FEATURE: switch map industry name

alphamind/data/engines/utilities.py

@@ -5,6 +5,7 @@ Created on 2017-12-25
 @author: cheng.li
 """
 
+import os
 from typing import Dict
 from typing import Iterable
 
@@ -51,19 +52,32 @@ def _map_factors(factors: Iterable[str], used_factor_tables) -> Dict:
     return factor_cols
 
 
-def _map_industry_category(category: str) -> str:
-    if category == 'sw':
-        return '申万行业分类'
-    elif category == 'sw_adj':
-        return '申万行业分类修订'
-    elif category == 'zz':
-        return '中证行业分类'
-    elif category == 'dx':
-        return '东兴行业分类'
-    elif category == 'zjh':
-        return '证监会行业V2012'
-    else:
-        raise ValueError("No other industry is supported at the current time")
+if "DB_VENDOR" in os.environ and os.environ["DB_VENDOR"].lower() == "rl":
+    def _map_industry_category(category: str) -> str:
+        if category == 'sw':
+            return '申万行业分类(2014)'
+        elif category == 'zz':
+            return '中证行业分类'
+        elif category == 'zx':
+            return '中信标普行业分类'
+        elif category == 'zjh':
+            return '证监会行业分类(2012)-证监会'
+        else:
+            raise ValueError("No other industry is supported at the current time")
+else:
+    def _map_industry_category(category: str) -> str:
+        if category == 'sw':
+            return '申万行业分类'
+        elif category == 'sw_adj':
+            return '申万行业分类修订'
+        elif category == 'zz':
+            return '中证行业分类'
+        elif category == 'dx':
+            return '东兴行业分类'
+        elif category == 'zjh':
+            return '证监会行业V2012'
+        else:
+            raise ValueError("No other industry is supported at the current time")
 
 
 def industry_list(category: str, level: int = 1) -> list: