fix20230109

apps/view_atlas/view.py

@@ -217,7 +217,6 @@ def find_up_thr1(industry_type, industry_id):
     for pname_one in res_one:  # 一级节点
         chain_name1 = pname_one["chain_name"]
         total_count = 0
-        node_one = ''
         # 一级的企业数量
         company_count = get_count(chain_name1)
         total_count += company_count
@@ -234,8 +233,7 @@ def find_up_thr1(industry_type, industry_id):
                     "chain_name_id": chain_two_obj.id
                     } for chain_two_obj in chain_two_obj_list]
 
-        for item in res_two:
-            pname_two = item
+        for pname_two in res_two:
             chain_name2 = pname_two["chain_name"]
             company_count = get_count(chain_name2)
             total_count += company_count
@@ -252,8 +250,7 @@ def find_up_thr1(industry_type, industry_id):
             res_three = [{"chain_name": chain_three_obj.industry_name,
                           "chain_name_id": chain_three_obj.id
                           } for chain_three_obj in chain_three_obj_list]
-            for itm in res_three:
-                pname_thr = itm
+            for pname_thr in res_three:
                 chain_name3 = pname_thr["chain_name"]
                 company_count = get_count(chain_name3)
                 total_count += company_count
@@ -266,13 +263,13 @@ def find_up_thr1(industry_type, industry_id):
                     "subNodeList": []
                 }
                 node_two["subNodeList"].append(node_thr)
-            # 一级
-            node_one = {
-                "node": "{}".format(chain_name1),
-                "level": 2,
-                "company_count": total_count,
-                "subNodeList": []
-            }
+            # # 一级
+            # node_one = {
+            #     "node": "{}".format(chain_name1),
+            #     "level": 2,
+            #     "company_count": total_count,
+            #     "subNodeList": []
+            # }
             node_one["subNodeList"].append(node_two)
         data["subNodeList"].append(node_one)
     return data