소스 검색

批量创建推送数据

locky 1 년 전
부모
커밋
0506a3a669
2개의 변경된 파일30개의 추가작업 그리고 2개의 파일을 삭제
  1. 20 2
      Service/DevicePushService.py
  2. 10 0
      Service/EquipmentInfoService.py

+ 20 - 2
Service/DevicePushService.py

@@ -155,6 +155,8 @@ class DevicePushService:
                 cls.is_send_app_push(
                     params['event_type'], params['event_tag'], params['app_push_config'], params['app_push'], uid)
 
+            equipment_info_list = []
+            equipment_info_model = EquipmentInfoService.randoms_choice_equipment_info()
             for up in params['uid_set_push_list']:
                 lang = up['lang']
                 tz = up['tz']
@@ -175,8 +177,22 @@ class DevicePushService:
                     # 保存推送消息
                     else:
                         params['userID_id'] = user_id
-                        cls.save_equipment_info(**params)
-
+                        equipment_info_list.append(
+                            equipment_info_model(
+                                device_user_id=params['userID_id'],
+                                event_time=params['n_time'],
+                                event_type=params['event_type'],
+                                device_uid=params['uid'],
+                                device_nick_name=params['nickname'],
+                                channel=params['channel'],
+                                alarm='Motion \tChannel:{}'.format(params['channel']),
+                                is_st=params['is_st'],
+                                add_time=int(time.time()),
+                                storage_location=params['storage_location'],
+                                event_tag=params['event_tag'],
+                                answer_status=True if params['dealings_type'] == 1 else False
+                            )
+                        )
                     saved_user_id_list.append(user_id)
 
                 # 推送
@@ -216,6 +232,8 @@ class DevicePushService:
             # 写入系统消息
             if sys_msg_list:
                 SysMsgModel.objects.bulk_create(sys_msg_list)
+            if equipment_info_list:
+                equipment_info_model.objects.bulk_create(equipment_info_list)
 
             return True
         except Exception as e:

+ 10 - 0
Service/EquipmentInfoService.py

@@ -19,6 +19,12 @@ from Model.models import EquipmentInfoMonday, EquipmentInfoTuesday, EquipmentInf
 from Object.utils import LocalDateTimeUtil
 from AnsjerPush.config import LOGGER
 
+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:
 
@@ -102,3 +108,7 @@ class EquipmentInfoService:
             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)