python 以字符串形式import

今天把本身寫了上萬行的代碼進行代碼重構～～佩服我本身的勇氣。。大量重複代碼(重複率90%)，真心看着噁心....

重構重構。。

 

遇到一個狀況就是，就是如何經過字符串import 某個模塊。

例如：有2個模塊 A 與 B ， 使用時但願是 call('A') 或者 call('B')

 

第一次考慮經過使用函數的方法：

 1 import A
 2 import B
 3 
 4 def get_module(module_name):
 5     if module_name == 'A':
 6         return A
 7     elif module_name == 'B':
 8         return B
 9     else:
10         raise Exception("Module is not existed")
11 
12 
13 
14 get_module('A')
15 get_module('B')

 

後來看了不少成熟代碼後，發現一個神器的模塊 importlib, 因此代碼改成

1 from importlib import import_module
2 
3 import_module('A')
4 import_module('B')
5 
6 # 一行搞定，python自帶模塊沒事仍是須要漲漲見識～～～

 

開心～～有學習一個新的模塊～～～