openmedialibrary_platform/Darwin/lib/python3.5/lib2to3/fixes/fix_exitfunc.py

73 lines
2.4 KiB
Python
Raw Normal View History

2013-10-11 17:28:32 +00:00
"""
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):
2015-11-25 01:25:01 +00:00
# First, find the sys import. We'll just hope it's global scope.
2013-10-11 17:28:32 +00:00
if "sys_import" in results:
if self.sys_import is None:
self.sys_import = results["sys_import"]
return
func = results["func"].clone()
2014-09-30 16:15:32 +00:00
func.prefix = ""
2013-10-11 17:28:32 +00:00
register = pytree.Node(syms.power,
2014-09-30 16:15:32 +00:00
Attr(Name("atexit"), Name("register"))
2013-10-11 17:28:32 +00:00
)
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())
2014-09-30 16:15:32 +00:00
names.append_child(Name("atexit", " "))
2013-10-11 17:28:32 +00:00
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,
2014-09-30 16:15:32 +00:00
[Name("import"), Name("atexit", " ")]
2013-10-11 17:28:32 +00:00
)
new = pytree.Node(syms.simple_stmt, [new_import])
containing_stmt.insert_child(position + 1, Newline())
containing_stmt.insert_child(position + 2, new)