Python：如果 __name__ == '__main__' 后导入并初始化 Argparse？

Question

如果我在脚本中使用 argparse 和 if __name__ == '__main__' 测试，我也想用作模块，我应该在该测试下导入 argparse 然后初始化它吗？我发现的样式指南都没有提到在脚本中使用 argparse，并且许多 argparse 脚本示例不使用“if name”测试或以不同方式使用它。到目前为止，这是我一直在做的事情：

#! /usr/bin/env python

def main(name):
    print('Hello, %s!' % name)

if __name__ == '__main__':
    import argparse
    parser = argparse.ArgumentParser(description = 'Say hello')
    parser.add_argument('name', help='your name, enter it')
    args = parser.parse_args()

    main(args.name)

我应该在顶部导入 argparse 和我的其他模块并在脚本正文中配置它吗？

Answer 1

我会将导入放在顶部，但将使用它的代码留在if __name__ 块中：

import argparse

# other code. . .

def main(name):
    print('Hello, %s!' % name)

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description = 'Say hello')
    parser.add_argument('name', help='your name, enter it')
    args = parser.parse_args()

    main(args.name)

将导入放在顶部可以阐明您的模块使用哪些模块。即使您不使用 argpase 也导入它对性能的影响可以忽略不计。

Answer 2

如果argparse 仅在该块中引用，则可以将import argparse 放在if __name__ == '__main__' 块中。显然，如果您的模块由另一个模块导入，则该块中的代码将不会运行，因此该模块必须为main 提供它自己的参数（可能使用它自己的ArgumentParser 实例）。