python脚本使用cmd执行,可通过获取用户输入的选项来执行不同的函数
时间: 2024-02-28 08:52:59 浏览: 142
可以使用Python的`argparse`模块来实现。这个模块可以帮助你解析命令行参数并且生成帮助信息。以下是一个示例脚本:
```python
import argparse
def function1():
print("Function 1 called")
def function2():
print("Function 2 called")
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Description of your script')
parser.add_argument('-f', '--function', choices=['function1', 'function2'], required=True, help='Which function to call')
args = parser.parse_args()
if args.function == 'function1':
function1()
elif args.function == 'function2':
function2()
```
在命令行中执行这个脚本时,你可以使用`-h`选项来查看帮助信息:
```
$ python my_script.py -h
usage: my_script.py [-h] -f {function1,function2}
Description of your script
optional arguments:
-h, --help show this help message and exit
-f {function1,function2}, --function {function1,function2}
Which function to call
```
你也可以执行这个脚本并指定要调用的函数:
```
$ python my_script.py -f function1
Function 1 called
```
阅读全文