From 99684540b2c6d05ebb3b21163da0a5e314a07052 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期一, 30 九月 2024 01:11:50 +0800
Subject: [PATCH] bug修复

---
 utils/ths_industry_util.py |   32 +++++++++++++++++++++++++++++++-
 1 files changed, 31 insertions(+), 1 deletions(-)

diff --git a/utils/ths_industry_util.py b/utils/ths_industry_util.py
index 3930479..1130676 100644
--- a/utils/ths_industry_util.py
+++ b/utils/ths_industry_util.py
@@ -6,10 +6,35 @@
 import time
 
 from code_attribute import global_data_loader
+from db.mysql_data_delegate import Mysqldb
 from utils import global_util, tool
 from db import mysql_data_delegate as mysql_data
 
+
 # 鑾峰彇琛屼笟鏄犲皠
+
+class ThsCodeIndustryManager:
+    __instance = None
+    __mysql = Mysqldb()
+    __code_industry = {}
+
+    def __new__(cls, *args, **kwargs):
+        if not cls.__instance:
+            cls.__instance = super(ThsCodeIndustryManager, cls).__new__(cls, *args, **kwargs)
+            cls.__load_data()
+        return cls.__instance
+
+    @classmethod
+    def __load_data(cls):
+        results = cls.__mysql.select_all("select _id,second_industry from ths_industry_codes")
+        if results:
+            for r in results:
+                code = r[0]
+                industry = r[1]
+                cls.__code_industry[code] = industry
+
+    def get_industry(self, code):
+        return self.__code_industry.get(code)
 
 
 def get_code_industry_maps():
@@ -122,7 +147,7 @@
         else:
             mysqldb.execute(
                 "update ths_industry_codes set second_industry='{}',zyltgb='{}',zyltgb_unit={} where _id='{}'".format(
-                     industry_name, zyltgb, zyltgb_unit, code))
+                    industry_name, zyltgb, zyltgb_unit, code))
 
 
 # 淇濆瓨琛屼笟浠g爜
@@ -135,6 +160,10 @@
             # 淇濆瓨
             code = data["code"]
             __save_code_industry(code, code_names.get(code), industry_name, data["zyltgb"], data["zyltgb_unit"])
+
+
+def save_code_industry(code, code_name, industry):
+    __save_code_industry(code, code_name, industry, 0, 0)
 
 
 # 鏍规嵁鍚嶇О鑾峰彇浠g爜
@@ -155,6 +184,7 @@
     else:
         return None
 
+
 if __name__ == "__main__":
     _code_map, _industry_map = get_code_industry_maps()
     print(_code_map, _industry_map)

--
Gitblit v1.8.0