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]


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)