FM / fixes

/opt/alt/python37/lib/python3.7/site-packages/libpasteurize/fixes/

/opt/alt/python37/lib/python3.7/site-packages/libpasteurize/fixes

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
feature_base.py	1723	0644	edit dl rm
fix_add_all_future_builtins.py	1269	0644	edit dl rm
fix_add_all__future__imports.py	676	0644	edit dl rm
fix_add_future_standard_library_import.py	663	0644	edit dl rm
fix_annotations.py	1581	0644	edit dl rm
fix_division.py	904	0644	edit dl rm
fix_features.py	2675	0644	edit dl rm
fix_fullargspec.py	438	0644	edit dl rm
fix_future_builtins.py	1450	0644	edit dl rm
fix_getcwd.py	873	0644	edit dl rm
fix_imports.py	4944	0644	edit dl rm
fix_imports2.py	8580	0644	edit dl rm
fix_kwargs.py	5991	0644	edit dl rm
fix_memoryview.py	551	0644	edit dl rm
fix_metaclass.py	3260	0644	edit dl rm
fix_newstyle.py	888	0644	edit dl rm
fix_next.py	1233	0644	edit dl rm
fix_printfunction.py	401	0644	edit dl rm
fix_raise.py	1099	0644	edit dl rm
fix_raise_.py	1225	0644	edit dl rm
fix_throw.py	835	0644	edit dl rm
fix_unpacking.py	5946	0644	edit dl rm
__init__.py	3719	0644	edit dl rm

Edit: /opt/alt/python37/lib/python3.7/site-packages/libpasteurize/fixes/fix_annotations.py (1581B)

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))

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))