# -*- 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 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) @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'检查当前订单是否绑定代理*****orderID:{order_id},serialNumber:{serial_number}') if not a_device_qs.exists(): return False LOGGER.info(f'当前设备属于代理商orderID:{order_id},serialNumber:{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.error('*****支付成功保存云服务代理订单异常orderID:{},errLine:{}, errMsg:{}' .format(order_id, 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, order_qs[0]['payTime']) 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_quarterly_settlement_dates(cls, start_date, months): """ 获取季度结算日期列表，按照以下规则： 1. 固定在四个季度结算日期（1月1日、4月1日、7月1日、10月1日）进行结算 2. 从购买时间到结算日不满1个月的不在当前季度结算，累积到下一个季度 3. 中间季度每季度计算3个月 4. 最后一个季度计算剩余的时间 :param start_date: 套餐开始日期(datetime) :param months: 套餐总月数 :return: 包含(结算日期timestamp, 该季度使用月数)的元组列表 """ # 固定的季度结算日期 QUARTER_DATES = [(1, 1), (4, 1), (7, 1), (10, 1)] # (月, 日) # 计算套餐结束日期 end_date = start_date + timedelta(days=int(months * 30.5)) # 初始化结果列表 result = [] # 找到开始日期后的第一个季度结算日 current_year = start_date.year current_quarter_idx = 0 # 找到开始日期之后的第一个季度结算日 for i, (month, day) in enumerate(QUARTER_DATES): quarter_date = datetime(current_year, month, day) if quarter_date > start_date: current_quarter_idx = i break else: # 如果当年没有更多季度结算日，则移到下一年的第一个季度结算日 current_year += 1 current_quarter_idx = 0 # 第一个季度的结算日期 month, day = QUARTER_DATES[current_quarter_idx] first_settlement_date = datetime(current_year, month, day) # 计算第一个季度的整月数 days_in_first_quarter = (first_settlement_date - start_date).days whole_months_first_quarter = int(days_in_first_quarter / 30.5) # 计算剩余的月数（总月数减去第一个季度的整月数，如果第一个季度有整月数） remaining_months = months # 如果第一个季度有整月数，则添加第一个季度的结算记录并减去已结算的月数 if whole_months_first_quarter >= 1: result.append((int(first_settlement_date.timestamp()), whole_months_first_quarter)) remaining_months -= whole_months_first_quarter else: # 即使不足1个月，也添加第一个季度的结算记录，但月数为0 # 这样可以确保在7月1日进行第一次结算 result.append((int(first_settlement_date.timestamp()), 0)) # 如果没有剩余月数，直接返回结果 if remaining_months <= 0: return result # 特殊处理年套餐（12个月）的情况 if months == 12 and whole_months_first_quarter == 0: # 确保总共有5次季度结算，最后一次是剩余的月数 # 第一次结算已经添加（7月1日，整月数为0） # 添加第二次结算（10月1日，整月数为3） current_quarter_idx = (current_quarter_idx + 1) % len(QUARTER_DATES) if current_quarter_idx == 0: current_year += 1 month, day = QUARTER_DATES[current_quarter_idx] settlement_date = datetime(current_year, month, day) result.append((int(settlement_date.timestamp()), 3)) # 添加第三次结算（1月1日，整月数为3） current_quarter_idx = (current_quarter_idx + 1) % len(QUARTER_DATES) if current_quarter_idx == 0: current_year += 1 month, day = QUARTER_DATES[current_quarter_idx] settlement_date = datetime(current_year, month, day) result.append((int(settlement_date.timestamp()), 3)) # 添加第四次结算（4月1日，整月数为3） current_quarter_idx = (current_quarter_idx + 1) % len(QUARTER_DATES) if current_quarter_idx == 0: current_year += 1 month, day = QUARTER_DATES[current_quarter_idx] settlement_date = datetime(current_year, month, day) result.append((int(settlement_date.timestamp()), 3)) # 添加第五次结算（7月1日，整月数为剩余的月数） current_quarter_idx = (current_quarter_idx + 1) % len(QUARTER_DATES) if current_quarter_idx == 0: current_year += 1 month, day = QUARTER_DATES[current_quarter_idx] settlement_date = datetime(current_year, month, day) # 剩余的月数为12减去前面已经结算的月数 remaining = 12 - (0 + 3 + 3 + 3) result.append((int(settlement_date.timestamp()), remaining)) return result # 非年套餐的处理逻辑 # 计算完整季度的数量（每季度3个月） full_quarters = int(remaining_months / 3) # 计算最后一个季度的剩余月数 last_quarter_months = remaining_months % 3 # 当前日期设置为第一个结算日期 current_date = first_settlement_date # 添加完整季度的结算记录 for _ in range(full_quarters): # 移到下一个季度结算日 current_quarter_idx = (current_quarter_idx + 1) % len(QUARTER_DATES) if current_quarter_idx == 0: current_year += 1 month, day = QUARTER_DATES[current_quarter_idx] settlement_date = datetime(current_year, month, day) # 添加完整季度的结算记录（3个月） result.append((int(settlement_date.timestamp()), 3)) # 更新当前日期 current_date = settlement_date # 如果有剩余月数，添加最后一个季度的结算记录 if last_quarter_months > 0: # 移到下一个季度结算日 current_quarter_idx = (current_quarter_idx + 1) % len(QUARTER_DATES) if current_quarter_idx == 0: current_year += 1 month, day = QUARTER_DATES[current_quarter_idx] settlement_date = datetime(current_year, month, day) # 添加最后一个季度的结算记录（剩余月数） result.append((int(settlement_date.timestamp()), last_quarter_months)) return result @staticmethod def calculate_months_in_period(start_date, end_date): """ 计算两个日期之间的整月数，不足一个月不计入 :param start_date: 开始日期 :param end_date: 结束日期 :return: 整月数 """ # 计算天数差 days_diff = (end_date - start_date).days # 转换为整月数(按平均每月30.5天计算) whole_months = int(days_diff / 30.5) return whole_months @staticmethod def calculate_quarterly_profit(profit, settlement_dates_with_months): """ 计算季度利润分配，基于整月数 :param profit: 总利润 :param settlement_dates_with_months: 包含(结算日期, 整月数)的元组列表 :return: 每个季度的利润列表 """ profit = Decimal(str(profit)).quantize(Decimal('0.01')) # 计算总整月数 total_months = sum(months for _, months in settlement_dates_with_months) # 如果总月数为0，返回空列表 if total_months == 0: return [] # 计算每月利润 monthly_profit = profit / Decimal(total_months) # 计算每个季度的利润 quarterly_amounts = [] for _, months in settlement_dates_with_months: # 计算当前季度的利润（基于整月数） amount = (monthly_profit * Decimal(months)).quantize(Decimal('0.01'), rounding=ROUND_DOWN) quarterly_amounts.append(amount) # 处理舍入误差，确保总和等于总利润 total_allocated = sum(quarterly_amounts) remainder = profit - total_allocated # 将剩余的分配到第一个季度 if remainder > Decimal('0') and quarterly_amounts: quarterly_amounts[0] += remainder return quarterly_amounts @classmethod def save_order_installment(cls, agent_order_id, package_type, package_id, profit, ac_id=None, pay_time=0): """ 保存代理订单分期信息(季度结算逻辑)，不满一个月的时间累积到下一个季度 :param cls: 类方法的约定参数 :param agent_order_id: 代理订单ID :param package_type: 套餐类型(1:云存, 2:4G) :param package_id: 套餐ID :param profit: 利润总额 :param ac_id: 代理客户ID :param pay_time: 订单支付时间 :return: 无返回值 """ try: # 转换支付时间为datetime对象 pay_time_dt = datetime.fromtimestamp(pay_time) # 获取套餐月数 if package_type == 1: # 云存 store = Store_Meal.objects.filter(id=package_id).first() if not store: LOGGER.info(f'云存套餐不存在: {package_id}') return months = store.expire else: # 4G combo = UnicomCombo.objects.filter(id=package_id).first() if not combo: LOGGER.info(f'4G套餐不存在: {package_id}') return months = int(combo.expiration_days / 30) if months <= 0 or profit <= 0: LOGGER.info(f'无效参数: months={months}, profit={profit}') return LOGGER.info( f'开始计算季度结算: 订单ID={agent_order_id}, 开始日期={pay_time_dt}, 套餐月数={months}, 总利润={profit}') # 获取季度结算日期和每个季度的整月数 settlement_dates_with_months = cls.get_quarterly_settlement_dates(pay_time_dt, months) # 记录季度结算日期和月数 for i, (date, months_used) in enumerate(settlement_dates_with_months): date_str = datetime.fromtimestamp(date).strftime('%Y-%m-%d') LOGGER.info(f'季度{i + 1}结算日期: {date_str}, 整月数: {months_used}') # 如果没有有效的结算日期，则退出 if not settlement_dates_with_months: LOGGER.info(f'没有有效的季度结算日期: start_date={pay_time_dt}, months={months}') return # 计算每个季度的利润分配 amounts = cls.calculate_quarterly_profit(profit, settlement_dates_with_months) # 记录每个季度的利润分配 for i, amount in enumerate(amounts): LOGGER.info(f'季度{i + 1}利润分配: {amount}') # 创建分期记录 n_time = int(time.time()) installment_list = [] for i, ((settlement_date, months_used), amount) in enumerate(zip(settlement_dates_with_months, amounts)): # 只有使用满一个月才创建结算记录 if months_used >= 1: installment_list.append(AgentDeviceOrderInstallment( ado_id=agent_order_id, period_number=len(settlement_dates_with_months), ac_id=ac_id, amount=amount, due_date=settlement_date, status=1, created_time=n_time, updated_time=n_time )) # 批量创建 if installment_list: 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'季度分期结算记录创建完成: {len(installment_list)}条, 订单ID: {agent_order_id}') else: LOGGER.info(f'没有创建季度分期结算记录: 订单ID: {agent_order_id}') except Exception as e: LOGGER.error(f'保存季度分期结算记录异常: 行号:{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)}')