Question

我需要创建一个安装/卸载的python脚本。目前我的脚本只能安装。我称之为：

python ./script.py --arg1 arg2 --arg2 arg2 --arg3 arg3

代码段如下：

def main(argv):
    subparser = parser.add_argument('--arg1', help='Enter arg1', dest="arg1")
    subparser = parser.add_argument('--arg2', help='Enter arg2', dest="arg2")
    subparser = parser.add_argument('--arg3', help='Enter arg3', dest="arg3")
    args = parser.parse_args()

    return args

我想修改上面的脚本，以便我能够按如下方式执行脚本：

python ./script.py --install --arg1 arg1 --arg2 arg2 --arg3 arg3

或

python ./script.py --uninstall

当我给--install作为命令行参数时，我期待arg1，arg2和arg3;但是当--uninstall作为命令行参数给出时，我不希望其他任何东西

我可以做我打算做的事吗？

Answer 1

现在我已经更新了我的python脚本，如下所示：

def main(argv):
    install = sys.argv[1]
    if install == "--uninstall":
            uninstall()
    elif install == "--install" or install == "--help":
            install()

def install():
    parser = argparse.ArgumentParser(description="Installing/Uninstalling")
    subparser = parser.add_argument('--install', dest='accumulate', action='store_const', 
                      const=None, default=None, 
                      help='Please provide either --install / --uninstall')
    subparser = parser.add_argument('--arg1', help='Enter arg1', dest="arg1")
    subparser = parser.add_argument('--arg2', help='Enter arg2', dest="arg2")
    subparser = parser.add_argument('--arg3', help='Enter arg3', dest="arg3")
    args = parser.parse_args()

    return args

如果这是正确的方法，请告诉我？还是有更好的方法？

使用命令行参数创建python脚本

1 个答案: