解决注册后未认证企业信息异常

Implements/Company/IndexImpl.py

@@ -25,37 +25,22 @@ class CompanyIndexImpl(CompanyIndex):
             ["企业名称", "综信评价结果", "ESG评价结果"]
         )
 
-        head_info = CompanyIndex.HeadInfo()
-        head_info.name = None if not company_data else company_data['企业名称']
-        head_info.verify_status = verify_status
+        head_info = {
+            "企业名称": None if not company_data else company_data['企业名称'],
+            "企业认证": "已认证" if verify_status == "是" else "未认证",
+            "综合信用评价": {
+                "信用等级": company_data['综信评价结果']['信用等级'] if company_data.__contains__('综合信用评价') else None,
+                "信用评分": company_data['综信评价结果']['信用评分'] if company_data.__contains__('综合信用评价') else None,
+                "评价时间": company_data['综信评价结果']['评价时间'] if company_data.__contains__('综合信用评价') else None,
+            },
+            "ESG评价": {
+                "评价等级": company_data['ESG评价结果']['评价等级'] if company_data.__contains__("ESG评价") else None,
+                "ESG得分": company_data['ESG评价结果']['ESG得分'] if company_data.__contains__("ESG评价") else None,
+                "评价时间": company_data['ESG评价结果']['评价时间'] if company_data.__contains__("ESG评价") else None,
+            }
+        }
 
-        latest_cc_rating = CompanyIndex.HeadInfo.LatestCCRating()
-        latest_esg_rating = CompanyIndex.HeadInfo.LatestESGRating()
-
-        try:
-            cc_rating = company_data['综信评价结果']
-            latest_cc_rating.level = cc_rating['信用等级']
-            latest_cc_rating.score = str(cc_rating['信用评分'])
-            latest_cc_rating.rating_time = cc_rating['评价时间']
-        except TypeError:
-            latest_cc_rating.level = None
-            latest_cc_rating.score = None
-            latest_cc_rating.rating_time = None
-
-        try:
-            esg_rating = company_data['ESG评价结果']
-            latest_esg_rating.level = esg_rating['评价等级']
-            latest_esg_rating.score = str(esg_rating['ESG得分'])
-            latest_esg_rating.rating_time = esg_rating['评价时间']
-        except TypeError:
-            latest_esg_rating.level = None
-            latest_esg_rating.score = None
-            latest_esg_rating.rating_time = None
-
-        head_info.latest_cc_rating = latest_cc_rating
-        head_info.latest_esg_rating = latest_esg_rating
-
-        self.head_info = head_info
+        return head_info
 
     def get_business_info(self):
         """企业主页 基本信息"""

Routes/Company/IndexRoute.py

@@ -15,8 +15,7 @@ def head_info_route(**kwargs):
     try:
         impl = CompanyIndexImpl()
         impl.cid = kwargs['cid']
-        impl.get_head_info()
-        result = impl.head_info.dict_to_return()
+        result = impl.get_head_info()
         return {"info": "查询结果", "result": result}, 200
     except ReturnConditionCheckFailed as e:
         return {"info": e.failed_info}, e.status_code