mirror of
https://github.com/python/cpython.git
synced 2025-07-31 23:23:11 +00:00
6ae94ee299
svn+ssh://pythondev@svn.python.org/sandbox/trunk/2to3/lib2to3 ........ r66805 | benjamin.peterson | 2008-10-04 20:11:02 -0500 (Sat, 04 Oct 2008) | 1 line mention what the fixes directory is for ........ r66841 | benjamin.peterson | 2008-10-07 17:48:12 -0500 (Tue, 07 Oct 2008) | 1 line use assertFalse and assertTrue ........ r66860 | benjamin.peterson | 2008-10-08 16:05:07 -0500 (Wed, 08 Oct 2008) | 1 line instead of abusing the pattern matcher, use start_tree to find a next binding ........ r66884 | benjamin.peterson | 2008-10-13 15:50:30 -0500 (Mon, 13 Oct 2008) | 1 line don't print tokens to stdout when -v is given ........ r66885 | benjamin.peterson | 2008-10-13 16:28:57 -0500 (Mon, 13 Oct 2008) | 1 line add the -x option to disable fixers ........ r66886 | benjamin.peterson | 2008-10-13 16:33:53 -0500 (Mon, 13 Oct 2008) | 1 line cut down on some crud ........ r66893 | benjamin.peterson | 2008-10-14 17:16:54 -0500 (Tue, 14 Oct 2008) | 1 line add an optional set literal fixer ........ r66907 | benjamin.peterson | 2008-10-15 16:59:41 -0500 (Wed, 15 Oct 2008) | 1 line don't write backup files by default ........ r66910 | benjamin.peterson | 2008-10-15 17:43:10 -0500 (Wed, 15 Oct 2008) | 1 line add the -n option; it stops backupfiles from being written ........
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
"""
|
|
Optional fixer to transform set() calls to set literals.
|
|
"""
|
|
|
|
# Author: Benjamin Peterson
|
|
|
|
from lib2to3 import fixer_base, pytree
|
|
from lib2to3.fixer_util import token, syms
|
|
|
|
|
|
|
|
class FixSetLiteral(fixer_base.BaseFix):
|
|
|
|
explicit = True
|
|
|
|
PATTERN = """power< 'set' trailer< '('
|
|
(atom=atom< '[' (items=listmaker< any ((',' any)* [',']) >
|
|
|
|
|
single=any) ']' >
|
|
|
|
|
atom< '(' items=testlist_gexp< any ((',' any)* [',']) > ')' >
|
|
)
|
|
')' > >
|
|
"""
|
|
|
|
def transform(self, node, results):
|
|
single = results.get("single")
|
|
if single:
|
|
# Make a fake listmaker
|
|
fake = pytree.Node(syms.listmaker, [single.clone()])
|
|
single.replace(fake)
|
|
items = fake
|
|
else:
|
|
items = results["items"]
|
|
|
|
# Build the contents of the literal
|
|
literal = [pytree.Leaf(token.LBRACE, "{")]
|
|
literal.extend(n.clone() for n in items.children)
|
|
literal.append(pytree.Leaf(token.RBRACE, "}"))
|
|
# Set the prefix of the right brace to that of the ')' or ']'
|
|
literal[-1].set_prefix(items.get_next_sibling().get_prefix())
|
|
maker = pytree.Node(syms.dictsetmaker, literal)
|
|
maker.set_prefix(node.get_prefix())
|
|
|
|
# If the original was a one tuple, we need to remove the extra comma.
|
|
if len(maker.children) == 4:
|
|
n = maker.children[2]
|
|
n.remove()
|
|
maker.children[-1].set_prefix(n.get_prefix())
|
|
|
|
# Finally, replace the set call with our shiny new literal.
|
|
return maker
|