astypes

ASTYPES

Biblioteca Python para detectar estaticamente tipos para nós AST.

Um bom caso de uso é um linhador que precisa executar algumas regras apenas para tipos específicos. Por exemplo, para verificar os argumentos de something.format(a=b) somente se something tiver tipo str .

python3 -m pip install astypes

Uso

ASTYPES usa astroid para inferir definições de nós. Portanto, se o seu código funcionar com nós AST, você precisará convertê -los em astroid primeiro:

 import astroid
import astypes

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

E quando você tem um nó astroidal, você pode obter seu tipo:

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

Exemplo:

 import astroid
import astypes

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

Para um exemplo de uso do mundo real, consulte os tipos inferiores. É uma ferramenta de CLI que adiciona automaticamente anotações de tipo no código Python usando astypes.

Como funciona

Você pode encontrar a maior parte da lógica em astypes/_handlers.py. Resumidamente:

1. O tipo de alguns nós é fácil de inferir. Por exemplo, 13 é sempre int .
2. Alguns nós também são inferidos, mas apenas se fizermos algumas suposições. As suposições que fazemos são as que são verdadeiras em 99% dos casos. Por exemplo, assumimos que a list(x) retorna list de tipos. Pode não ser verdade se você sombrear list com outra coisa.
3. Se o tipo não puder ser assumido apenas olhando para o nó, tentamos usar astroid para inferir o tipo.
4. Se o valor retornado for uma chamada de função, usamos astroid para encontrar a definição da função. A anotação anotada de retorno da função é o que precisamos.
5. Se a função resolvida não tiver anotação, usamos o TypeSeshed_client para obter sua anotação do TypeShed. Por exemplo, para todas as funções internas.