为条件语句提供sys.argv

Question

我的功能设置如下：

args = sys.argv[1:]
if args:
    if args[0] in Constants.envs.keys():
        environment = Constants.envs[args[0]]

但我觉得这看起来并不干净。有没有更好的方法来检查参数，如果有效，可以设置变量吗？

Answer 1

为什么不组合两个if语句？

args = sys.argv[1:]
if args and args[0] in Constants.envs:
    environment = Constants.envs[args[0]]

您还可以使用dict的get方法，如果缺少密钥，则返回默认值。这意味着您无需检查args[0] in Constants.envs.keys()。

args = sys.argv[1:]
if args:
    environment = Constants.envs.get(args[0])  # will return None if the key is missing