[已解决]如何将程序的启动方式修改为通过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
然后在输入短语进行输出。

求大佬们帮忙看看怎么改动啊

最佳答案

月排行榜 / 总排行榜

super_niu

2020-12-22 16:24:59

import sys

str = sys.argv[1].split('--')[-1]

print(str)

# 下面是cmd界面

C:\Users\Niu\Desktop>py python.py --generate
generate

跳转到最佳答案楼层

super_niu

import sys

str = sys.argv[1].split('--')[-1]

print(str)

# 下面是cmd界面

C:\Users\Niu\Desktop>py python.py --generate
generate