要在python中输入参数,可以使用getopt module。这里的参数可以是可选的,只要有正确的标志就可以按任何顺序输入。在
在下面的示例中,用户有两个可选参数要设置,即输入文件名和数据库名。可以使用调用代码python example.py -f test.txt -d HelloWorld
或者python example.py file=test.txt database=HelloWorld
或者两者兼而有之。
可以更改标志和名称以反映您的需要。在import getopt
def main(argv):
inputFileName = ”
databaseName = ”
try:
opts, args = getopt.getopt(argv,”f:d:”,[“file=”,”database=”])
except getopt.GetoptError:
print(‘-f -d -c ‘)
sys.exit()
for opt, arg in opts:
if opt in (‘-f’,’ file’):
inputFileName = arg
elif opt in (‘-d’,’ database’):
databaseName = arg
if __name__ == “__main__”:
main(sys.argv[1:])