返回指数名称

app/service/fund_service.py

@@ -176,9 +176,10 @@ def fund_index_compare(fund_id, invest_type=2, index_id='000905.SH'):
 
     xlabels = ["" for i in range(len(compare))]
 
+    index_name = get_index_name(index_id).values[0][0]
     chart_data = {
         "xlabels": xlabels,
-        "index": {'name': '中证500', 'data': compare["index_return_ratio"].values},
+        "index": {'name': index_name, 'data': compare["index_return_ratio"].values},
         "fund": {'name': fund_id, 'data': compare[fund_id + "_return_ratio"].values},
     }
     r = draw_index_combination_chart(chart_data)

app/service/portfolio_diagnose.py

@@ -344,6 +344,16 @@ def get_fund_name(fund):
         return df
 
 
+def get_index_name(index_id):
+    with TAMP_SQL(tamp_fund_engine) as tamp_fund:
+        tamp_fund_session = tamp_fund.session
+        sql = "SELECT `name` FROM index_basic WHERE `ts_code`='{}'".format(index_id)
+        # df = pd.read_sql(sql, con)
+        cur = tamp_fund_session.execute(sql)
+        data = cur.fetchall()
+        df = pd.DataFrame(list(data), columns=['index_name'])
+        return df
+
 # 获取排名信息
 fund_rank = get_fund_rank()
 # 获取探普产品池