python - 将参数传递给 Python 脚本

Question

我想通过两种方式将参数传递给 Python 脚本，

python main.py --source=aws

和

python main.py source aws

这是我当前的代码，

parser = argparse.ArgumentParser()
parser.add_argument("--s", "--source", help='Flag to choose source')

这使得第一个选项成为可能。如何使第二个选项成为可能？

Answer 1

Argparse 无法做到这一点。唯一的方法是使用 sys.argv 过滤标准输入。

import argparse
import sys

mangle_my_args = ['s', 'source']
arguments=['--'+arg if arg in mangle_my_args else arg for arg in sys.argv[1:]]
parser = argparse.ArgumentParser()
parser.add_argument("--s", "--source", help='Flag to choose source')
print(parser.parse_args(arguments))