Commit fa4f0b67 authored by Dr.李's avatar Dr.李

change utilitiy

parent 5774831d
This diff is collapsed.
...@@ -24,27 +24,18 @@ def nearest_report_date(date): ...@@ -24,27 +24,18 @@ def nearest_report_date(date):
return dt.date(last_year, q, last).strftime("%Y%m%d") return dt.date(last_year, q, last).strftime("%Y%m%d")
def fd_alive_funds(trade_dt, l1_code=None): def fd_alive_funds(trade_dt, type_style, class_code):
if l1_code is not None: sec_codes = read_sql(f"""
security_ids = read_sql(f""" select SECODE from TQ_FD_TYPE
select SECURITYID from TQ_FD_TYPECLASS WHERE
WHERE ISVALID = 1 AND
ISVALID = 1 AND TYPESTYLE = '{type_style}' AND
L1CODE = {l1_code} AND CLASSCODE = '{class_code}' AND
BEGINDATE <= '{trade_dt}' AND BEGINDATE <= '{trade_dt}' AND
(ENDDATE>='{trade_dt}' or ENDDATE = '19000101') (ENDDATE>='{trade_dt}' or ENDDATE = '19000101')
ORDER BY SECURITYID; ORDER BY SECODE;
""")["SECURITYID"].unique().tolist() """)["SECODE"].unique().tolist()
else: return sec_codes
security_ids = read_sql(f"""
select SECURITYID from TQ_FD_TYPECLASS
WHERE
ISVALID = 1 AND
BEGINDATE <= '{trade_dt}' AND
(ENDDATE>='{trade_dt}' or ENDDATE = '19000101')
ORDER BY SECURITYID;
""")["SECURITYID"].unique().tolist()
return security_ids
def fd_basicinfo(security_ids=None, trade_dt=None): def fd_basicinfo(security_ids=None, trade_dt=None):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment