Question

我有一个脚本，它解析了一些参数，并且有一些昂贵的导入，但只有当用户提供有效的输入参数时才需要这些导入，否则程序退出。此外，当用户说python script.py --help时，根本不需要执行那些昂贵的导入。

我能想到这样一个剧本：

import argparse

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('--argument', type=str)
    args = parser.parse_args()
    return args

if __name__ == "__main__":
    args = parse_args()

import gensim # expensive import
import blahblahblah

def the_rest_of_the_code(args):
    pass

if __name__ == "__main__":
    the_rest_of_the_code(args)

这可以胜任，但对我来说看起来并不优雅。有没有更好的任务建议？

编辑：导入非常昂贵：

$ time python -c "import gensim"
Using TensorFlow backend.

real    0m12.257s
user    0m10.756s
sys 0m0.348s

Answer 1

您可以有条件地导入，也可以在try块中导入，也可以在代码中的任何位置导入。

所以你可以这样做：

import cheaplib

if __name__ == "__main__":
    args = parse_args()
    if expensive_arg in args:
        import expensivelib
    do_stuff(args)

或者更清楚的是，只导入将使用它的函数中的lib。

def expensive_function():
    import expensivelib
    ...

Answer 2

不确定它比你现有的更好，但你可以懒散加载它：

def load_gensim():
    global gensim
    import gensim

如果您只想确保参数有意义，可以使用包装器main模块检查参数，然后加载另一个模块并调用它。

main.py：

args = check_args()
if args is not None:
    import mymodule
    mymodule.main(args)

mymodule.py：

import gensim
def main(args):
    # do work

如何在昂贵的导入之前优雅地解析python中的argumens？

2 个答案: