Python -類型提示 Type Hints
- 2021 年 8 月 16 日
- 筆記
- Python, 測試進階知識系列 - Python
為什麼會有類型提示
- Python是一種動態類型語言,這意味著我們在編寫程式碼的時候更為自由,運行時不需要指定變數類型
- 但是與此同時 IDE 無法像靜態類型語言那樣分析程式碼,及時給我們相應的提示,比如字元串的 split 方法
def split_str(s): strs = s.split(",")
由於不知道參數 s 是什麼類型,所以當你敲 s. 的時候不會出現 split 的語法提示
解決上述問題,類型提示
Python 3.6 新增了兩個特性 PEP 484 和 PEP 526
幫助 IDE 為我們提供更智慧的提示
這些新特性不會影響語言本身,只是增加一點提示
類型提示分類
主要分兩個
- 變數提示:PEP 526 特性加的
- 函數參數提示:PEP 484 特性加的
變數類型提示
沒有使用類型提示
想說明變數的數據類型只能通過注釋
# 'primes' is a list of integers primes = [] # type: List[int] # 'captain' is a string (Note: initial value is a problem) captain = ... # type: str class Starship: # 'stats' is a class variable stats = {} # type: Dict[str, int]
使用了類型提示
from typing import List, ClassVar, Dict # int 變數,默認值為 0 num: int = 0 # bool 變數,默認值為 True bool_var: bool = True # 字典變數,默認為空 dict_var: Dict = {} # 列表變數,且列表元素為 int primes: List[int] = [] class Starship: # 類變數,字典類型,鍵-字元串,值-整型 stats: ClassVar[Dict[str, int]] = {} # 實例變數,標註了是一個整型 num: int
這裡會用到 typing 模組,後面會再展開詳解
假設變數標註了類型,傳錯了會報錯嗎?
from typing import List # int 變數,默認值為 0 num: int = 0 # bool 變數,默認值為 True bool_var: bool = True # 字典變數,默認為空 dict_var: Dict = {} # 列表變數,且列表元素為 int primes: List[int] = [] num = "123" bool_var = 123 dict_var = [] primes = ["1", "2"] print(num, bool_var, dict_var, primes) # 輸出結果 123 123 [] ['1', '2']
它並不會報錯,但是會有 warning,是 IDE 的智慧語法提示
所以,這個類型提示更像是一個規範約束,並不是一個語法限制
變數類型提示-元組打包
# 正常的元組打包 a = 1, 2, 3 # 加上類型提示的元組打包 t: Tuple[int, ...] = (1, 2, 3) print(t) t = 1, 2, 3 print(t) # py3.8+ 才有的寫法 t: Tuple[int, ...] = 1, 2, 3 print(t) t = 1, 2, 3 print(t) # 輸出結果 (1, 2, 3) (1, 2, 3) (1, 2, 3) (1, 2, 3)
為什麼要加 …
不加的話,元組打包的時候,會有一個 warning 提示
變數類型提示-元組解包
# 正常元組解包 message = (1, 2, 3) a, b, c = message print(a, b, c) # 輸出 1 2 3 # 加上類型提示的元組解包 header: str kind: int body: Optional[List[str]] # 不會 warning 的栗子 header, kind, body = ("str", 123, ["1", "2", "3"]) # 會提示 warning 的栗子 header, kind, body = (123, 123, ["1", "2", "3"])
Optional 會在後面講 typing 的時候詳解
在類裡面使用
class BasicStarship: captain: str = 'Picard' # 實例變數,有默認值 damage: int # 實例變數,沒有默認值 stats: ClassVar[Dict[str, int]] = {} # 類變數,有默認值
ClassVar
- 是 typing 模組的一個特殊類
- 它向靜態類型檢查器指示不應在類實例上設置此變數
函數參數類型提示
不僅提供了函數參數列表的類型提示,也提供了函數返回的類型提示
栗子一
# 參數 name 類型提示 str,而函數返回值類型提示也是 str def greeting(name: str) -> str: return 'Hello ' + name
栗子二
def greeting(name: str, obj: Dict[str, List[int]]) -> None: print(name, obj)