Usuwanie podpowiedzi typu programowo w języku Python

Question

Mam napisany kod źródłowy dla Pythona 3.5, który chcę wykonać w Pythonie 3.4. Jedyną opcją z 3,5, której używam, która nie jest dostępna w wersji 3.4, są wskazówki typu, więc chciałbym napisać skrypt, aby usunąć je całkowicie.

Na pierwszy rzut oka wydawało się to łatwe i postanowiłem napisać kilka wyrażeń, aby to zrobić, ale potem pomyślałem o kilku skrajnych przypadkach i nie byłem pewien, jak rozwiązać problem w celu uzyskania bardziej złożonej funkcji, takiej jak ta:

def foo(bar: Dict[T, List[T]], 
     baz: Callable[[T], int] = lambda x: (x+3)/7, 
     **kwargs) -> List[T]: 

Zasadniczo musiałbym przeanalizować całą sprawę i odbudować listę argumentów bez adnotacji typu. Jak miałbym się do tego podejść?

Answer 1

OK, rozumiem: D

wbudowanego ast moduł

w Pythonie do analizowania kodu źródłowego, a następnie doskonałe astunparse biblioteka ponownie wygenerować kod źródłowy z przeanalizowanej ast. Wtedy wszystko, co pozostało jest usunięcie adnotacji Typ:

import ast 
import astunparse 

source=""" 
import typing 
from typing import Dict, T, Callable 
from typing import List 

def foo(bar: Dict[T, List[T]], 
     baz: Callable[[T], int] = lambda x: (x+3)/7, 
     **kwargs) -> List[T]: 
    pass 
""" 

class TypeHintRemover(ast.NodeTransformer): 

    def visit_FunctionDef(self, node): 
     # remove the return type defintion 
     node.returns = None 
     # remove all argument annotations 
     if node.args.args: 
      for arg in node.args.args: 
       arg.annotation = None 
     return node 

    def visit_Import(self, node): 
     node.names = [n for n in node.names if n.name != 'typing'] 
     return node if node.names else None 

    def visit_ImportFrom(self, node): 
     return node if node.module != 'typing' else None 

# parse the source code into an AST 
parsed_source = ast.parse(source) 
# remove all type annotations, function return type definitions 
# and import statements from 'typing' 
transformed = TypeHintRemover().visit(parsed_source) 
# convert the AST back to source code 
print(astunparse.unparse(transformed)) 

wizyt TypeHintRemover wszystkich węzłów w AST i usuwa wszelkie podpowiedzi typu w ciągu argumentów funkcji, definicje typu zwrot z każdej funkcji i wszystkie instrukcje importu, które odnoszą się do moduł "pisania".

Wynikiem jest:

def foo(bar, baz=(lambda x: ((x + 3)/7)), **kwargs): 
    pass