#!/usr/bin/env python3 # -*- coding: utf-8 -*- import logging import random import time from bulk_update.helper import bulk_update from django.db import transaction from django.db.models import Count from django.views import View from Model.models import SerialNumberModel, UserModel, UserSerialNumberModel, UserUIDModel, CompanySerialModel from Object.RedisObject import RedisObject from Object.ResponseObject import ResponseObject from Object.TokenObject import TokenObject from Service.AlgorithmService import AlgorithmBaseOn35 from Service.CommonService import CommonService from Service.ModelService import ModelService class SerialNumberView(View): def get(self, request, *args, **kwargs): request.encoding = 'utf-8' request_dict = request.GET operation = kwargs.get('operation') return self.validate(request_dict, operation) def post(self, request, *args, **kwargs): request.encoding = 'utf-8' request_dict = request.POST operation = kwargs.get('operation') return self.validate(request_dict, operation) def validate(self, request_dict, operation): response = ResponseObject() if operation == 'create': return self.do_create(request_dict, response) elif operation == 'getSerial': return self.do_get_serial_number(request_dict, response) token = request_dict.get('token', None) token = TokenObject(token) if token.code != 0: return response.json(token.code) if operation == 'quantity': return self.do_quantity(request_dict,token.userID, response) elif operation == 'allot': return self.do_allot(request_dict, response) else: return response.json(309) # 查询当前可用的UID的数量 def do_quantity(self, request_dict, userID, response): user_qs = UserModel.objects.filter(id=userID) if not user_qs.exists(): return response.json(9) user = user_qs[0] chenyunsuper = SerialNumberModel.objects.filter().count() result = UserSerialNumberModel.objects.filter(user__id=user.id, serial_number__use_status=1).aggregate(num=Count('serial_number__use_status')) chenyundev = result['num'] chenyun = SerialNumberModel.objects.filter(use_status=1).count() res_data = {'code': 0, 'chenyun': chenyun, 'chenyunsuper': chenyunsuper, 'chenyundev': chenyundev} return response.json(0, {'data': res_data}) def do_create(self, request_dict, response): quantity = int(request_dict.get('quantity', 0)) if not quantity: return response.json(444) try: try: sum = SerialNumberModel.objects.last().id except: sum = 0 serial_number_bulk = [] now_time = int(time.time()) algorithm = AlgorithmBaseOn35() for i in range(quantity): serial_number = algorithm.getLetter(sum) sum += 1 # sum每次递增1 # 前面补0至六位 serial_number = (6-len(serial_number))*'0' + serial_number serial_number_bulk.append(SerialNumberModel(serial_number=serial_number, add_time=now_time)) # 开启事务写入 with transaction.atomic(): SerialNumberModel.objects.bulk_create(serial_number_bulk) return response.json(0) except Exception as e: print(e) return response.json(500, repr(e)) # 分配序列号 @transaction.atomic def do_allot(self, request_dict, response): username = request_dict.get('username', None) quantity = int(request_dict.get('quantity', None)) token = request_dict.get('token', None) token = TokenObject(token) if token.code != 0: return response.json(token.code) user = UserModel.objects.get(id=token.userID) if not user or '0' not in user.permission: return response.json(404) # 要分配的对象 allot_user_qs = UserModel.objects.filter(username=username) if not allot_user_qs.exists(): return response.json(444, 'username') # 取出对应区域可用的UID分配给allot_user sn_qs = SerialNumberModel.objects.filter(use_status=0)[0:quantity] sns = [] for sn in sn_qs: sns.append(sn.serial_number) cs_qs = CompanySerialModel.objects.filter(serial_number__in=sns, status=1) sns = [] for cs in cs_qs: sns.append(cs.serial_number) sn_qs = SerialNumberModel.objects.filter(serial_number__in=sns) count = sn_qs.count() if count < quantity: return response.json(444, '序列号不足') updates = [] datas = [] count = 0 if sn_qs.exists(): sn_qs = sn_qs[0:quantity] now_time = int(time.time()) for i in range(len(sn_qs)): item = sn_qs[i] serialNumberModel = SerialNumberModel( id=item.id, serial_number=item.serial_number, status=item.status, use_status=1, add_time=item.add_time ) user_serial_number = UserSerialNumberModel() user_serial_number.serial_number = serialNumberModel user_serial_number.user = allot_user_qs[0] user_serial_number.add_time = now_time user_serial_number.update_time = now_time datas.append(user_serial_number) updates.append(serialNumberModel) if len(updates) % 5000 == 0: bulk_update(updates) UserSerialNumberModel.objects.bulk_create(datas) if len(updates) > 0: bulk_update(updates) UserSerialNumberModel.objects.bulk_create(datas) del datas del updates return response.json(0) else: return response.json(444) # 提供给pc端获取序列号 def do_get_serial_number(self, request_dict, response): quantity = 1 #只能取一个 company_id = request_dict.get('company_id', None) token = request_dict.get('token', None) time_stamp = request_dict.get('time_stamp', None) mac = request_dict.get('mac', None) if token and time_stamp and quantity and company_id: token = int(CommonService.decode_data(token)) time_stamp = int(time_stamp) now_time = int(time.time()) distance = now_time - time_stamp if token != time_stamp or distance > 60000 or distance < -60000: # 为了全球化时间控制在一天内 return response.json(404) redisObject = RedisObject() key = 'serial_lock' value = redisObject.lpop(key) count = 0 while value is False and count < 5: time.sleep(1) value = redisObject.lpop(key) count += 1 if count == 5 and value is False: # 暂时注释 return response.json(5) userid = '3' user_qs = UserModel.objects.filter(id=userid) if not user_qs.exists(): return response.json(9) user = user_qs[0] result = UserSerialNumberModel.objects.filter(user__id=user.id, serial_number__use_status=1).aggregate( num=Count('serial_number__use_status')) us_qs = UserSerialNumberModel.objects.filter(user__id=user.id, serial_number__use_status=1).values( 'serial_number__serial_number') sns = [] for us in us_qs: sns.append(us['serial_number__serial_number']) cs_qs = CompanySerialModel.objects.filter(serial_number__in=sns, status=1, company__secret=company_id).values('serial_number', 'company__mark') sns = [] mark = "" if cs_qs.exists(): for cs in cs_qs: sns.append(cs['serial_number']) mark = cs_qs[0]['company__mark'] sn_qs = SerialNumberModel.objects.filter(serial_number__in=sns) count = result['num'] if count < quantity: return response.json(444, '序列号不足') # company_serial_qs = company_serial_qs.values('id', 'serial_number__serial_number', 'company__mark') save_point = transaction.savepoint() # 设置事务保存点 try: if sn_qs.exists(): sn_qs = sn_qs[0: quantity] serial_number = sn_qs[0].serial_number serialNumberModel = SerialNumberModel( id=sn_qs[0].id, serial_number=serial_number, status=sn_qs[0].status, use_status=2, add_time=sn_qs[0].add_time ) updates = [] updates.append(serialNumberModel) bulk_update(updates) except Exception as e: if save_point: transaction.savepoint_rollback(save_point) # 事务回滚 redisObject.rpush(key, value) return response.json(0, {'serial_number': serial_number + mark, 'mac': mac}) else: return response.json(444)