support mod lib property

ricequant · Mar 27, 2017 · d78be21 · d78be21
1 parent 45aff16
commit d78be21
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/rqalpha/mod/__init__.py b/rqalpha/mod/__init__.py
@@ -41,7 +41,9 @@ def set_env(self, environment):
             self._mod_list.append((mod_name, mod_config))
 
         for idx, (mod_name, user_mod_config) in enumerate(self._mod_list):
-            if mod_name in SYSTEM_MOD_LIST:
+            if hasattr(user_mod_config, 'lib'):
+                lib_name = user_mod_config.lib
+            elif mod_name in SYSTEM_MOD_LIST:
                 lib_name = "rqalpha.mod.rqalpha_mod_" + mod_name
             else:
                 lib_name = "rqalpha_mod_" + mod_name