Index: third_party/google-endpoints/libpasteurize/fixes/fix_imports.py |
diff --git a/third_party/google-endpoints/libpasteurize/fixes/fix_imports.py b/third_party/google-endpoints/libpasteurize/fixes/fix_imports.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..4db0d548f52e19ff11093d66ff8f9430090a8965 |
--- /dev/null |
+++ b/third_party/google-endpoints/libpasteurize/fixes/fix_imports.py |
@@ -0,0 +1,113 @@ |
+u""" |
+Fixer for standard library imports renamed in Python 3 |
+""" |
+ |
+from lib2to3 import fixer_base |
+from lib2to3.fixer_util import Name, is_probably_builtin, Newline, does_tree_import |
+from lib2to3.pygram import python_symbols as syms |
+from lib2to3.pgen2 import token |
+from lib2to3.pytree import Node, Leaf |
+ |
+from libfuturize.fixer_util import touch_import_top |
+# from ..fixer_util import NameImport |
+ |
+# used in simple_mapping_to_pattern() |
+MAPPING = {u"reprlib": u"repr", |
+ u"winreg": u"_winreg", |
+ u"configparser": u"ConfigParser", |
+ u"copyreg": u"copy_reg", |
+ u"queue": u"Queue", |
+ u"socketserver": u"SocketServer", |
+ u"_markupbase": u"markupbase", |
+ u"test.support": u"test.test_support", |
+ u"dbm.bsd": u"dbhash", |
+ u"dbm.ndbm": u"dbm", |
+ u"dbm.dumb": u"dumbdbm", |
+ u"dbm.gnu": u"gdbm", |
+ u"html.parser": u"HTMLParser", |
+ u"html.entities": u"htmlentitydefs", |
+ u"http.client": u"httplib", |
+ u"http.cookies": u"Cookie", |
+ u"http.cookiejar": u"cookielib", |
+# "tkinter": "Tkinter", |
+ u"tkinter.dialog": u"Dialog", |
+ u"tkinter._fix": u"FixTk", |
+ u"tkinter.scrolledtext": u"ScrolledText", |
+ u"tkinter.tix": u"Tix", |
+ u"tkinter.constants": u"Tkconstants", |
+ u"tkinter.dnd": u"Tkdnd", |
+ u"tkinter.__init__": u"Tkinter", |
+ u"tkinter.colorchooser": u"tkColorChooser", |
+ u"tkinter.commondialog": u"tkCommonDialog", |
+ u"tkinter.font": u"tkFont", |
+ u"tkinter.ttk": u"ttk", |
+ u"tkinter.messagebox": u"tkMessageBox", |
+ u"tkinter.turtle": u"turtle", |
+ u"urllib.robotparser": u"robotparser", |
+ u"xmlrpc.client": u"xmlrpclib", |
+ u"builtins": u"__builtin__", |
+} |
+ |
+# generic strings to help build patterns |
+# these variables mean (with http.client.HTTPConnection as an example): |
+# name = http |
+# attr = client |
+# used = HTTPConnection |
+# fmt_name is a formatted subpattern (simple_name_match or dotted_name_match) |
+ |
+# helps match 'queue', as in 'from queue import ...' |
+simple_name_match = u"name='%s'" |
+# helps match 'client', to be used if client has been imported from http |
+subname_match = u"attr='%s'" |
+# helps match 'http.client', as in 'import urllib.request' |
+dotted_name_match = u"dotted_name=dotted_name< %s '.' %s >" |
+# helps match 'queue', as in 'queue.Queue(...)' |
+power_onename_match = u"%s" |
+# helps match 'http.client', as in 'http.client.HTTPConnection(...)' |
+power_twoname_match = u"power< %s trailer< '.' %s > any* >" |
+# helps match 'client.HTTPConnection', if 'client' has been imported from http |
+power_subname_match = u"power< %s any* >" |
+# helps match 'from http.client import HTTPConnection' |
+from_import_match = u"from_import=import_from< 'from' %s 'import' imported=any >" |
+# helps match 'from http import client' |
+from_import_submod_match = u"from_import_submod=import_from< 'from' %s 'import' (%s | import_as_name< %s 'as' renamed=any > | import_as_names< any* (%s | import_as_name< %s 'as' renamed=any >) any* > ) >" |
+# helps match 'import urllib.request' |
+name_import_match = u"name_import=import_name< 'import' %s > | name_import=import_name< 'import' dotted_as_name< %s 'as' renamed=any > >" |
+# helps match 'import http.client, winreg' |
+multiple_name_import_match = u"name_import=import_name< 'import' dotted_as_names< names=any* > >" |
+ |
+def all_patterns(name): |
+ u""" |
+ Accepts a string and returns a pattern of possible patterns involving that name |
+ Called by simple_mapping_to_pattern for each name in the mapping it receives. |
+ """ |
+ |
+ # i_ denotes an import-like node |
+ # u_ denotes a node that appears to be a usage of the name |
+ if u'.' in name: |
+ name, attr = name.split(u'.', 1) |
+ simple_name = simple_name_match % (name) |
+ simple_attr = subname_match % (attr) |
+ dotted_name = dotted_name_match % (simple_name, simple_attr) |
+ i_from = from_import_match % (dotted_name) |
+ i_from_submod = from_import_submod_match % (simple_name, simple_attr, simple_attr, simple_attr, simple_attr) |
+ i_name = name_import_match % (dotted_name, dotted_name) |
+ u_name = power_twoname_match % (simple_name, simple_attr) |
+ u_subname = power_subname_match % (simple_attr) |
+ return u' | \n'.join((i_name, i_from, i_from_submod, u_name, u_subname)) |
+ else: |
+ simple_name = simple_name_match % (name) |
+ i_name = name_import_match % (simple_name, simple_name) |
+ i_from = from_import_match % (simple_name) |
+ u_name = power_onename_match % (simple_name) |
+ return u' | \n'.join((i_name, i_from, u_name)) |
+ |
+ |
+class FixImports(fixer_base.BaseFix): |
+ |
+ PATTERN = u' | \n'.join([all_patterns(name) for name in MAPPING]) |
+ PATTERN = u' | \n'.join((PATTERN, multiple_name_import_match)) |
+ |
+ def transform(self, node, results): |
+ touch_import_top(u'future', u'standard_library', node) |
+ |