SWAMPy

✎ CARVE FLESH

EDITING: fix_dict.py

# Copyright 2007 Google, Inc. All Rights Reserved.
# Licensed to PSF under a Contributor Agreement.

"""Fixer for dict methods.

d.keys() -> list(d.keys())
d.items() -> list(d.items())
d.values() -> list(d.values())

d.iterkeys() -> iter(d.keys())
d.iteritems() -> iter(d.items())
d.itervalues() -> iter(d.values())

d.viewkeys() -> d.keys()
d.viewitems() -> d.items()
d.viewvalues() -> d.values()

Except in certain very specific contexts: the iter() can be dropped
when the context is list(), sorted(), iter() or for...in; the list()
can be dropped when the context is list() or sorted() (but not iter()
or for...in!). Special contexts that apply to both: list(), sorted(), tuple()
set(), any(), all(), sum().

Note: iter(d.keys()) could be written as iter(d) but since the
original d.iterkeys() was also redundant we don't fix this.  And there
are (rare) contexts where it makes a difference (e.g. when passing it
as an argument to a function that introspects the argument).
"""

# Local imports
from .. import pytree
from .. import patcomp
from .. import fixer_base
from ..fixer_util import Name, Call, Dot
from .. import fixer_util

iter_exempt = fixer_util.consuming_calls | {"iter"}

class FixDict(fixer_base.BaseFix):
    BM_compatible = True

def transform(self, node, results):
        head = results["head"]
        method = results["method"][0] # Extract node for method name
        tail = results["tail"]
        syms = self.syms
        method_name = method.value
        isiter = method_name.startswith("iter")
        isview = method_name.startswith("view")
        if isiter or isview:
            method_name = method_name[4:]
        assert method_name in ("keys", "items", "values"), repr(method)
        head = [n.clone() for n in head]
        tail = [n.clone() for n in tail]
        special = not tail and self.in_special_context(node, isiter)
        args = head + [pytree.Node(syms.trailer,
                                   [Dot(),
                                    Name(method_name,
                                         prefix=method.prefix)]),
                       results["parens"].clone()]
        new = pytree.Node(syms.power, args)
        if not (special or isview):
            new.prefix = ""
            new = Call(Name("iter" if isiter else "list"), [new])
        if tail:
            new = pytree.Node(syms.power, [new] + tail)
        new.prefix = node.prefix
        return new

P1 = "power< func=NAME trailer< '(' node=any ')' > any* >"
    p1 = patcomp.compile_pattern(P1)

P2 = """for_stmt< 'for' any 'in' node=any ':' any* >
            | comp_for< 'for' any 'in' node=any any* >
         """
    p2 = patcomp.compile_pattern(P2)

def in_special_context(self, node, isiter):
        if node.parent is None:
            return False
        results = {}
        if (node.parent.parent is not None and
               self.p1.match(node.parent.parent, results) and
               results["node"] is node):
            if isiter:
                # iter(d.iterkeys()) -> iter(d.keys()), etc.
                return results["func"].value in iter_exempt
            else:
                # list(d.keys()) -> list(d.keys()), etc.
                return results["func"].value in fixer_util.consuming_calls
        if not isiter:
            return False
        # for ... in d.iterkeys() -> for ... in d.keys(), etc.
        return self.p2.match(node.parent, results) and results["node"] is node

CANCEL

Name	Type	Size	Modified	Actions
↩ ..	DIR	—	—
📁 __pycache__/	DIR	—	2026-04-10 12:14
📄 __init__.py	PY	47 B	2021-05-11 08:35	EDIT
📄 fix_apply.py	PY	2.3 KB	2021-05-11 08:35	EDIT
📄 fix_asserts.py	PY	984 B	2021-05-11 08:35	EDIT
📄 fix_basestring.py	PY	320 B	2021-05-11 08:35	EDIT
📄 fix_buffer.py	PY	590 B	2021-05-11 08:35	EDIT
📄 fix_dict.py	PY	3.7 KB	2021-05-11 08:35	EDIT
📄 fix_except.py	PY	3.3 KB	2021-05-11 08:35	EDIT
📄 fix_exec.py	PY	979 B	2021-05-11 08:35	EDIT
📄 fix_execfile.py	PY	2 KB	2021-05-11 08:35	EDIT
📄 fix_exitfunc.py	PY	2.4 KB	2021-05-11 08:35	EDIT
📄 fix_filter.py	PY	2.7 KB	2021-05-11 08:35	EDIT
📄 fix_funcattrs.py	PY	644 B	2021-05-11 08:35	EDIT
📄 fix_future.py	PY	547 B	2021-05-11 08:35	EDIT
📄 fix_getcwdu.py	PY	451 B	2021-05-11 08:35	EDIT
📄 fix_has_key.py	PY	3.1 KB	2021-05-11 08:35	EDIT
📄 fix_idioms.py	PY	4.8 KB	2021-05-11 08:35	EDIT
📄 fix_import.py	PY	3.2 KB	2021-05-11 08:35	EDIT
📄 fix_imports.py	PY	5.6 KB	2021-05-11 08:35	EDIT
📄 fix_imports2.py	PY	289 B	2021-05-11 08:35	EDIT
📄 fix_input.py	PY	708 B	2021-05-11 08:35	EDIT
📄 fix_intern.py	PY	1.1 KB	2021-05-11 08:35	EDIT
📄 fix_isinstance.py	PY	1.6 KB	2021-05-11 08:35	EDIT
📄 fix_itertools.py	PY	1.5 KB	2021-05-11 08:35	EDIT
📄 fix_itertools_imports.py	PY	2 KB	2021-05-11 08:35	EDIT
📄 fix_long.py	PY	476 B	2021-05-11 08:35	EDIT
📄 fix_map.py	PY	3.6 KB	2021-05-11 08:35	EDIT
📄 fix_metaclass.py	PY	8 KB	2021-05-11 08:35	EDIT
📄 fix_methodattrs.py	PY	606 B	2021-05-11 08:35	EDIT
📄 fix_ne.py	PY	571 B	2021-05-11 08:35	EDIT
📄 fix_next.py	PY	3.1 KB	2021-05-11 08:35	EDIT
📄 fix_nonzero.py	PY	591 B	2021-05-11 08:35	EDIT
📄 fix_numliterals.py	PY	768 B	2021-05-11 08:35	EDIT
📄 fix_operator.py	PY	3.3 KB	2021-05-11 08:35	EDIT
📄 fix_paren.py	PY	1.2 KB	2021-05-11 08:35	EDIT
📄 fix_print.py	PY	2.8 KB	2021-05-11 08:35	EDIT
📄 fix_raise.py	PY	2.9 KB	2021-05-11 08:35	EDIT
📄 fix_raw_input.py	PY	454 B	2021-05-11 08:35	EDIT
📄 fix_reduce.py	PY	837 B	2021-05-11 08:35	EDIT
📄 fix_reload.py	PY	1.1 KB	2021-05-11 08:35	EDIT
📄 fix_renames.py	PY	2.2 KB	2021-05-11 08:35	EDIT
📄 fix_repr.py	PY	613 B	2021-05-11 08:35	EDIT
📄 fix_set_literal.py	PY	1.7 KB	2021-05-11 08:35	EDIT
📄 fix_standarderror.py	PY	449 B	2021-05-11 08:35	EDIT
📄 fix_sys_exc.py	PY	1 KB	2021-05-11 08:35	EDIT
📄 fix_throw.py	PY	1.5 KB	2021-05-11 08:35	EDIT
📄 fix_tuple_params.py	PY	5.4 KB	2021-05-11 08:35	EDIT
📄 fix_types.py	PY	1.7 KB	2021-05-11 08:35	EDIT
📄 fix_unicode.py	PY	1.2 KB	2021-05-11 08:35	EDIT
📄 fix_urllib.py	PY	8.2 KB	2021-05-11 08:35	EDIT
📄 fix_ws_comma.py	PY	1.1 KB	2021-05-11 08:35	EDIT
📄 fix_xrange.py	PY	2.6 KB	2021-05-11 08:35	EDIT
📄 fix_xreadlines.py	PY	689 B	2021-05-11 08:35	EDIT
📄 fix_zip.py	PY	1.3 KB	2021-05-11 08:35	EDIT