ꦫꦣꦺꦤ꧀ꦄꦤ꧀ꦠꦱꦺꦤ

PATH: opt / cloudlinux / venv / lib / python3.11 / site-packages / libpasteurize / fixes

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

[+] ..
[+] __pycache__
[-] fix_kwargs.py [open]
[-] fix_newstyle.py [open]
[-] fix_imports.py [open]
[-] fix_fullargspec.py [open]
[-] fix_metaclass.py [open]
[-] feature_base.py [open]
[-] fix_printfunction.py [open]
[-] fix_next.py [open]
[-] fix_add_future_standard_library_import.py [open]
[-] fix_throw.py [open]
[-] fix_features.py [open]
[-] fix_raise_.py [open]
[-] fix_unpacking.py [open]
[-] __init__.py [open]
[-] fix_annotations.py [open]
[-] fix_memoryview.py [open]
[-] fix_future_builtins.py [open]
[-] fix_getcwd.py [open]
[-] fix_add_all__future__imports.py [open]
[-] fix_raise.py [open]
[-] fix_add_all_future_builtins.py [open]
[-] fix_imports2.py [open]
[-] fix_division.py [open]