FastAPI的小兄弟
 ，開發(fā)命令行工具更給力

FastAPI的開發(fā)組織叫做tiangolo ，他家除了FastAPI外
，還有另一個項目也非常好用，叫做typer。

三年前 ，我寫過一篇文章一日一技 ：快速實現(xiàn)Python 命令行參數(shù)介紹另一個命令行工具fire. 而typer做得比fire還要好。

首先使用pip來安裝它
：

python3 -m pip install typer

函數(shù)參數(shù)等于命令行參數(shù)

我們首先來看看typer怎么使用
。創(chuàng)建一個example_1.py文件 ，寫入如下代碼 。

import typer


def main(name: str, salary: int):
print(f'{ name}月薪{ salary}元')


if __name__ == '__main__':
typer.run(main)

直接運行，Python會報錯：

使用參數(shù)--help可以查看這個腳本的命令行參數(shù)：

于是我們根據(jù)這里的提示 ，輸入正確的參數(shù)，從而正常運行程序 ：

子命令與自動補全更好用

假設(shè)我們有一個神經(jīng)網(wǎng)絡(luò)的程序 ，其中的入口函數(shù)代碼如下：

def train_data(train_folder: str, test_folder: str, rate: float = 0.8):
"""
訓(xùn)練人臉檢測模型
"""
print(f'使用文件夾{ train_folder}中的數(shù)據(jù)進行訓(xùn)練')
print(f'使用{ test_folder}中的數(shù)據(jù)用來驗證訓(xùn)練效果，確保準(zhǔn)確率>{ rate}')
return True

def predict(folder: str):
"""
使用訓(xùn)練好的模型預(yù)測
"""
print(f'對文件夾{ folder}中的數(shù)據(jù)進行預(yù)測
。')

顯然，這個程序可以用來訓(xùn)練數(shù)據(jù)
，也可以用來預(yù)測數(shù)據(jù)，所以有兩種不同的命令 ，每一種命令有不同的參數(shù)
。

這種情況下，使用typer非常方便
，只需要加兩個裝飾器就可以了：

運行效果如下圖所示
：

輸入具體的子命令，還可以查看每個子命令的參數(shù) ：

因此