0UT1S's

opt/alt/python310/lib64/python3.10/lib2to3/fixes/

By Order of Mr.0UT1S

N/A

[DIR] __pycache__

N/A

47 bytes

2.29 KB

984 bytes

fix_basestring.py

320 bytes

590 bytes

3.67 KB

3.27 KB

979 bytes

fix_execfile.py

2.00 KB

fix_exitfunc.py

2.44 KB

2.70 KB

fix_funcattrs.py

644 bytes

547 bytes

451 bytes

3.12 KB

4.76 KB

3.18 KB

5.55 KB

fix_imports2.py

289 bytes

708 bytes

1.12 KB

fix_isinstance.py

1.57 KB

fix_itertools.py

1.51 KB

fix_itertools_imports.py

2.04 KB

476 bytes

3.55 KB

fix_metaclass.py

8.00 KB

fix_methodattrs.py

606 bytes

571 bytes

3.10 KB

591 bytes

fix_numliterals.py

768 bytes

fix_operator.py

3.35 KB

1.20 KB

2.78 KB

2.86 KB

fix_raw_input.py

454 bytes

837 bytes

1.06 KB

2.17 KB

613 bytes

fix_set_literal.py

1.66 KB

fix_standarderror.py

449 bytes

1.01 KB

1.54 KB

fix_tuple_params.py

5.43 KB

1.73 KB

1.23 KB

8.17 KB

fix_ws_comma.py

1.06 KB

2.63 KB

fix_xreadlines.py

689 bytes

1.26 KB

"""
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)