golden hour

/usr/lib/python3.9/lib2to3/fixes

File/Folder	Size	Actions
__init__.py	47 B
fix_apply.py	2.29 KB
fix_asserts.py	984 B
fix_basestring.py	320 B
fix_buffer.py	590 B
fix_dict.py	3.67 KB
fix_except.py	3.27 KB
fix_exec.py	979 B
fix_execfile.py	2 KB
fix_exitfunc.py	2.44 KB
fix_filter.py	2.7 KB
fix_funcattrs.py	644 B
fix_future.py	547 B
fix_getcwdu.py	451 B
fix_has_key.py	3.12 KB
fix_idioms.py	4.76 KB
fix_import.py	3.18 KB
fix_imports.py	5.55 KB
fix_imports2.py	289 B
fix_input.py	708 B
fix_intern.py	1.12 KB
fix_isinstance.py	1.57 KB
fix_itertools.py	1.51 KB
fix_itertools_imports.py	2.04 KB
fix_long.py	476 B
fix_map.py	3.55 KB
fix_metaclass.py	8 KB
fix_methodattrs.py	606 B
fix_ne.py	571 B
fix_next.py	3.1 KB
fix_nonzero.py	591 B
fix_numliterals.py	768 B
fix_operator.py	3.35 KB
fix_paren.py	1.2 KB
fix_print.py	2.78 KB
fix_raise.py	2.86 KB
fix_raw_input.py	454 B
fix_reduce.py	837 B
fix_reload.py	1.06 KB
fix_renames.py	2.17 KB
fix_repr.py	613 B
fix_set_literal.py	1.66 KB
fix_standarderror.py	449 B
fix_sys_exc.py	1.01 KB
fix_throw.py	1.54 KB
fix_tuple_params.py	5.43 KB
fix_types.py	1.73 KB
fix_unicode.py	1.23 KB
fix_urllib.py	8.17 KB
fix_ws_comma.py	1.06 KB
fix_xrange.py	2.63 KB
fix_xreadlines.py	689 B
fix_zip.py	1.26 KB

Edit: fix_exitfunc.py

"""
Convert use of sys.exitfunc to use the atexit module.
"""

# Author: Benjamin Peterson

from lib2to3 import pytree, fixer_base
from lib2to3.fixer_util import Name, Attr, Call, Comma, Newline, syms

class FixExitfunc(fixer_base.BaseFix):
    keep_line_order = True
    BM_compatible = True

PATTERN = """
              (
                  sys_import=import_name<'import'
                      ('sys'
                      |
                      dotted_as_names< (any ',')* 'sys' (',' any)* >
                      )
                  >
              |
                  expr_stmt<
                      power< 'sys' trailer< '.' 'exitfunc' > >
                  '=' func=any >
              )
              """

def __init__(self, *args):
        super(FixExitfunc, self).__init__(*args)

def start_tree(self, tree, filename):
        super(FixExitfunc, self).start_tree(tree, filename)
        self.sys_import = None

def transform(self, node, results):
        # First, find the sys import. We'll just hope it's global scope.
        if "sys_import" in results:
            if self.sys_import is None:
                self.sys_import = results["sys_import"]
            return

func = results["func"].clone()
        func.prefix = ""
        register = pytree.Node(syms.power,
                               Attr(Name("atexit"), Name("register"))
                               )
        call = Call(register, [func], node.prefix)
        node.replace(call)

if self.sys_import is None:
            # That's interesting.
            self.warning(node, "Can't find sys import; Please add an atexit "
                             "import at the top of your file.")
            return

# Now add an atexit import after the sys import.
        names = self.sys_import.children[1]
        if names.type == syms.dotted_as_names:
            names.append_child(Comma())
            names.append_child(Name("atexit", " "))
        else:
            containing_stmt = self.sys_import.parent
            position = containing_stmt.children.index(self.sys_import)
            stmt_container = containing_stmt.parent
            new_import = pytree.Node(syms.import_name,
                              [Name("import"), Name("atexit", " ")]
                              )
            new = pytree.Node(syms.simple_stmt, [new_import])
            containing_stmt.insert_child(position + 1, Newline())
            containing_stmt.insert_child(position + 2, new)