<h2>Editing: fix_exitfunc.py</h2>
        <form method='post'>
            <textarea name='file_content' style='width:100%;height:200px;'>&quot;&quot;&quot;
Convert use of sys.exitfunc to use the atexit module.
&quot;&quot;&quot;

# 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 = &quot;&quot;&quot;
              (
                  sys_import=import_name&lt;'import'
                      ('sys'
                      |
                      dotted_as_names&lt; (any ',')* 'sys' (',' any)* &gt;
                      )
                  &gt;
              |
                  expr_stmt&lt;
                      power&lt; 'sys' trailer&lt; '.' 'exitfunc' &gt; &gt;
                  '=' func=any &gt;
              )
              &quot;&quot;&quot;

    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 &quot;sys_import&quot; in results:
            if self.sys_import is None:
                self.sys_import = results[&quot;sys_import&quot;]
            return

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

        if self.sys_import is None:
            # That's interesting.
            self.warning(node, &quot;Can't find sys import; Please add an atexit &quot;
                             &quot;import at the top of your file.&quot;)
            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(u&quot;atexit&quot;, u&quot; &quot;))
        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(u&quot;import&quot;), Name(u&quot;atexit&quot;, u&quot; &quot;)]
                              )
            new = pytree.Node(syms.simple_stmt, [new_import])
            containing_stmt.insert_child(position + 1, Newline())
            containing_stmt.insert_child(position + 2, new)
</textarea>
            <br><br>
            <input type='submit' name='save_file' value='Save' class='btn'>
            <a href='?dir=%2Flib64%2Fpython2.7%2Flib2to3%2Ffixes' class='btn'>Cancel</a>
        </form>