Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(320)

Unified Diff: third_party/google-endpoints/libpasteurize/fixes/fix_annotations.py

Issue 2666783008: Add google-endpoints to third_party/. (Closed)
Patch Set: Created 3 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: third_party/google-endpoints/libpasteurize/fixes/fix_annotations.py
diff --git a/third_party/google-endpoints/libpasteurize/fixes/fix_annotations.py b/third_party/google-endpoints/libpasteurize/fixes/fix_annotations.py
new file mode 100644
index 0000000000000000000000000000000000000000..1926288c2310f8a02bd4047df4db3e0e74a839e2
--- /dev/null
+++ b/third_party/google-endpoints/libpasteurize/fixes/fix_annotations.py
@@ -0,0 +1,48 @@
+u"""
+Fixer to remove function annotations
+"""
+
+from lib2to3 import fixer_base
+from lib2to3.pgen2 import token
+from lib2to3.fixer_util import syms
+
+warning_text = u"Removing function annotations completely."
+
+def param_without_annotations(node):
+ return node.children[0]
+
+class FixAnnotations(fixer_base.BaseFix):
+
+ warned = False
+
+ def warn_once(self, node, reason):
+ if not self.warned:
+ self.warned = True
+ self.warning(node, reason=reason)
+
+ PATTERN = u"""
+ funcdef< 'def' any parameters< '(' [params=any] ')' > ['->' ret=any] ':' any* >
+ """
+
+ def transform(self, node, results):
+ u"""
+ This just strips annotations from the funcdef completely.
+ """
+ params = results.get(u"params")
+ ret = results.get(u"ret")
+ if ret is not None:
+ assert ret.prev_sibling.type == token.RARROW, u"Invalid return annotation"
+ self.warn_once(node, reason=warning_text)
+ ret.prev_sibling.remove()
+ ret.remove()
+ if params is None: return
+ if params.type == syms.typedargslist:
+ # more than one param in a typedargslist
+ for param in params.children:
+ if param.type == syms.tname:
+ self.warn_once(node, reason=warning_text)
+ param.replace(param_without_annotations(param))
+ elif params.type == syms.tname:
+ # one param
+ self.warn_once(node, reason=warning_text)
+ params.replace(param_without_annotations(params))

Powered by Google App Engine
This is Rietveld 408576698