astypes

astitis

Biblioteca de Python para detectar estáticamente tipos de nodos AST.

Un buen caso de uso es un enlace que necesita ejecutar algunas reglas solo para tipos particulares. Por ejemplo, para verificar los argumentos de something.format(a=b) solo si something tiene tipo str .

python3 -m pip install astypes

Uso

Astipes usa el Atroides para inferir definiciones de nodos. Entonces, si su código funciona con nodos AST, primero deberá convertirlos en Atroides:

 import astroid
import astypes

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

Y cuando tienes un nodo atroides, puedes obtener su tipo:

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

Ejemplo:

 import astroid
import astypes

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

Para un ejemplo de uso del mundo real, consulte los tipos de infierno. Es una herramienta CLI que agrega automáticamente anotaciones de tipo en el código Python usando Astipes.

Cómo funciona

Puede encontrar la mayor parte de la lógica en Astipes/_handlers.py. En breve:

1. El tipo de algunos nodos es fácil de inferir. Por ejemplo, 13 es siempre int .
2. Algunos nodos también se pueden inferir, pero solo si hacemos algunas suposiciones. Las suposiciones que hacemos son las que son ciertas en el 99% de los casos. Por ejemplo, suponemos que esa list(x) devuelve list de tipos. Puede que no sea cierto si list con algo más.
3. Si no se puede asumir el tipo solo mirando el nodo, intentamos usar el astroide para inferir el tipo.
4. Si el valor devuelto es una llamada de función, usamos Atroides para encontrar la definición de la función. La anotación de retorno anotada de la función es lo que necesitamos.
5. Si la función resuelta no tiene anotación, usamos typeshed_client para obtener su anotación de Tipeshed. Por ejemplo, para todas las funciones incorporadas.