异常保护

Administrator

4 天以前 48fb7a00951f91bdc707e5dd2d196e5bccb752c3

 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))


# 保存行业代码
@@ -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)


# 根据名称获取代码
@@ -155,6 +184,7 @@
    else:
        return None


if __name__ == "__main__":
    _code_map, _industry_map = get_code_industry_maps()
    print(_code_map, _industry_map)