python 以字符串形式import

今天把自己写了上万行的代码进行代码重构～～佩服我自己的勇气。。大量重复代码(重复率90%)，真心看着恶心....

重构重构。。

 

遇到一个情况就是，就是如何通过字符串import 某个模块。

例如：有2个模块 A 与 B ， 使用时希望是 call('A') 或者 call('B')

 

第一次考虑通过使用函数的方法：

import A
import B

def get_module(module_name):
    if module_name == 'A':
        return A
    elif module_name == 'B':
        return B
    else:
        raise Exception("Module is not existed")



get_module('A')
get_module('B')

 

后来看了很多成熟代码后，发现一个神器的模块 importlib, 所以代码改为

from importlib import import_module

import_module('A')
import_module('B')

# 一行搞定，python自带模块没事还是需要涨涨见识～～～

 

开心～～有学习一个新的模块～～～