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

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

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

首先使用pip來安裝它：

python3 -m pip install typer

函數參數等于命令行參數

我們首先來看看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會報錯：

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

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

子命令與自動補全更好用

假設我們有一個神經網絡的程序，其中的入口函數代碼如下  ：

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

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

。')

顯然，這個程序可以用來訓練數據，也可以用來預測數據 
，所以有兩種不同的命令