Current Dir: /snap/core22/2437/usr/lib/python3.10/lib2to3/fixes/

[DIR]  __pycache__ [ delete | rename ]
[FILE] fix_apply.py [ edit | delete | rename ]
[FILE] fix_asserts.py [ edit | delete | rename ]
[FILE] fix_basestring.py [ edit | delete | rename ]
[FILE] fix_buffer.py [ edit | delete | rename ]
[FILE] fix_dict.py [ edit | delete | rename ]
[FILE] fix_except.py [ edit | delete | rename ]
[FILE] fix_exec.py [ edit | delete | rename ]
[FILE] fix_execfile.py [ edit | delete | rename ]
[FILE] fix_exitfunc.py [ edit | delete | rename ]
[FILE] fix_filter.py [ edit | delete | rename ]
[FILE] fix_funcattrs.py [ edit | delete | rename ]
[FILE] fix_future.py [ edit | delete | rename ]
[FILE] fix_getcwdu.py [ edit | delete | rename ]
[FILE] fix_has_key.py [ edit | delete | rename ]
[FILE] fix_idioms.py [ edit | delete | rename ]
[FILE] fix_import.py [ edit | delete | rename ]
[FILE] fix_imports.py [ edit | delete | rename ]
[FILE] fix_imports2.py [ edit | delete | rename ]
[FILE] fix_input.py [ edit | delete | rename ]
[FILE] fix_intern.py [ edit | delete | rename ]
[FILE] fix_isinstance.py [ edit | delete | rename ]
[FILE] fix_itertools.py [ edit | delete | rename ]
[FILE] fix_itertools_imports.py [ edit | delete | rename ]
[FILE] fix_long.py [ edit | delete | rename ]
[FILE] fix_map.py [ edit | delete | rename ]
[FILE] fix_metaclass.py [ edit | delete | rename ]
[FILE] fix_methodattrs.py [ edit | delete | rename ]
[FILE] fix_ne.py [ edit | delete | rename ]
[FILE] fix_next.py [ edit | delete | rename ]
[FILE] fix_nonzero.py [ edit | delete | rename ]
[FILE] fix_numliterals.py [ edit | delete | rename ]
[FILE] fix_operator.py [ edit | delete | rename ]
[FILE] fix_paren.py [ edit | delete | rename ]
[FILE] fix_print.py [ edit | delete | rename ]
[FILE] fix_raise.py [ edit | delete | rename ]
[FILE] fix_raw_input.py [ edit | delete | rename ]
[FILE] fix_reduce.py [ edit | delete | rename ]
[FILE] fix_reload.py [ edit | delete | rename ]
[FILE] fix_renames.py [ edit | delete | rename ]
[FILE] fix_repr.py [ edit | delete | rename ]
[FILE] fix_set_literal.py [ edit | delete | rename ]
[FILE] fix_standarderror.py [ edit | delete | rename ]
[FILE] fix_sys_exc.py [ edit | delete | rename ]
[FILE] fix_throw.py [ edit | delete | rename ]
[FILE] fix_tuple_params.py [ edit | delete | rename ]
[FILE] fix_types.py [ edit | delete | rename ]
[FILE] fix_unicode.py [ edit | delete | rename ]
[FILE] fix_urllib.py [ edit | delete | rename ]
[FILE] fix_ws_comma.py [ edit | delete | rename ]
[FILE] fix_xrange.py [ edit | delete | rename ]
[FILE] fix_xreadlines.py [ edit | delete | rename ]
[FILE] fix_zip.py [ edit | delete | rename ]
[FILE] __init__.py [ edit | delete | rename ]

Viewing: /snap/core22/2437/usr/lib/python3.10/lib2to3/fixes/fix_except.py

"""Fixer for except statements with named exceptions.

The following cases will be converted:

- "except E, T:" where T is a name:

    except E as T:

- "except E, T:" where T is not a name, tuple or list:

        except E as t:
            T = t

    This is done because the target of an "except" clause must be a
    name.

- "except E, T:" where T is a tuple or list literal:

        except E as t:
            T = t.args
"""
# Author: Collin Winter

# Local imports
from .. import pytree
from ..pgen2 import token
from .. import fixer_base
from ..fixer_util import Assign, Attr, Name, is_tuple, is_list, syms

def find_excepts(nodes):
    for i, n in enumerate(nodes):
        if n.type == syms.except_clause:
            if n.children[0].value == 'except':
                yield (n, nodes[i+2])

class FixExcept(fixer_base.BaseFix):
    BM_compatible = True

    PATTERN = """
    try_stmt< 'try' ':' (simple_stmt | suite)
                  cleanup=(except_clause ':' (simple_stmt | suite))+
                  tail=(['except' ':' (simple_stmt | suite)]
                        ['else' ':' (simple_stmt | suite)]
                        ['finally' ':' (simple_stmt | suite)]) >
    """

    def transform(self, node, results):
        syms = self.syms

        tail = [n.clone() for n in results["tail"]]

        try_cleanup = [ch.clone() for ch in results["cleanup"]]
        for except_clause, e_suite in find_excepts(try_cleanup):
            if len(except_clause.children) == 4:
                (E, comma, N) = except_clause.children[1:4]
                comma.replace(Name("as", prefix=" "))

                if N.type != token.NAME:
                    # Generate a new N for the except clause
                    new_N = Name(self.new_name(), prefix=" ")
                    target = N.clone()
                    target.prefix = ""
                    N.replace(new_N)
                    new_N = new_N.clone()

                    # Insert "old_N = new_N" as the first statement in
                    #  the except body. This loop skips leading whitespace
                    #  and indents
                    #TODO(cwinter) suite-cleanup
                    suite_stmts = e_suite.children
                    for i, stmt in enumerate(suite_stmts):
                        if isinstance(stmt, pytree.Node):
                            break

                    # The assignment is different if old_N is a tuple or list
                    # In that case, the assignment is old_N = new_N.args
                    if is_tuple(N) or is_list(N):
                        assign = Assign(target, Attr(new_N, Name('args')))
                    else:
                        assign = Assign(target, new_N)

                    #TODO(cwinter) stopgap until children becomes a smart list
                    for child in reversed(suite_stmts[:i]):
                        e_suite.insert_child(0, child)
                    e_suite.insert_child(i, assign)
                elif N.prefix == "":
                    # No space after a comma is legal; no space after "as",
                    # not so much.
                    N.prefix = " "

        #TODO(cwinter) fix this when children becomes a smart list
        children = [c.clone() for c in node.children[:3]] + try_cleanup + tail
        return pytree.Node(node.type, children)

Upload File: