astypes

Asttypes

Python -Bibliothek, um Typen für AST -Knoten statisch zu erkennen.

Ein guter Anwendungsfall ist ein Linter, der einige Regeln nur für bestimmte Typen ausführen muss. Zum Beispiel, um Argumente something something.format(a=b) zu überprüfen str

python3 -m pip install astypes

Verwendung

ASTypes verwendet Astroid, um Definitionen von Knoten zu schließen. Wenn Ihr Code also mit AST -Knoten funktioniert, müssen Sie sie zuerst in Astroid konvertieren:

 import astroid
import astypes

module = astroid . parse ( source_code )
node = astypes . find_node ( module , ast_node )

Und wenn Sie einen Astroidknoten haben, können Sie seinen Typ erhalten:

 node_type = astype . get_node ( node )
print ( node_type . annotation )

Beispiel:

 import astroid
import astypes

node = astroid . extract_node ( '1 + 2.3' )
t = astypes . get_type ( node )
print ( t . annotation )  # 'float'

Für ein Beispiel für ein reales Verwendungsbeispiel finden Sie infer-Typen. Es handelt sich um ein CLI -Tool, das mithilfe von ASTypes automatisch Typ Anmerkungen in Python -Code hinzufügt.

Wie es funktioniert

Sie finden den größten Teil der Logik in Asttypes/_Handlers.py. Zusamenfassend:

1. Die Art einiger Knoten ist leicht zu schließen. Zum Beispiel ist 13 immer int .
2. Einige Knoten können auch abgeleitet werden, aber nur, wenn wir einige Annahmen treffen. Annahmen, die wir treffen, sind diejenigen, die in 99% der Fälle zutreffen. Zum Beispiel gehen wir davon aus, dass die list(x) list zurückgibt. Es ist möglicherweise nicht der Fall, wenn Sie mit etwas anderem list .
3. Wenn der Typ nicht nur durch Betrachtung des Knotens angenommen werden kann, versuchen wir, Astroid zu verwenden, um den Typ zu schließen.
4. Wenn der zurückgegebene Wert ein Funktionsaufruf ist, verwenden wir Astroid, um die Definition der Funktion zu finden. Die kommentierte Rückkehrannotation der Funktion ist das, was wir brauchen.
5. Wenn die aufgelöste Funktion keine Annotation hat, verwenden wir typeshed_client, um ihre Annotation von typeshed zu erhalten. Zum Beispiel für alle integrierten Funktionen.