如何将程序的启动方式修改为通过cmd命令行带参数启动
源码是一个自动写诗的代码。启动部分源码如下：
def denerate_poem(func_select,start_words):
results = func_select(start_words, ix2word, word2ix)
rslt = ''.join(results)
print(rslt)
input_words = input('请输入一个4字短语作为藏头诗每句诗的第一个字：')
denerate_poem(gen_acrostic,start_words=input_words)
这里的 func_select 有2个可选项，一个是：generate，另一个是gen_acrostic。
目前运行的时候是直接以gen_acrostic的模式，和输入的短语进行输出。
想要修改为在cmd命令行中启动，样式如：
python main.py --gen_acrostic
然后在输入短语进行输出。
求大佬们帮忙看看怎么改动啊
import sys
str = sys.argv.split('--')[-1]
print(str)
# 下面是cmd界面
C:\Users\Niu\Desktop>py python.py --generate
generate
页:
[1]