REDROOM

Navigation

Breadcrumbs

/ / lib64 / python3.6 / lib2to3 / fixes

Create

Create File | Create Folder | Upload Files | Run Command

Preview: fix_exitfunc.py Size: 2.44 KB

/lib64/python3.6/lib2to3/fixes/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)

Directory Contents

Dirs: 1 × Files: 53

Sort:

Enable drag-to-move

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2026-02-06 23:01:23	Edit Download recursive recursive
fix_apply.py	2.37 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_asserts.py	984 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_basestring.py	320 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_buffer.py	590 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_dict.py	3.67 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_except.py	3.27 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_exec.py	979 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_execfile.py	2.00 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_exitfunc.py	2.44 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_filter.py	2.59 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_funcattrs.py	644 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_future.py	547 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_getcwdu.py	451 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_has_key.py	3.12 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_idioms.py	4.76 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_import.py	3.18 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_imports.py	5.55 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_imports2.py	289 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_input.py	708 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_intern.py	1.21 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_isinstance.py	1.57 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_itertools.py	1.51 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_itertools_imports.py	2.04 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_long.py	476 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_map.py	3.55 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_metaclass.py	8.00 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_methodattrs.py	606 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_ne.py	571 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_next.py	3.10 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_nonzero.py	591 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_numliterals.py	768 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_operator.py	3.39 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_paren.py	1.20 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_print.py	2.78 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_raise.py	2.86 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_raw_input.py	454 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_reduce.py	837 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_reload.py	1.13 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_renames.py	2.17 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_repr.py	613 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_set_literal.py	1.66 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_standarderror.py	449 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_sys_exc.py	1.01 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_throw.py	1.54 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_tuple_params.py	5.43 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_types.py	1.73 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_unicode.py	1.23 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_urllib.py	8.16 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_ws_comma.py	1.06 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_xrange.py	2.63 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_xreadlines.py	689 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
fix_zip.py	1.26 KB	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive
__init__.py	47 B	lrw-r--r--	2018-12-23 21:37:14	Edit Download recursive recursive