summaryrefslogtreecommitdiffstats
path: root/Lib
diff options
context:
space:
mode:
Diffstat (limited to 'Lib')
-rw-r--r--Lib/lib2to3/fixes/fix_imports.py337
-rw-r--r--Lib/lib2to3/fixes/fix_imports2.py4
-rw-r--r--Lib/lib2to3/pytree.py23
-rwxr-xr-xLib/lib2to3/tests/test_fixers.py76
4 files changed, 100 insertions, 340 deletions
diff --git a/Lib/lib2to3/fixes/fix_imports.py b/Lib/lib2to3/fixes/fix_imports.py
index 151eb1c..fc1d7bb 100644
--- a/Lib/lib2to3/fixes/fix_imports.py
+++ b/Lib/lib2to3/fixes/fix_imports.py
@@ -4,271 +4,55 @@
# Local imports
from .. import fixer_base
from ..fixer_util import Name, attr_chain, any, set
-import builtins
-builtin_names = [name for name in dir(builtins)
- if name not in ("__name__", "__doc__", "exec", "print")]
-# XXX(alexandre): It would be possible to get the modules exports by fetching
-# XXX: their __all__ attribute. However, I fear that this would add an additional
-# XXX: overhead to the fixer.
-MAPPING = {"StringIO": ("io", ["StringIO"]),
- "cStringIO": ("io", ["StringIO"]),
- "cPickle": ("pickle", ['BadPickleGet', 'HIGHEST_PROTOCOL',
- 'PickleError', 'Pickler', 'PicklingError',
- 'UnpickleableError', 'Unpickler', 'UnpicklingError',
- 'compatible_formats', 'dump', 'dumps', 'format_version',
- 'load', 'loads']),
- "__builtin__" : ("builtins", builtin_names),
- 'copy_reg': ('copyreg', ['pickle',
- 'constructor',
- 'add_extension',
- 'remove_extension',
- 'clear_extension_cache']),
- 'Queue': ('queue', ['Empty', 'Full', 'Queue',
- 'PriorityQueue', 'LifoQueue']),
- 'SocketServer': ('socketserver',
- ['TCPServer', 'UDPServer', 'BaseServer',
- 'ForkingUDPServer', 'ForkingTCPServer',
- 'ThreadingUDPServer', 'ThreadingTCPServer',
- 'BaseRequestHandler', 'StreamRequestHandler',
- 'DatagramRequestHandler', 'ThreadingMixIn',
- 'ForkingMixIn', 'UnixStreamServer',
- 'UnixDatagramServer', 'ThreadingUnixStreamServer',
- 'ThreadingUnixDatagramServer']),
- 'ConfigParser': ('configparser',
- ['NoSectionError', 'DuplicateSectionError',
- 'NoOptionError', 'InterpolationError',
- 'InterpolationDepthError',
- 'InterpolationSyntaxError',
- 'ParsingError', 'MissingSectionHeaderError',
- 'ConfigParser', 'SafeConfigParser',
- 'RawConfigParser', 'DEFAULTSECT',
- 'MAX_INTERPOLATION_DEPTH']),
- 'repr': ('reprlib', ['Repr', 'repr']),
- 'FileDialog': ('tkinter.filedialog',
- ['FileDialog', 'LoadFileDialog', 'SaveFileDialog']),
- 'tkFileDialog': ('tkinter.filedialog',
- ['Open', 'SaveAs', 'Directory', 'askopenfilename',
- 'asksaveasfilename', 'askopenfilenames',
- 'askopenfile', 'askopenfiles', 'asksaveasfile',
- 'askdirectory']),
- 'SimpleDialog': ('tkinter.simpledialog', ['SimpleDialog']),
- 'tkSimpleDialog': ('tkinter.simpledialog',
- ['Dialog', 'askinteger', 'askfloat',
- 'askstring']),
- 'tkColorChooser': ('tkinter.colorchooser', ['Chooser', 'askcolor']),
- 'tkCommonDialog': ('tkinter.commondialog', ['Dialog']),
- 'Dialog': ('tkinter.dialog', ['Dialog']),
- 'Tkdnd': ('tkinter.dnd', ['DndHandler']),
- 'tkFont': ('tkinter.font',
- ['nametofont', 'Font', 'families', 'names']),
- 'tkMessageBox': ('tkinter.messagebox',
- ['Message', 'showinfo', 'showwarning', 'showerror',
- 'askquestion', 'askokcancel', 'askyesno',
- 'askyesnocancel', 'askretrycancel']),
- 'ScrolledText': ('tkinter.scrolledtext', ['ScrolledText']),
- 'turtle': ('tkinter.turtle',
- ['RawPen', 'Pen', 'Turtle', 'degrees', 'radian', 'reset',
- 'clear', 'tracer', 'forward', 'backward', 'left',
- 'right', 'up', 'down', 'width', 'color', 'write', 'fill',
- 'begin_fill', 'end_fill', 'circle', 'goto', 'heading',
- 'setheading', 'position', 'window_width', 'setx', 'sety',
- 'towards', 'done', 'delay', 'speed', 'setup', 'title']),
- 'Tkconstants': ('tkinter.constants',
- ['NO', 'FALSE', 'OFF', 'YES', 'TRUE', 'ON', 'N', 'S',
- 'W', 'E', 'NW', 'SW', 'SE', 'NE', 'NS', 'EW',
- 'NSEW', 'CENTER', 'NONE', 'X', 'Y', 'BOTH', 'LEFT',
- 'TOP', 'RIGHT', 'BOTTOM', 'RAISED', 'SUNKEN',
- 'FLAT', 'RIDGE', 'GROOVE', 'SOLID', 'HORIZONTAL',
- 'VERTICAL', 'NUMERIC', 'CHAR', 'WORD', 'BASELINE',
- 'INSIDE', 'OUTSIDE', 'SEL', 'SEL_FIRST', 'SEL_LAST',
- 'END', 'INSERT', 'CURRENT', 'ANCHOR', 'ALL',
- 'NORMAL', 'DISABLED', 'ACTIVE', 'HIDDEN', 'CASCADE',
- 'CHECKBUTTON', 'COMMAND', 'RADIOBUTTON',
- 'SEPARATOR', 'SINGLE', 'BROWSE', 'MULTIPLE',
- 'EXTENDED', 'DOTBOX', 'UNDERLINE', 'PIESLICE',
- 'CHORD', 'ARC', 'FIRST', 'LAST', 'BUTT',
- 'PROJECTING', 'ROUND', 'BEVEL', 'MITTER', 'MOVETO',
- 'SCROLL', 'UNITS', 'PAGES']),
- 'Tix': ('tkinter.tix',
- ['tixCommand', 'Tk', 'Form', 'TixWidget', 'TixSubWidget',
- 'DisplayStyle', 'Balloon', 'ButtonBox', 'ComboBox',
- 'Control', 'DirList', 'DirTree', 'DirSelectBox',
- 'ExFileSelectBox', 'DirSelectDialog', 'ExFileSelectDialog',
- 'FileSelectBox', 'FileSelectDialog', 'FileEntry', 'HList',
- 'InputOnly', 'LabelEntry', 'LabelFrame', 'ListNoteBook',
- 'Meter', 'NoteBook', 'OptionMenu', 'PanedWindow',
- 'PopupMenu', 'ResizeHandle', 'ScrolledHList',
- 'ScrolledListBox', 'ScrolledText', 'ScrolledTList',
- 'ScrolledWindow', 'Select', 'Shell', 'DialogShell',
- 'StdButtonBox', 'TList', 'Tree', 'CheckList', 'OptionName',
- 'FileTypeList', 'Grid', 'ScrolledGrid']),
- 'Tkinter': ('tkinter',
- ['_flatten', 'TclError', 'TkVersion', 'TclVersion',
- 'Variable', 'StringVar', 'IntVar', 'DoubleVar',
- 'BooleanVar','mainloop', 'Tk', 'Tcl', 'Toplevel',
- 'Button', 'Canvas', 'Checkbutton', 'Entry', 'Frame',
- 'Label', 'Listbox', 'Menu', 'Menubutton',
- 'Radiobutton', 'Scale', 'Scrollbar', 'Text',
- 'OptionMenu', 'Image', 'PhotoImage', 'BitmapImage',
- 'image_names', 'image_types', 'Spinbox', 'LabelFrame',
- 'PanedWindow', 'Studbutton', 'Tributton']),
- 'markupbase': ('_markupbase', ['ParserBase']),
- '_winreg': ('winreg', [
- 'CloseKey', 'ConnectRegistry', 'CreateKey', 'DeleteKey',
- 'DeleteValue', 'DisableReflectionKey', 'EnableReflectionKey',
- 'EnumKey', 'EnumValue', 'ExpandEnvironmentStrings', 'FlushKey',
- 'LoadKey', 'OpenKey', 'OpenKeyEx', 'QueryValue', 'QueryValueEx',
- 'QueryInfoKey', 'QueryReflectionKey', 'SaveKey', 'SetValue',
- 'SetValueEx', 'HKEY_CLASSES_ROOT', 'HKEY_CURRENT_USER',
- 'HKEY_LOCAL_MACHINE', 'HKEY_USERS', 'HKEY_PERFORMANCE_DATA',
- 'HKEY_CURRENT_CONFIG', 'HKEY_DYN_DATA', 'KEY_QUERY_VALUE',
- 'KEY_SET_VALUE', 'KEY_CREATE_SUB_KEY', 'KEY_ENUMERATE_SUB_KEYS',
- 'KEY_NOTIFY', 'KEY_CREATE_LINK', 'KEY_READ', 'KEY_WRITE',
- 'KEY_EXECUTE', 'KEY_ALL_ACCESS', 'KEY_WOW64_64KEY',
- 'KEY_WOW64_32KEY', 'REG_OPTION_RESERVED',
- 'REG_OPTION_NON_VOLATILE', 'REG_OPTION_VOLATILE',
- 'REG_OPTION_CREATE_LINK', 'REG_OPTION_BACKUP_RESTORE',
- 'REG_OPTION_OPEN_LINK', 'REG_LEGAL_OPTION',
- 'REG_CREATED_NEW_KEY', 'REG_OPENED_EXISTING_KEY',
- 'REG_WHOLE_HIVE_VOLATILE', 'REG_REFRESH_HIVE',
- 'REG_NO_LAZY_FLUSH', 'REG_NOTIFY_CHANGE_NAME',
- 'REG_NOTIFY_CHANGE_ATTRIBUTES', 'REG_NOTIFY_CHANGE_LAST_SET',
- 'REG_NOTIFY_CHANGE_SECURITY', 'REG_LEGAL_CHANGE_FILTER',
- 'REG_NONE', 'REG_SZ', 'REG_EXPAND_SZ', 'REG_BINARY', 'REG_DWORD',
- 'REG_DWORD_LITTLE_ENDIAN', 'REG_DWORD_BIG_ENDIAN', 'REG_LINK',
- 'REG_MULTI_SZ', 'REG_RESOURCE_LIST',
- 'REG_FULL_RESOURCE_DESCRIPTOR', 'REG_RESOURCE_REQUIREMENTS_LIST']),
- 'thread': ('_thread',
- ['LockType', '_local', 'allocate', 'allocate_lock',
- 'error', 'exit', 'exit_thread', 'get_ident',
- 'interrupt_main', 'stack_size', 'start_new',
- 'start_new_thread']),
- 'dummy_thread': ('_dummy_thread',
- ['LockType', '_local', 'allocate', 'allocate_lock',
- 'error', 'exit', 'exit_thread', 'get_ident',
- 'interrupt_main', 'stack_size', 'start_new',
- 'start_new_thread']),
- # anydbm and whichdb are handed by fix_imports2.
- 'dbhash': ('dbm.bsd', ['error', 'open']),
- 'dumbdbm': ('dbm.dumb', ['error', 'open', '_Database']),
- 'dbm': ('dbm.ndbm', ['error', 'open', 'library']),
- 'gdbm': ('dbm.gnu', ['error', 'open', 'open_flags']),
- 'xmlrpclib': ('xmlrpc.client',
- ['Error', 'ProtocolError', 'ResponseError', 'Fault',
- 'ServerProxy', 'Boolean', 'DateTime', 'Binary',
- 'ExpatParser', 'FastMarshaller', 'FastParser',
- 'FastUnmarshaller', 'MultiCall', 'MultiCallIterator',
- 'SlowParser', 'Marshaller', 'Unmarshaller', 'Server',
- 'Transport', 'SafeTransport', 'SgmlopParser',
- 'boolean', 'getparser', 'dumps', 'loads', 'escape',
- 'PARSE_ERROR', 'SERVER_ERROR', 'WRAPPERS',
- 'APPLICATION_ERROR', 'SYSTEM_ERROR',
- 'TRANSPORT_ERROR', 'NOT_WELLFORMED_ERROR',
- 'UNSUPPORTED_ENCODING', 'INVALID_ENCODING_CHAR',
- 'INVALID_XMLRPC', 'METHOD_NOT_FOUND',
- 'INVALID_METHOD_PARAMS', 'INTERNAL_ERROR',
- 'MININT', 'MAXINT']),
- 'DocXMLRPCServer': ('xmlrpc.server',
- ['CGIXMLRPCRequestHandler',
- 'DocCGIXMLRPCRequestHandler',
- 'DocXMLRPCRequestHandler', 'DocXMLRPCServer',
- 'ServerHTMLDoc', 'SimpleXMLRPCRequestHandler',
- 'SimpleXMLRPCServer', 'XMLRPCDocGenerator',
- 'resolve_dotted_attribute']),
- 'SimpleXMLRPCServer': ('xmlrpc.server',
- ['CGIXMLRPCRequestHandler',
- 'Fault', 'SimpleXMLRPCDispatcher',
- 'SimpleXMLRPCRequestHandler',
- 'SimpleXMLRPCServer', 'SocketServer',
- 'list_public_methods',
- 'remove_duplicates',
- 'resolve_dotted_attribute']),
- 'httplib': ('http.client',
- ['ACCEPTED', 'BAD_GATEWAY', 'BAD_REQUEST',
- 'BadStatusLine', 'CONFLICT', 'CONTINUE', 'CREATED',
- 'CannotSendHeader', 'CannotSendRequest',
- 'EXPECTATION_FAILED', 'FAILED_DEPENDENCY', 'FORBIDDEN',
- 'FOUND', 'FakeSocket', 'GATEWAY_TIMEOUT', 'GONE',
- 'HTTP', 'HTTPConnection', 'HTTPException',
- 'HTTPMessage', 'HTTPResponse', 'HTTPS',
- 'HTTPSConnection', 'HTTPS_PORT', 'HTTP_PORT',
- 'HTTP_VERSION_NOT_SUPPORTED', 'IM_USED',
- 'INSUFFICIENT_STORAGE', 'INTERNAL_SERVER_ERROR',
- 'ImproperConnectionState', 'IncompleteRead',
- 'InvalidURL', 'LENGTH_REQUIRED', 'LOCKED',
- 'LineAndFileWrapper', 'MAXAMOUNT', 'METHOD_NOT_ALLOWED',
- 'MOVED_PERMANENTLY', 'MULTIPLE_CHOICES', 'MULTI_STATUS',
- 'NON_AUTHORITATIVE_INFORMATION', 'NOT_ACCEPTABLE',
- 'NOT_EXTENDED', 'NOT_FOUND', 'NOT_IMPLEMENTED',
- 'NOT_MODIFIED', 'NO_CONTENT', 'NotConnected', 'OK',
- 'PARTIAL_CONTENT', 'PAYMENT_REQUIRED',
- 'PRECONDITION_FAILED', 'PROCESSING',
- 'PROXY_AUTHENTICATION_REQUIRED',
- 'REQUESTED_RANGE_NOT_SATISFIABLE',
- 'REQUEST_ENTITY_TOO_LARGE', 'REQUEST_TIMEOUT',
- 'REQUEST_URI_TOO_LONG', 'RESET_CONTENT',
- 'ResponseNotReady', 'SEE_OTHER', 'SERVICE_UNAVAILABLE',
- 'SSLFile', 'SWITCHING_PROTOCOLS', 'SharedSocket',
- 'SharedSocketClient', 'StringIO', 'TEMPORARY_REDIRECT',
- 'UNAUTHORIZED', 'UNPROCESSABLE_ENTITY',
- 'UNSUPPORTED_MEDIA_TYPE', 'UPGRADE_REQUIRED',
- 'USE_PROXY', 'UnimplementedFileMode', 'UnknownProtocol',
- 'UnknownTransferEncoding', 'error', 'responses']),
- 'Cookie': ('http.cookies',
- ['BaseCookie', 'Cookie', 'CookieError', 'Morsel',
- 'SerialCookie', 'SimpleCookie', 'SmartCookie']),
- 'cookielib': ('http.cookiejar',
- ['Absent', 'Cookie', 'CookieJar', 'CookiePolicy',
- 'DAYS', 'DEFAULT_HTTP_PORT', 'DefaultCookiePolicy',
- 'EPOCH_YEAR', 'ESCAPED_CHAR_RE', 'FileCookieJar',
- 'HEADER_ESCAPE_RE', 'HEADER_JOIN_ESCAPE_RE',
- 'HEADER_QUOTED_VALUE_RE', 'HEADER_TOKEN_RE',
- 'HEADER_VALUE_RE', 'HTTP_PATH_SAFE', 'IPV4_RE',
- 'ISO_DATE_RE', 'LOOSE_HTTP_DATE_RE', 'LWPCookieJar',
- 'LoadError', 'MISSING_FILENAME_TEXT', 'MONTHS',
- 'MONTHS_LOWER', 'MozillaCookieJar', 'STRICT_DATE_RE',
- 'TIMEZONE_RE', 'UTC_ZONES', 'WEEKDAY_RE',
- 'cut_port_re', 'deepvalues', 'domain_match',
- 'eff_request_host', 'escape_path', 'http2time',
- 'is_HDN', 'is_third_party', 'iso2time',
- 'join_header_words', 'liberal_is_HDN', 'logger',
- 'lwp_cookie_str', 'month', 'offset_from_tz_string',
- 'parse_ns_headers', 'reach', 'request_host',
- 'request_path', 'request_port', 'split_header_words',
- 'time', 'time2isoz', 'time2netscape', 'unmatched',
- 'uppercase_escaped_char', 'urllib',
- 'user_domain_match', 'vals_sorted_by_key']),
- 'BaseHTTPServer': ('http.server',
- ['BaseHTTPRequestHandler',
- 'DEFAULT_ERROR_MESSAGE', 'HTTPServer']),
- 'SimpleHTTPServer': ('http.server', ['SimpleHTTPRequestHandler']),
- 'CGIHTTPServer': ('http.server',
- ['CGIHTTPRequestHandler', 'executable',
- 'nobody_uid', 'nobody']),
- # 'test.test_support': ('test.support',
- # ["Error", "TestFailed", "TestSkipped", "ResourceDenied",
- # "import_module", "verbose", "use_resources",
- # "max_memuse", "record_original_stdout",
- # "get_original_stdout", "unload", "unlink", "rmtree",
- # "forget", "is_resource_enabled", "requires",
- # "find_unused_port", "bind_port",
- # "fcmp", "is_jython", "TESTFN", "HOST",
- # "FUZZ", "findfile", "verify", "vereq", "sortdict",
- # "check_syntax_error", "open_urlresource", "WarningMessage",
- # "catch_warning", "CleanImport", "EnvironmentVarGuard",
- # "TransientResource", "captured_output", "captured_stdout",
- # "TransientResource", "transient_internet", "run_with_locale",
- # "set_memlimit", "bigmemtest", "bigaddrspacetest",
- # "BasicTestRunner", "run_unittest", "run_doctest",
- # "threading_setup", "threading_cleanup", "reap_children"]),
- 'commands': ('subprocess', ['getstatusoutput', 'getoutput']),
- 'UserString' : ('collections', ['UserString']),
- 'UserList' : ('collections', ['UserList']),
- 'urlparse' : ('urllib.parse',
- ['urlparse', 'urlunparse', 'urlsplit',
- 'urlunsplit', 'urljoin', 'urldefrag']),
- 'robotparser' : ('urllib.robotparser', ['RobotFileParser']),
+MAPPING = {'StringIO': 'io',
+ 'cStringIO': 'io',
+ 'cPickle': 'pickle',
+ '__builtin__' : 'builtins',
+ 'copy_reg': 'copyreg',
+ 'Queue': 'queue',
+ 'SocketServer': 'socketserver',
+ 'ConfigParser': 'configparser',
+ 'repr': 'reprlib',
+ 'FileDialog': 'tkinter.filedialog',
+ 'tkFileDialog': 'tkinter.filedialog',
+ 'SimpleDialog': 'tkinter.simpledialog',
+ 'tkSimpleDialog': 'tkinter.simpledialog',
+ 'tkColorChooser': 'tkinter.colorchooser',
+ 'tkCommonDialog': 'tkinter.commondialog',
+ 'Dialog': 'tkinter.dialog',
+ 'Tkdnd': 'tkinter.dnd',
+ 'tkFont': 'tkinter.font',
+ 'tkMessageBox': 'tkinter.messagebox',
+ 'ScrolledText': 'tkinter.scrolledtext',
+ 'turtle': 'tkinter.turtle',
+ 'Tkconstants': 'tkinter.constants',
+ 'Tix': 'tkinter.tix',
+ 'Tkinter': 'tkinter',
+ 'markupbase': '_markupbase',
+ '_winreg': 'winreg',
+ 'thread': '_thread',
+ 'dummy_thread': '_dummy_thread',
+ # anydbm and whichdb are handled by fix_imports2
+ 'dbhash': 'dbm.bsd',
+ 'dumbdbm': 'dbm.dumb',
+ 'dbm': 'dbm.ndbm',
+ 'gdbm': 'dbm.gnu',
+ 'xmlrpclib': 'xmlrpc.client',
+ 'DocXMLRPCServer': 'xmlrpc.server',
+ 'SimpleXMLRPCServer': 'xmlrpc.server',
+ 'httplib': 'http.client',
+ 'Cookie': 'http.cookies',
+ 'cookielib': 'http.cookiejar',
+ 'BaseHTTPServer': 'http.server',
+ 'SimpleHTTPServer': 'http.server',
+ 'CGIHTTPServer': 'http.server',
+ #'test.test_support': 'test.support',
+ 'commands': 'subprocess',
+ 'UserString' : 'collections',
+ 'UserList' : 'collections',
+ 'urlparse' : 'urllib.parse',
+ 'robotparser' : 'urllib.robotparser',
}
@@ -278,31 +62,26 @@ def alternates(members):
def build_pattern(mapping=MAPPING):
bare = set()
- for old_module, (new_module, members) in list(mapping.items()):
+ for old_module, new_module in mapping.items():
bare.add(old_module)
- bare.update(members)
- members = alternates(members)
yield """import_name< 'import' (module=%r
| dotted_as_names< any* module=%r any* >) >
""" % (old_module, old_module)
yield """import_from< 'from' module_name=%r 'import'
- ( %s | import_as_name< %s 'as' any > |
- import_as_names< any* >) >
- """ % (old_module, members, members)
- yield """import_from< 'from' module_name=%r 'import' star='*' >
+ ( any | import_as_name< any 'as' any > |
+ import_as_names< any* >) >
""" % old_module
yield """import_name< 'import'
dotted_as_name< module_name=%r 'as' any > >
""" % old_module
# Find usages of module members in code e.g. urllib.foo(bar)
- yield """power< module_name=%r trailer< '.' %s > any* >
- """ % (old_module, members)
+ yield """power< module_name=%r
+ trailer<'.' any > any* >
+ """ % old_module
yield """bare_name=%s""" % alternates(bare)
-
class FixImports(fixer_base.BaseFix):
PATTERN = "|".join(build_pattern())
-
order = "pre" # Pre-order tree traversal
mapping = MAPPING
@@ -325,19 +104,15 @@ class FixImports(fixer_base.BaseFix):
import_mod = results.get("module")
mod_name = results.get("module_name")
bare_name = results.get("bare_name")
- star = results.get("star")
if import_mod or mod_name:
- new_name, members = self.mapping[(import_mod or mod_name).value]
+ new_name = self.mapping[(import_mod or mod_name).value]
if import_mod:
self.replace[import_mod.value] = new_name
import_mod.replace(Name(new_name, prefix=import_mod.get_prefix()))
elif mod_name:
- if star:
- self.cannot_convert(node, "Cannot handle star imports.")
- else:
- mod_name.replace(Name(new_name, prefix=mod_name.get_prefix()))
+ mod_name.replace(Name(new_name, prefix=mod_name.get_prefix()))
elif bare_name:
bare_name = bare_name[0]
new_name = self.replace.get(bare_name.value)
diff --git a/Lib/lib2to3/fixes/fix_imports2.py b/Lib/lib2to3/fixes/fix_imports2.py
index 6c1f3a1..e82f7e4 100644
--- a/Lib/lib2to3/fixes/fix_imports2.py
+++ b/Lib/lib2to3/fixes/fix_imports2.py
@@ -4,8 +4,8 @@ from . import fix_imports
MAPPING = {
- 'whichdb': ('dbm', ['whichdb']),
- 'anydbm': ('dbm', ['error', 'open']),
+ 'whichdb': 'dbm',
+ 'anydbm': 'dbm',
}
diff --git a/Lib/lib2to3/pytree.py b/Lib/lib2to3/pytree.py
index 65fbb67..efdeb05 100644
--- a/Lib/lib2to3/pytree.py
+++ b/Lib/lib2to3/pytree.py
@@ -652,20 +652,35 @@ class WildcardPattern(BasePattern):
if self.name:
r[self.name] = nodes[:count]
yield count, r
+ elif self.name == "bare_name":
+ yield self._bare_name_matches(nodes)
else:
for count, r in self._recursive_matches(nodes, 0):
if self.name:
r[self.name] = nodes[:count]
yield count, r
+ def _bare_name_matches(self, nodes):
+ """Special optimized matcher for bare_name."""
+ count = 0
+ r = {}
+ done = False
+ max = len(nodes)
+ while not done and count < max:
+ done = True
+ for leaf in self.content:
+ if leaf[0].match(nodes[count], r):
+ count += 1
+ done = False
+ break
+ r[self.name] = nodes[:count]
+ return count, r
+
def _recursive_matches(self, nodes, count):
"""Helper to recursively yield the matches."""
assert self.content is not None
if count >= self.min:
- r = {}
- if self.name:
- r[self.name] = nodes[:0]
- yield 0, r
+ yield 0, {}
if count < self.max:
for alt in self.content:
for c0, r0 in generate_matches(alt, nodes):
diff --git a/Lib/lib2to3/tests/test_fixers.py b/Lib/lib2to3/tests/test_fixers.py
index 117de49..6d8b34e 100755
--- a/Lib/lib2to3/tests/test_fixers.py
+++ b/Lib/lib2to3/tests/test_fixers.py
@@ -1413,7 +1413,7 @@ class Test_imports(FixerTestCase):
from ..fixes.fix_imports import MAPPING as modules
def test_import_module(self):
- for old, (new, members) in list(self.modules.items()):
+ for old, new in self.modules.items():
b = "import %s" % old
a = "import %s" % new
self.check(b, a)
@@ -1423,24 +1423,17 @@ class Test_imports(FixerTestCase):
self.check(b, a)
def test_import_from(self):
- for old, (new, members) in list(self.modules.items()):
- for member in members:
- b = "from %s import %s" % (old, member)
- a = "from %s import %s" % (new, member)
- self.check(b, a)
-
- s = "from foo import %s" % member
- self.unchanged(s)
-
- b = "from %s import %s" % (old, ", ".join(members))
- a = "from %s import %s" % (new, ", ".join(members))
+ for old, new in self.modules.items():
+ b = "from %s import foo" % old
+ a = "from %s import foo" % new
self.check(b, a)
- s = "from foo import %s" % ", ".join(members)
- self.unchanged(s)
+ b = "from %s import foo, bar" % old
+ a = "from %s import foo, bar" % new
+ self.check(b, a)
def test_import_module_as(self):
- for old, (new, members) in list(self.modules.items()):
+ for old, new in self.modules.items():
b = "import %s as foo_bar" % old
a = "import %s as foo_bar" % new
self.check(b, a)
@@ -1450,50 +1443,27 @@ class Test_imports(FixerTestCase):
self.check(b, a)
def test_import_from_as(self):
- for old, (new, members) in list(self.modules.items()):
- for member in members:
- b = "from %s import %s as foo_bar" % (old, member)
- a = "from %s import %s as foo_bar" % (new, member)
- self.check(b, a)
+ for old, new in self.modules.items():
+ b = "from %s import foo as bar" % old
+ a = "from %s import foo as bar" % new
+ self.check(b, a)
def test_star(self):
- for old in self.modules:
- s = "from %s import *" % old
- self.warns_unchanged(s, "Cannot handle star imports")
+ for old, new in self.modules.items():
+ b = "from %s import *" % old
+ a = "from %s import *" % new
+ self.check(b, a)
def test_import_module_usage(self):
- for old, (new, members) in list(self.modules.items()):
- for member in members:
- b = """
- import %s
- foo(%s, %s.%s)
- """ % (old, old, old, member)
- a = """
- import %s
- foo(%s, %s.%s)
- """ % (new, new, new, member)
- self.check(b, a)
-
- def test_from_import_usage(self):
- for old, (new, members) in list(self.modules.items()):
- for member in members:
- b = """
- from %s import %s
- foo(%s, %s())
- """ % (old, member, member, member)
- a = """
- from %s import %s
- foo(%s, %s())
- """ % (new, member, member, member)
- self.check(b, a)
+ for old, new in self.modules.items():
b = """
- from %s import %s
- foo(%s)
- """ % (old, ", ".join(members), ", ".join(members))
+ import %s
+ foo(%s.bar)
+ """ % (old, old)
a = """
- from %s import %s
- foo(%s)
- """ % (new, ", ".join(members), ", ".join(members))
+ import %s
+ foo(%s.bar)
+ """ % (new, new)
self.check(b, a)
generated by cgit v0.12 at 2024-11-15 16:57:08 (GMT)