argparse - 如何將參數從 args 傳遞到 function？

Question

import argparse
from queries import most_common_cities

parser = argparse.ArgumentParser(description='A script that does operations with database data and returns values')

parser.add_argument('-c', '--most_common_cities',
                nargs=1,
                type=positive_int,
                help='Specify how many common cities.')

args = parser.parse_args()

if args.most_common_cities:
result = most_common_cities(n) # "n" should be an arg passed by user
print(result)

如何將 arguments 從 CLI 傳遞到我的 function arg？ 當有人使用命令時：

python argp.py --most_common_cities 5

它應該返回 5 個最常見的城市。

Answer 1

刪除nargs=1 ，則args.most_common_cities將是傳入的實際值。

nargs=1將其包裝在一個列表中。

parser.add_argument('-c', '--most_common_cities',
                type=int,
                help='Specify how many common cities.')

args = parser.parse_args(['-c', '5'])
n = args.most_common_cities
print(n)
print(type(n))
# 5
# <class 'int'>

Answer 2

我使用以下命令啟動了您的腳本：

python3 test.py --most_common_cities 5

您可以通過以下方式訪問 arguments：

import argparse

parser = argparse.ArgumentParser(description='A script that does operations with database data and returns values')

parser.add_argument('-c', '--most_common_cities',
                nargs=1,
                type=int,
                help='Specify how many common cities.')

args = parser.parse_args()

arguments = vars(parser.parse_args())
print(arguments) #{'most_common_cities': [5]}

#then you can access the value with:
arguments['most_common_cities']