# -*- encoding: utf-8 -*- """ @File : AgentOrderController.py @Time : 2024/3/14 10:53 @Author : stephen @Email : zhangdongming@asj6.wecom.work @Software: PyCharm """ import threading import time from datetime import datetime, timedelta from decimal import Decimal, ROUND_DOWN from django.http import QueryDict from django.views import View from AgentModel.models import AgentDevice, AgentCloudServicePackage, AgentCustomerPackage, AgentDeviceOrder, \ AgentDeviceOrderInstallment, AgentAccount from Ansjer.config import LOGGER from Model.models import Order_Model, Store_Meal, UnicomCombo, TimeZoneInfo from Object.CeleryBeatObject import CeleryBeatObj from Object.ResponseObject import ResponseObject from Object.TokenObject import TokenObject from Service.CommonService import CommonService class AgentOrderView(View): def get(self, request, *args, **kwargs): request.encoding = 'utf-8' operation = kwargs.get('operation') return self.validation(request.GET, request, operation) def post(self, request, *args, **kwargs): request.encoding = 'utf-8' operation = kwargs.get('operation') return self.validation(request.POST, request, operation) def delete(self, request, *args, **kwargs): request.encoding = 'utf-8' operation = kwargs.get('operation') delete = QueryDict(request.body) if not delete: delete = request.GET return self.validation(delete, request, operation) def put(self, request, *args, **kwargs): request.encoding = 'utf-8' operation = kwargs.get('operation') put = QueryDict(request.body) return self.validation(put, request, operation) def validation(self, request_dict, request, operation): response = ResponseObject() tko = TokenObject( request.META.get('HTTP_AUTHORIZATION'), returntpye='pc') if operation == 'addOrder': # 添加代理商订单 order_id = request_dict.get('order_id', None) uid = request_dict.get('uid', None) order_type = request_dict.get('order_type', None) package_id = request_dict.get('package_id', None) self.check_agent_service_package(order_id, uid, int(package_id)) return response.json(0) elif operation == 'addSettlementJob': self.add_settlement_job() return response.json(0) elif operation == 'delSettlementJob': self.del_settlement_job() return response.json(0) elif operation == 'updateSettlementJob': self.update_settlement_job() return response.json(0) @classmethod def update_settlement_job(cls): celery_beat_obj = CeleryBeatObj() job_name = 'Agent-updateSettlement' time_zone_info_qs = TimeZoneInfo.objects.filter(tz=8).values('zone_info') if time_zone_info_qs.exists(): time_zone = time_zone_info_qs[0]['zone_info'] cron_tuple = ('*/3', '*', '*', '*', '*', time_zone) celery_beat_obj.update_task(name=job_name, crontab=cron_tuple) @classmethod def add_settlement_job(cls): celery_beat_obj = CeleryBeatObj() job_name = 'Agent-updateSettlement' SMART_SCENE_TASK = 'Controller.CeleryTasks.tasks.update_installment_settlement_order' celery_beat_obj.creat_crontab_task( timezone_offset=8, name=job_name, task=SMART_SCENE_TASK, minute='*/2') @classmethod def del_settlement_job(cls): celery_beat_obj = CeleryBeatObj() job_name = 'Agent-updateSettlement' celery_beat_obj.del_task(job_name) @classmethod def check_agent_service_package(cls, order_id, uid, package_id): """ 检查是否代理服务套餐 @param package_id: 套餐id @param order_id: 订单ID @param uid: UID @return: True | False """ try: serial_number = CommonService.get_serial_number_by_uid(uid) a_device_qs = AgentDevice.objects.filter(serial_number=serial_number) \ .values('ac_id', 'type', 'status') LOGGER.info(f'******check_agent_service_package检查是否代理*****orderId:{order_id}') if not a_device_qs.exists(): return False LOGGER.info(f'******check_agent_service_package当前设备属于代理商*****serial_number:{serial_number}') asy = threading.Thread(target=cls.save_agent_package, args=(order_id, serial_number, a_device_qs[0]['ac_id'], package_id)) asy.start() return True except Exception as e: LOGGER.info('*****AgentOrderView.check_agent_service_package:errLine:{}, errMsg:{}' .format(e.__traceback__.tb_lineno, repr(e))) return False @classmethod def save_agent_package(cls, order_id, serial_number, ac_id, package_id): """ 保存代理套餐 """ try: order_qs = Order_Model.objects.filter(orderID=order_id, status=1).values('price', 'payTime', 'order_type') if not order_qs.exists(): LOGGER.info(f'******save_agent_package当前代理客户未添加此套餐******ac_id:{ac_id},package_id:{package_id}') return order_type = order_qs[0]['order_type'] package_type = 2 if order_type in [2, 3, 5] else 1 # 判断订单信息是云存还是4G package_id = int(package_id) agent_package_qs = AgentCloudServicePackage.objects.filter(type=package_type, package_id=package_id, status=1) if not agent_package_qs.exists(): LOGGER.info(f'******save_agent_package当前套餐未设置代理******order_id:{order_id},serial_number:{serial_number}') return agent_package = agent_package_qs.first() # 代理云服务套餐 LOGGER.info(f'******save_agent_package代理套餐******service_name:{agent_package_qs.first().service_name}') acp_qs = AgentCustomerPackage.objects.filter(ac_id=ac_id, cs_id=agent_package.id).values('id') if not acp_qs.exists(): LOGGER.info(f'******save_agent_package当前代理客户未添加此套餐******ac_id:{ac_id},package_id:{package_id}') return # 组装数据 now_time = int(time.time()) pay_price = Decimal(order_qs[0]['price']).quantize(Decimal('0.00')) profit = cls.calculate_order_profit(agent_package, pay_price) dict_data = {'ac_id': ac_id, 'serial_number': serial_number, 'csp_id': agent_package.id, 'order_id': order_id, 'status': 1, 'profit_amount': pay_price, 'profit': profit, 'pay_time': order_qs[0]['payTime'], 'created_time': now_time, 'updated_time': now_time} agent_order_obj = AgentDeviceOrder.objects.create(**dict_data) # 保存分期结算记录 cls.save_order_installment(agent_order_obj.id, package_type, package_id, profit, ac_id) LOGGER.info(f'******save_agent_package代理订单存表结束：{dict_data}') except Exception as e: LOGGER.info('*****AgentOrderView.save_agent_package:errLine:{}, errMsg:{}' .format(e.__traceback__.tb_lineno, repr(e))) @classmethod def calculate_order_profit(cls, agent_package, price): """ 计算利润 @param agent_package: 套餐配置 @param price: 支付价格 @return: 利润 """ profit = 0 price = Decimal(price).quantize(Decimal('0.00')) if agent_package.profit_type == 1: profit = agent_package.profit elif agent_package.profit_type == 2: profit_value = Decimal(agent_package.profit).quantize(Decimal('0.00')) cost = Decimal(agent_package.cost).quantize(Decimal('0.00')) profit = (price - cost) * (profit_value / 100) profit = profit.quantize(Decimal('0.00')) return profit @classmethod def get_settlement_interval(cls, package_type, package_id): try: if package_type == 1: # 云存 store_qs = Store_Meal.objects.filter(id=package_id).values('day', 'bucket_id', 'expire', 'icloud_store_meal_id') if not store_qs.exists(): return [] # 根据套餐周期计算往后每个月26号作为结算时间 return cls.get_future_timestamps(store_qs[0]['expire']) elif package_type == 2: # 4G combo4g_qs = UnicomCombo.objects.filter(id=package_id).values('expiration_days', 'expiration_type') if not combo4g_qs.exists(): return [] # 目前4G套餐都是基于按天类型创建 if combo4g_qs[0]['expiration_type'] == 0 and combo4g_qs[0]['expiration_days'] > 0: months = int(combo4g_qs[0]['expiration_days'] / 30) # 根据套餐周期计算往后每个月26号作为结算时间 return cls.get_future_timestamps(months) except Exception as e: LOGGER.info('*****AgentOrderView.get_settlement_interval:errLine:{}, errMsg:{}' .format(e.__traceback__.tb_lineno, repr(e))) return [] @staticmethod def get_future_timestamps(months): """ 生成未来若干个月的第一个月的26号11点的timestamp列表。 参数： months -- 未来需要生成timestamp的月份数量 返回值： timestamps -- 包含未来months个月第一个月的26号11点的timestamp的列表 """ current_time = datetime.now() # 获取当前时间，注意这会是系统当前时区的时间 current_month = current_time.month current_year = current_time.year timestamps = [] for _ in range(months): # 如果当前月已经是需要生成的月份，则年份和月份不变 if current_month == 1 and _ == 0: next_year = current_year next_month = current_month + 1 else: # 计算下一个月的年和月 if current_month == 12: next_month = 1 next_year = current_year + 1 else: next_month = current_month + 1 next_year = current_year # 生成下个月的26号11点的时间点 next_date = datetime(next_year, next_month, 26, 11, 0, 0) # 如果生成的日期超过了当月的实际天数（比如2月没有26号），则需要回退到当月的最后一天 last_day_of_month = (datetime(next_year, next_month, 1) + timedelta(days=31)).replace(day=1) - timedelta( days=1) if next_date > last_day_of_month: next_date = last_day_of_month.replace(hour=11, minute=0, second=0, microsecond=0) timestamps.append(int(next_date.timestamp())) # 更新当前月份和年份为下一次循环使用 current_month = next_month current_year = next_year return timestamps @staticmethod def distribute_commission(commission, periods): # 转换佣金和期数为Decimal类型，并设置精度 commission = Decimal(str(commission)).quantize(Decimal('0.01')) periods = Decimal(periods) # 每期基础金额（向下取整到最接近的0.01） base_amount = (commission / periods).quantize(Decimal('0.01'), rounding=ROUND_DOWN) # 初始化每期分配的金额列表 distributed_amounts = [base_amount] * int(periods) # 计算按照基础金额分配后的总和 total_distributed = sum(distributed_amounts) # 计算剩余需要分配的金额 remainder = commission - total_distributed # 分配剩余金额 if remainder > Decimal('0'): # 从第一期开始分配剩余金额 for i in range(len(distributed_amounts)): if remainder >= Decimal('0.01'): distributed_amounts[i] += Decimal('0.01') remainder -= Decimal('0.01') else: # 如果剩余金额不足0.01，则将其全部加到当前期 distributed_amounts[i] += remainder break return distributed_amounts @classmethod def save_order_installment(cls, agent_order_id, package_type, package_id, profit, ac_id=None): """ 保存代理订单分期信息 :param cls: 类方法的约定参数 :param agent_order_id: 代理订单ID :param package_type: 套餐类型 :param package_id: 套餐ID :param profit: 利润总额 :return: 无返回值 :param ac_id: 代理客户ID """ try: # 根据包裹类型和ID获取结算时间间隔列表 time_list = cls.get_settlement_interval(package_type, package_id) period_number = len(time_list) # 计算分期总数 # 输入合理性检查 if period_number == 0 or profit <= 0: LOGGER.info(f'Invalid input parameters: period_number={period_number}, profit={profit}') return n_time = int(datetime.now().timestamp()) # 获取当前时间戳 # 利润总额按分期数平均分配 amount_list = cls.distribute_commission(profit, period_number) installment_list = [] # 遍历分期数，生成分期记录列表 for time_point in range(period_number): installment_list.append(AgentDeviceOrderInstallment(ado_id=agent_order_id, period_number=period_number, ac_id=ac_id, amount=amount_list[time_point], due_date=time_list[time_point], status=1, created_time=n_time, updated_time=n_time)) # 分批处理大量数据，避免数据库压力过大 batch_size = 100 for i in range(0, len(installment_list), batch_size): AgentDeviceOrderInstallment.objects.bulk_create( installment_list[i:i + batch_size] ) # 记录创建完成的日志 LOGGER.info(f'*****AgentOrderView.save_OrderInstallment分期结算记录创建完成:{len(installment_list)} records') except Exception as e: # 记录异常信息 LOGGER.error('*****AgentOrderView.save_OrderInstallment:errLine:{}, errMsg:{}' .format(e.__traceback__.tb_lineno, repr(e))) @staticmethod def update_periodic_settlement(): """ 更新周期结算信息 返回值: - 无返回值 """ try: # 根据条件查询需要更新结算信息的订单分期记录 adoi_qs = AgentDeviceOrderInstallment.objects.filter(status=1, due_date__lte=int(time.time())) if not adoi_qs: # 如果没有找到符合条件的记录，直接返回 return ids = [] a_account_list = [] adoi_set = set() n_time = int(time.time()) for item in adoi_qs: # 准备分期记录的id列表和账户记录列表 ids.append(item.id) adoi_set.add(item.ado_id) a_account_list.append(AgentAccount(ac_id=item.ac_id, amount=item.amount, remark=f'周期结算', status=1, created_time=n_time, updated_time=n_time)) batch_size = 100 # 分批更新分期记录状态 for i in range(0, len(ids), batch_size): AgentDeviceOrderInstallment.objects.filter(id__in=ids[i:i + batch_size]) \ .update(status=2, settlement_time=n_time, updated_time=n_time) # 分批创建账户记录 for i in range(0, len(a_account_list), batch_size): AgentAccount.objects.bulk_create(a_account_list[i:i + batch_size]) # 检查是否所有分期都已结算，如果是，则更新订单状态为已结算 for ado in adoi_set: adoi_qs = AgentDeviceOrderInstallment.objects.filter(ado_id=ado, status=1) if not adoi_qs.exists(): AgentDeviceOrder.objects.filter(id=ado, status=1) \ .update(status=2, settlement_time=n_time, updated_time=n_time) except Exception as e: # 记录异常信息 LOGGER.error( f'*****AgentOrderView.update_periodic_settlement:errLine:{e.__traceback__.tb_lineno}, errMsg:{str(e)}')