如何使用 argparse 模塊在 python 腳本中將文件作為參數傳遞？

Question

我正在使用 argparse 模塊在 python 中編寫一個自動化腳本，我想在其中使用 -s 作為選項，它將文件/文件路徑作為參數。 有人可以幫我做到這一點嗎？

示例：./argtest.py -s /home/test/hello.txt

Answer 1

只需這樣做：

import argparse

parser = argparse.ArgumentParser(description="My program!", formatter_class=argparse.RawTextHelpFormatter)
parser.add_argument("-s", type=argparse.FileType('r'), help="Filename to be passed")
args = vars(parser.parse_args())

open_file = args.s

如果要打開文件進行寫入，只需將type=argparse.FileType('r')中的r更改為w即可。 您也可以將其更改為a 、 r+ 、 w+等。

Answer 2

您可以使用

import argparse

parse = argparse.ArgumentParser()
parse.add_argument("-s")
args = parse.parse_args()
# print argument of -s
print('argument: ',args.s)

假設上面的代碼存儲在文件example.py中

$ python example.py -s /home/test/hello.txt
argument: /home/test/hello.txt

您可以單擊此處(Python3.x) 或此處(Python2.x) 了解更多信息。