ASJPush/Service/EquipmentInfoService.py

# -*- encoding: utf-8 -*-
"""
@File    : EquipmentInfoService.py
@Time    : 2022/4/14 17:28
@Author  : stephen
@Email   : zhangdongming@asj6.wecom.work
@Software: PyCharm
"""
import random

from AnsjerPush.config import LOGGER
from Model.models import EquipmentInfo1, EquipmentInfo2, EquipmentInfo3, \
    EquipmentInfo4, EquipmentInfo5, EquipmentInfo6, EquipmentInfo7, EquipmentInfo8, EquipmentInfo9, EquipmentInfo10, \
    EquipmentInfo11, EquipmentInfo12, EquipmentInfo13, EquipmentInfo14, EquipmentInfo15, EquipmentInfo16, \
    EquipmentInfo17, EquipmentInfo18, EquipmentInfo19, EquipmentInfo20

EQUIPMENT_INFO_LIST = [
    EquipmentInfo1, EquipmentInfo2, EquipmentInfo3, EquipmentInfo4, EquipmentInfo5, EquipmentInfo6,
    EquipmentInfo7, EquipmentInfo8, EquipmentInfo9, EquipmentInfo10,
    EquipmentInfo11, EquipmentInfo12, EquipmentInfo13, EquipmentInfo14, EquipmentInfo15, EquipmentInfo16,
    EquipmentInfo17, EquipmentInfo18, EquipmentInfo19, EquipmentInfo20]

EQUIPMENT_INFO_KEY_LIST = [
    'EquipmentInfo1', 'EquipmentInfo2', 'EquipmentInfo3', 'EquipmentInfo4', 'EquipmentInfo5', 'EquipmentInfo6',
    'EquipmentInfo7', 'EquipmentInfo8', 'EquipmentInfo9', 'EquipmentInfo10',
    'EquipmentInfo11', 'EquipmentInfo12', 'EquipmentInfo13', 'EquipmentInfo14', 'EquipmentInfo15', 'EquipmentInfo16',
    'EquipmentInfo17', 'EquipmentInfo18', 'EquipmentInfo19', 'EquipmentInfo20']
EQUIPMENT_INFO_DICT = {
    'EquipmentInfo1': EquipmentInfo1,
    'EquipmentInfo2': EquipmentInfo2,
    'EquipmentInfo3': EquipmentInfo3,
    'EquipmentInfo4': EquipmentInfo4,
    'EquipmentInfo5': EquipmentInfo5,
    'EquipmentInfo6': EquipmentInfo6,
    'EquipmentInfo7': EquipmentInfo7,
    'EquipmentInfo8': EquipmentInfo8,
    'EquipmentInfo9': EquipmentInfo9,
    'EquipmentInfo10': EquipmentInfo10,
    'EquipmentInfo11': EquipmentInfo11,
    'EquipmentInfo12': EquipmentInfo12,
    'EquipmentInfo13': EquipmentInfo13,
    'EquipmentInfo14': EquipmentInfo14,
    'EquipmentInfo15': EquipmentInfo15,
    'EquipmentInfo16': EquipmentInfo16,
    'EquipmentInfo17': EquipmentInfo17,
    'EquipmentInfo18': EquipmentInfo18,
    'EquipmentInfo19': EquipmentInfo19,
    'EquipmentInfo20': EquipmentInfo20
}


class EquipmentInfoService:

    @staticmethod
    def randoms_insert_equipment_info(**kwargs):
        """
        随机一个推送数据表存储数据
        """
        equipment_info_list = [
            EquipmentInfo1, EquipmentInfo2, EquipmentInfo3, EquipmentInfo4, EquipmentInfo5, EquipmentInfo6,
            EquipmentInfo7, EquipmentInfo8, EquipmentInfo9, EquipmentInfo10,
            EquipmentInfo11, EquipmentInfo12, EquipmentInfo13, EquipmentInfo14, EquipmentInfo15, EquipmentInfo16,
            EquipmentInfo17, EquipmentInfo18, EquipmentInfo19, EquipmentInfo20]
        try:
            random_model = random.choice(equipment_info_list)
            LOGGER.info('随机存表参数:{},表:{}'.format(kwargs, random_model))
            random_model.objects.create(**kwargs)
        except Exception as e:
            LOGGER.info('随机存储推送数据异常: error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @staticmethod
    def randoms_choice_equipment_info():
        return random.choice(EQUIPMENT_INFO_LIST)

    @staticmethod
    def randoms_choice_equipment_info_key():
        return random.choice(EQUIPMENT_INFO_KEY_LIST)