InitController.py 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122
  1. # @Author : Rocky
  2. # @File : InitController.py
  3. # @Time : 2023/4/11 17:26
  4. import json
  5. import ssl
  6. import time
  7. from django.http import HttpResponse
  8. from django.views import View
  9. from redis.connection import SSLConnection
  10. from Model.models import Device_Info, SceneLog, EquipmentInfo1
  11. from Object.RedisObject import RedisObject
  12. from Object.ResponseObject import ResponseObject
  13. import redis
  14. class InitView(View):
  15. def get(self, request, *args, **kwargs):
  16. request.encoding = 'utf-8'
  17. operation = kwargs.get('operation')
  18. return self.validation(request.GET, operation)
  19. def post(self, request, *args, **kwargs):
  20. request.encoding = 'utf-8'
  21. operation = kwargs.get('operation')
  22. return self.validation(request.POST, operation)
  23. def validation(self, request_dict, operation):
  24. if operation == 'health-check': # 负载均衡器健康检测接口
  25. return self.health_check(request_dict)
  26. elif operation == 'oci_redis_test':
  27. return self.oci_redis_test(request_dict)
  28. elif operation == 'oci_redis_test_2':
  29. return self.oci_redis_test_2(request_dict)
  30. @staticmethod
  31. def health_check(request_dict):
  32. try:
  33. redis_obj = RedisObject()
  34. redis_obj.set_data('health_check', 1)
  35. response = ResponseObject()
  36. Device_Info.objects.filter().values('id').first()
  37. SceneLog.objects.filter().values('id').first()
  38. return response.json(0)
  39. except Exception as e:
  40. return HttpResponse(repr(e), status=500)
  41. @staticmethod
  42. def oci_redis_test(request_dict):
  43. response = ResponseObject()
  44. try:
  45. key = request_dict.get('key', None)
  46. value = request_dict.get('value', None)
  47. # import rediscluster
  48. # from rediscluster import ClusterConnectionPool
  49. # password = '012bzaKUhleHc2645465'
  50. # startup_nodes = [
  51. # {"host": "10.60.1.113", "port": 6379},
  52. # ]
  53. # pool = ClusterConnectionPool(startup_nodes=startup_nodes, password=password)
  54. # r = rediscluster.RedisCluster(connection_pool=pool)
  55. # r.ping()
  56. # r.set(name=key, value=value, ex=60)
  57. # redis_value = r.get(name=key)
  58. redis_obj = RedisObject()
  59. redis_obj.set_data(key, value, 60)
  60. redis_value = redis_obj.get_data(key)
  61. res = {
  62. 'redis_value': redis_value
  63. }
  64. return response.json(0, res)
  65. except Exception as e:
  66. return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  67. @staticmethod
  68. def oci_redis_test_2(request_dict):
  69. try:
  70. redis_obj = RedisObject()
  71. pipe = redis_obj.CONN.pipeline()
  72. response = ResponseObject()
  73. equipment_info_kwargs = {
  74. 'device_user_id': '163417566733313800138000',
  75. 'event_time': int(time.time()),
  76. 'event_type': 0,
  77. 'device_uid': 'H47UZJ7PHY2NXKNW111A',
  78. 'device_nick_name': 'redis_test',
  79. 'channel': 1,
  80. 'alarm': 'Motion',
  81. 'is_st': 0,
  82. 'add_time': int(time.time()),
  83. 'storage_location': 1,
  84. 'event_tag': '',
  85. 'answer_status': 0
  86. }
  87. # 保存到redis列表
  88. equipment_info_value = json.dumps(equipment_info_kwargs)
  89. equipment_info_key = 'equipment_info'
  90. pipe.rpush(equipment_info_key, equipment_info_value)
  91. pipe.rpush(equipment_info_key, equipment_info_value)
  92. pipe.execute()
  93. equipment_info_redis_list = redis_obj.lrange(equipment_info_key, 0, 99)
  94. redis_obj.ltrim(equipment_info_key, 100, -1)
  95. equipment_info_list = []
  96. for equipment_info in equipment_info_redis_list:
  97. equipment_info_data = eval(equipment_info)
  98. # 设备昵称存在表情,解码utf-8
  99. if equipment_info_data.get('device_nick_name') is not None:
  100. equipment_info_data['device_nick_name'] = equipment_info_data['device_nick_name']. \
  101. encode('UTF-8', 'ignore').decode('UTF-8')
  102. equipment_info_list.append(EquipmentInfo1(**equipment_info_data))
  103. EquipmentInfo1.objects.bulk_create(equipment_info_list)
  104. return response.json(0)
  105. except Exception as e:
  106. return HttpResponse(repr(e), status=500)