using hb classification

fof/utility.py

@@ -497,15 +497,25 @@ def fd_fund_mktv(security_ids, trade_dt):
     return df
 
 
-def fd_rl_classification(security_ids, trade_dt):
+def fd_rl_classification(security_ids, trade_dt, source="hb"):
     sec_id_strs = _join_ids(security_ids)
-    query = f"""
-    SELECT security_id as SECURITYID,  rl_class, rl_class_name, rl_sub_class, rl_sub_class_name FROM fd_rl_classification
-    WHERE
-        trade_date = '{_to_yyyymmdd(trade_dt)}' AND
-        security_id in ({sec_id_strs}) AND
-        flag = 1
-    """
+    if source == "rl":
+        table = "fd_rl_classification"
+        query = f"""
+        SELECT security_id as SECURITYID,  rl_class, rl_class_name, rl_sub_class, rl_sub_class_name FROM {table}
+        WHERE
+            trade_date = '{_to_yyyymmdd(trade_dt)}' AND
+            security_id in ({sec_id_strs}) AND
+            flag = 1
+        """
+    elif source == "hb":
+        table = "hb_fd_classification"
+        query = f"""
+        SELECT security_code as SECURITYID,  rl_class, rl_class_name, rl_sub_class, rl_sub_class_name FROM {table}
+        WHERE
+            security_code in ({sec_id_strs}) AND
+            flag = 1
+        """
     return read_sql(query, source="mysql").sort_values("SECURITYID").reset_index(drop=True)