修改 周边风险清洗BUG

company/routes.py

@@ -52,4 +52,4 @@ def general_rating():
     financial_analysis(rid)
     risk_analysis_etl(rid)
 
-    return {"info": "数据准备完成"}
+    return {"info": "数据准备完成"}, 200

company/scripts.py

@@ -411,6 +411,7 @@ def financial_analysis(rid):
     index_level = json.loads(df_rate.apply(lambda x: '优' if x.values[0] >= 1 else ('良' if x.values[0] >= 0.75 else ('中' if x.values[0] >= 0.5 else ('低' if x.values[0] >= 0.25 else '差')))).to_json())
     df_index = pd.DataFrame({'值': index_value, "级别": index_level})
     result['指标详情'] = df_iterrows(df_index)
+
     insert_data_to_tfse('企业', '财务要素分析', result)
 
 
@@ -512,7 +513,10 @@ def risk_analysis_etl(rid):
     # 关联风险表格
     def associate_risk():
         data = associate_risk_detail(rating_result['企业名称'])
+        if data:
             return df_iterrows(pd.DataFrame(data).sort_values('total', ascending=False).set_index('title'))
+        else:
+            return data
 
     # 变更记录表格
     def change_log():
@@ -545,4 +549,5 @@ def risk_analysis_etl(rid):
     result['经营风险'] = operating_risk()
     result['周边风险'] = associate_risk()
     result['变更记录'] = change_log()
+
     insert_data_to_tfse('企业', '风险要素分析', result)

rating/scripts/risk_detail.py

@@ -15,7 +15,17 @@ def associate_risk_detail(param):
     """
     res = find_tyc_db(db='天眼风险', col='周边风险', query={'企业名称': param})
     result = []
+
     if res:
+        pass
+    else:
+        return result
+
+    if res[0]['周边风险']['list']:
+        pass
+    else:
+        return result
+
     df = pd.DataFrame(res[0]['周边风险']['list'])[['total', 'title', 'list']]
     detail_list = list()
     for row in df.iterrows():
@@ -23,6 +33,7 @@ def associate_risk_detail(param):
         detail_list.append((df_sub1['title'] + df_sub1['riskCount'].apply(lambda x: ' ' + str(x) + '条')).to_list())
     df['detail'] = detail_list
     result = list(json.loads(df[['title', 'total', 'detail']].T.to_json()).values())
+
     return result
 
 
@@ -49,7 +60,3 @@ def change_log_detail(param):
         df = df.drop(df[df['companyId'] == 0].index)
         result = list(json.loads(df[['title', 'total', 'detail']].T.to_json()).values())
     return result
-
-
-if __name__ == '__main__':
-    change_log_detail('四川省通江山霸王野生食品有限公司')