astypes

astypes

python庫以靜態檢測AST節點的類型。

一個好的用例是只需對特定類型運行一些規則的襯裡。例如，僅當something具有類型str時，檢查something.format(a=b) 。

python3 -m pip install astypes

用法

Astypes使用星體來推斷節點的定義。因此，如果您的代碼與AST節點一起使用，則需要首先將它們轉換為Astroid：

 import astroid
import astypes

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

當您有一個星體節點時，您可以得到它的類型：

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

例子：

 import astroid
import astypes

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

對於真實的用法示例，請查看sublytypes。這是一個CLI工具，可以使用ASTYPES自動將類型註釋添加到Python代碼中。

它如何工作

您可以在astypes/_handlers.py中找到大多數邏輯。簡而言之：

1. 某些節點的類型很容易推斷。例如， 13始終是int 。
2. 也可以推斷出某些節點，但前提是我們做出一些假設。我們做出的假設是99％的案件中的假設。例如，我們假設list(x)返回類型list 。如果您與其他內容列出了陰影list ，那可能是不正確的。
3. 如果不能僅通過查看節點來假設類型，我們會嘗試使用Astroid來推斷類型。
4. 如果返回的值是函數調用，我們使用Astroid來找到函數的定義。函數的註釋回報註釋是我們需要的。
5. 如果已解決的函數沒有註釋，我們使用Typeshed_client從Typeshed獲取其註釋。例如，對於所有內置功能。