ASJServer/AdminController/CloudServiceManage/AgentCustomerController.py

# -*- encoding: utf-8 -*-
"""
@File    : AgentCustomerController.py
@Time    : 2024/3/7 16:56
@Author  : stephen
@Email   : zhangdongming@asj6.wecom.work
@Software: PyCharm
"""
import re
import time
import json
from decimal import Decimal

from django.db.models import Sum, F, Q
from django.http import QueryDict
from django.views import View
from django.core.paginator import Paginator
from datetime import datetime, timedelta
from Ansjer.config import CONFIG_CN, LOGGER, CONFIG_INFO

from AgentModel.models import AgentCustomerInfo, AgentCustomerCard, AgentCustomerPackage, AgentCloudServicePackage, \
    AgentDeviceOrder, AgentAccountWithdraw, AgentDevice, AgentAccount, ApplyAgent, AgentDeviceOrderInstallment
from Model.models import UnicomCombo, Store_Meal, Device_User, CouponCombo

from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject


class AgentCustomerView(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):
        AgentCustomerInfo.objects.filter()
        language = request_dict.get('language', 'en')
        response = ResponseObject(language, 'pc')
        if operation == 'getUnicomAndIcloud':
            return self.get_unicom_and_icloud(request_dict, response)
        elif operation == 'applyAgent':
            return self.apply_agent(request_dict, response)
        else:
            tko = TokenObject(
                request.META.get('HTTP_AUTHORIZATION'),
                returntpye='pc')
            if tko.code != 0:
                return response.json(tko.code)
            response.lang = tko.lang
            userID = tko.userID
            # 代理用户界面（代理顾客个人信息）
            if operation == 'getAgentInfo':
                return self.get_agent_info(userID, response)

            # 代理云存套餐
            elif operation == 'getAgentServicePackage':
                return self.get_agent_service_package(request_dict, response)
            elif operation == 'addAgentServicePackage':
                return self.add_agent_service_package(userID, request_dict, response)
            elif operation == 'updateAgentServicePackage':
                return self.update_agent_service_package(userID, request_dict, response)
            elif operation == 'delAgentServicePackage':
                return self.del_agent_service_package(request_dict, response)

            # 代理客户绑定套餐
            elif operation == 'getCustomerList':
                return self.get_customer_list(request_dict, response)
            elif operation == 'getCustomerPackageList':
                return self.get_cumstomer_package_list(request_dict, response)
            elif operation == 'batchRebindCustomerPackage':
                return self.batch_rebind_customer_packages(userID, request_dict, response)
            elif operation == 'getAgentServicePackageList':
                return self.get_agent_service_package_list(response)

            elif operation == 'getAgentSettleOrders':
                return self.get_agent_settle_order(userID, request_dict, response)
            elif operation == 'getDataStatistics':
                return self.get_data_statistics(userID, response)

            # 代理商提现功能
            elif operation == 'getAgentAccountWithdraw':
                return self.get_agent_account_withdraw(userID, request_dict, response)
            elif operation == 'getCheckBalance':
                return self.get_check_balance(userID, response)
            elif operation == 'agentApplyWithdraw':
                return self.agent_apply_withdraw(userID, request_dict, response)
            elif operation == 'getWithdrawalReview':
                return self.get_withdrawal_review(request_dict, response)
            elif operation == 'updateWithdrawalReview':
                return self.update_withdrawal_review(userID, request_dict, response)
            else:
                return response.json(444, 'operation')

    @classmethod
    def get_unicom_and_icloud(cls, request_dict, response):
        """
        查询云存储套餐和物联网卡套餐列表
        @param response: 响应对象
        @return: 包含套餐信息的响应
        """
        try:
            lang = request_dict.get('lang', 'cn')

            # 1. 优化查询 - 云存储套餐
            store_meals_qs = Store_Meal.objects.filter(
                ~Q(pay_type__in=[10, 11]),
                lang__lang=lang,
                is_show=0
            ).exclude(app_type=2)

            # 根据配置区分国内外套餐
            if CONFIG_INFO == CONFIG_CN:
                store_meals_qs = store_meals_qs.filter(is_ai=0, bucket__mold=0)
            else:
                store_meals_qs = store_meals_qs.filter(bucket__mold=1)

            # 一次性获取所有需要的字段和关联数据
            store_meals = store_meals_qs.annotate(
                title=F('lang__title'),
                content=F('lang__content'),
                new_title=F('lang__new_title'),
                discount_content=F('lang__discount_content'),
                store_day=F('bucket__storeDay'),
                bucket_name=F('bucket__bucket'),
                bucket_area=F('bucket__area')
            ).order_by('-pixel_level', '-is_ai').values(
                "id", "title", "content", "price", "day", "currency", "store_day",
                "new_title", "bucket_name", "bucket_area", "commodity_code", "commodity_type",
                "is_discounts", "virtual_price", "expire", "discount_price",
                "discount_content", "symbol", "cycle_config_id",
                'pixel_level', 'is_ai', 'app_type'
            )

            # 2. 批量处理优惠券信息 - 避免N+1查询
            ios_meal_ids = [meal['id'] for meal in store_meals if meal.get('app_type') == 1]
            coupon_map = {}
            if ios_meal_ids:
                coupon_combos = CouponCombo.objects.filter(
                    combo_id__in=ios_meal_ids
                ).values_list('combo_id', flat=True).distinct()
                coupon_map = {combo_id: True for combo_id in coupon_combos}

            # 3. 使用列表推导式构建套餐列表
            store_meal_list = []
            for meal in store_meals:
                # 构建基础名称
                name_parts = [meal['title'],
                              f'{meal.get("symbol")}{meal["price"]}',
                              f"({str(meal['expire'])}month+",
                              '≥4K' if meal.get('pixel_level') == 1 else '<4K']

                # 添加AI标识
                if meal.get('is_ai') == 1:
                    name_parts.append('+AI')

                # 处理iOS套餐
                if meal.get('app_type') == 1:
                    name_parts.append('+ios')
                    # 检查优惠券 - 使用预处理的映射
                    if meal['id'] in coupon_map:
                        name_parts.append('+discount')

                # 添加周期订阅标识
                if meal.get('cycle_config_id', 0) == 1:
                    name_parts.append('+cycle')

                # 关闭括号并连接字符串
                name_parts.append(')')

                full_name = ''.join(name_parts)

                store_meal_list.append({
                    'id': meal['id'],
                    'name': full_name
                })

            # 4. 物联网卡套餐查询
            unicom_combo_list = list(
                UnicomCombo.objects.filter(
                    is_show=1,
                    is_del=False
                ).distinct().values('id', 'combo_name')
            )
            # 统一字段名
            for combo in unicom_combo_list:
                combo['name'] = combo.pop('combo_name')

            return response.json(0, {
                'storeMeals': store_meal_list,
                'unicomCombos': unicom_combo_list,
            })

        except Exception as e:
            LOGGER.error(f"get_unicom_and_icloud error: {str(e)}", exc_info=True)
            return response.json(500, 'Server error')

    def get_agent_info(self, userID, response):
        """
        查询用户信息
        @param userID: userID
        @param response: 响应对象
        @return:
        """
        try:
            # 使用userID查询AgentCustomerInfo获取基本信息
            agent_info = AgentCustomerInfo.objects.filter(user_id=userID, status=1).first()
            if not agent_info:
                return response.json(444, {'error': '没有找到这个代理用户'})
            # userID查询用户电话
            device_user_qs = Device_User.objects.filter(userID=userID).first()

            # 代理客户ID（ac_id）查询AgentCustomerCard获取银行卡信息。
            card_details = AgentCustomerCard.objects.filter(ac_id=agent_info.id, status=1).first()

            # ac_id查询AgentCustomerPackage来找到关联的云服务套餐ID（cs_id）
            package_ids = AgentCustomerPackage.objects.filter(ac_id=agent_info.id).values_list('cs_id',
                                                                                               flat=True)
            # cs_id查询AgentCloudServicePackage获取服务套餐详情。
            service_packages = AgentCloudServicePackage.objects.filter(id__in=package_ids, status=1).values()

            result = {
                'agent_info': {
                    'ac_id': agent_info.id,
                    'company_name': agent_info.company_name,
                    'phone': device_user_qs.phone,
                    'created_time': agent_info.created_time,
                    'service_packages': list(service_packages),
                }
            }

            if card_details:
                # 获取银行卡号
                card_no = card_details.card_no
                # 检查银行卡号字符串长度，如果小于等于8，则返回"null"
                if len(card_no) <= 8:
                    masked_card_no = "null"
                else:
                    # 保留前四位和后四位，用"*"代替中间的数字
                    masked_card_no = card_no[:4] + '*' * (len(card_no) - 8) + card_no[-4:]

                result['agent_info'].update({
                    'card_name': card_details.name,
                    'card_no': masked_card_no,
                    'card_address': card_details.card_address,
                })

            return response.json(0, result)
        except Exception as e:
            return response.json(500, {'error': str(e)})

    def get_agent_service_package(self, request_dict, response):
        """
        查询所有代理云服务套餐
        @param request_dict: 请求参数
        @request_dict page: 页码
        @request_dict page_size: 查询分页数
        @param response: 响应对象
        @return:
        """
        page = int(request_dict.get('page', 1))
        page_size = int(request_dict.get('page_size', 10))
        try:
            # 查询所有有效的代理云服务套餐
            all_packages = AgentCloudServicePackage.objects.filter(status=1).order_by('type', '-created_time')
            # 创建分页对象
            paginator = Paginator(all_packages, page_size)
            # 获取请求页的数据
            packages_page = paginator.page(page)
            # 准备响应数据，转换查询集为列表形式
            agents_list = []
            for agent_info in packages_page:
                package_name = agent_info.package_id
                if agent_info.type == 1:
                    store_meals = Store_Meal.objects.filter(id=agent_info.package_id).values('bucket__bucket').first()
                    if store_meals:
                        package_name = store_meals['bucket__bucket']
                else:
                    unicom_combos = UnicomCombo.objects.filter(id=agent_info.package_id).first()
                    if unicom_combos:
                        package_name = unicom_combos.combo_name
                agents = {
                    'id': agent_info.id,
                    'service_name': agent_info.service_name,
                    'package_name': package_name,
                    'type': agent_info.type,
                    'profit_type': agent_info.profit_type,
                    'cost': agent_info.cost,
                    'profit': agent_info.profit,
                    'status': agent_info.status,
                    'created_time': agent_info.created_time
                }
                agents_list.append(agents)
            # 返回分页数据
            return response.json(0, {
                'page': page,
                'page_size': page_size,
                'total': paginator.count,
                'num_pages': paginator.num_pages,
                'list': agents_list
            })
        except Exception as e:
            # 出错时返回错误信息
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def add_agent_service_package(self, userID, request_dict, response):
        """
        添加代理云服务套餐
        @param request_dict: 请求参数
        @request_dict package_id: 代理服务id
        @request_dict service_name: 代理服务名称
        @request_dict package_type: 套餐类型 1:云存,2:4G
        @response_dict profit_type: 利润分配类型 1:固定值,2:百分比
        @response_dict cost: 成本
        @response_dict profit: 利润值
        @param response: 响应对象
        @return:
        """
        package_id = request_dict.get('package_id', None)
        service_name = request_dict.get('service_name', None)
        package_type = int(request_dict.get('package_type', 0))  # 默认为0，确保类型安全
        profit_type = int(request_dict.get('profit_type', 1))  # 默认值为1
        profit = request_dict.get('profit', 0)
        cost = request_dict.get('cost', 0)

        try:
            # 创建AgentCloudServicePackage实例并保存
            if not all([package_id, service_name]):
                return response.json(444)
            if package_type == 1:
                query_set = Store_Meal.objects.filter(is_show=0, id=package_id)
            elif package_type == 2:
                query_set = UnicomCombo.objects.filter(is_show=1, is_del=False, id=package_id)
            else:
                return response.json(444, 'error package_type')
            if not query_set.exists():
                return response.json(173)
            AgentCloudServicePackage.objects.create(
                service_name=service_name,
                package_id=package_id,
                type=package_type,
                profit_type=profit_type,
                profit=profit,
                status=1,
                cost=cost,
                created_by=userID,
                created_time=int(time.time()),
                updated_by=userID,
                updated_time=int(time.time())
            )
            return response.json(0)
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def update_agent_service_package(self, userID, request_dict, response):
        """
        更新代理云服务套餐
        @param request_dict: 请求参数
        @request_dict package_id: 代理服务id
        @request_dict service_name: 代理服务名称
        @response_dict profit_type: 利润分配类型 1:固定值,2:百分比
        @response_dict cost: 成本
        @response_dict profit: 利润值
        @param response: 响应对象
        @return:
        """
        id = request_dict.get('id', None)
        service_name = request_dict.get('service_name', None)
        profit_type = request_dict.get('profit_type', None)
        cost = request_dict.get('cost', None)
        profit = request_dict.get('profit', None)

        if not all([id, service_name, profit_type, cost, profit]):
            return response.json(444)

        try:
            ac_service_package = AgentCloudServicePackage.objects.get(pk=id)
            ac_service_package.service_name = service_name
            ac_service_package.profit_type = profit_type
            ac_service_package.cost = cost
            ac_service_package.profit = profit
            ac_service_package.updated_time = int(time.time())
            ac_service_package.updated_by = userID
            ac_service_package.save()
            return response.json(0)

        except AgentCloudServicePackage.DoesNotExist:
            return response.json(173)
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def del_agent_service_package(self, request_dict, response):
        """
        删除代理云服务套餐
        @param userID: 用户ID（执行删除操作的用户）
        @param request_dict: 请求参数
        @request_dict package_id: 代理服务id
        @param response: 响应对象
        @return:
        """
        id = request_dict.get('id', None)
        if not id:
            return response.json(444, 'Missing package_id')
        try:
            ac_service_package = AgentCloudServicePackage.objects.get(pk=id)
            # 假删除，否则查利润会出问题
            ac_service_package.status = 0
            ac_service_package.save()
            return response.json(0)

        except AgentCloudServicePackage.DoesNotExist:
            return response.json(173, 'Package does not exist.')

        except Exception as e:
            return response.json(500, {'error': str(e)})

    def get_customer_list(self, request_dict, response):
        """
        查询代理商信息，并进行分页处理
        @param request_dict: 请求对象，用于接收分页参数
        @param response: 响应对象
        @return:
        """
        # 接收分页参数
        page = int(request_dict.get('page', 1))
        page_size = int(request_dict.get('page_size', 10))
        search_query = request_dict.get('search_query', '')
        user_id = request_dict.get('user_id', None)
        phone = request_dict.get('phone', None)
        userEmail = request_dict.get('userEmail', None)

        try:
            # 基础查询条件
            agent_infos_query = AgentCustomerInfo.objects.filter(status=1)

            # 精确查询条件
            if user_id:
                agent_infos_query = agent_infos_query.filter(user_id=user_id)

            # 获取所有符合条件的AgentCustomerInfo记录
            agent_infos = agent_infos_query.values('id', 'user_id', 'company_name').order_by('id')

            # 对结果进行分页
            paginator = Paginator(agent_infos, page_size)
            packages_page = paginator.page(page)

            # 准备最终的代理商列表
            agents_list = []
            for agent_info in packages_page:
                # 模糊查询
                if search_query:
                    card_info_query = AgentCustomerCard.objects.filter(ac_id=agent_info['id'],
                                                                       name__icontains=search_query)
                    if not card_info_query.exists() and search_query not in agent_info['company_name']:
                        continue
                else:
                    card_info_query = AgentCustomerCard.objects.filter(ac_id=agent_info['id'])

                # 获取卡信息
                card_info = card_info_query.values('name', 'card_no', 'card_address').first()

                # 获取用户信息，根据需要进行精确查询
                user_info_query = Device_User.objects.filter(userID=agent_info['user_id'])
                if phone:
                    user_info_query = user_info_query.filter(phone=phone)
                if userEmail:
                    user_info_query = user_info_query.filter(userEmail=userEmail)

                user_info = user_info_query.values('phone', 'userEmail').first()

                # 组合信息
                agent_record = {
                    'id': agent_info['id'],
                    'company_name': agent_info['company_name'],
                    'user_id': agent_info['user_id'],
                    'phone': user_info.get('phone') if user_info else None,
                    'userEmail': user_info.get('userEmail') if user_info else None,
                    'card_name': card_info.get('name') if card_info else None,
                    'card_no': card_info.get('card_no') if card_info else None,
                    'card_address': card_info.get('card_address') if card_info else None,
                }
                agents_list.append(agent_record)

            # 返回分页后的结果
            return response.json(0, {
                'page': page,
                'page_size': page_size,
                'total': paginator.count,
                'num_pages': paginator.num_pages,
                'list': agents_list
            })
        except Exception as e:
            print(e)
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def get_cumstomer_package_list(self, request_dict, response):
        """
        查询代理商服务套餐列表
        @param response: 响应对象
        @return:
        """
        ac_id = request_dict.get('id', None)
        try:
            if ac_id is None:
                return response.json(444, 'Missing ac_id')
            # ac_id查询AgentCustomerPackage来找到关联的云服务套餐ID（cs_id）
            package_ids = AgentCustomerPackage.objects.filter(ac_id=ac_id).values_list('cs_id', flat=True)
            # cs_id查询AgentCloudServicePackage获取服务套餐详情。
            service_packages = AgentCloudServicePackage.objects.filter(id__in=package_ids, status=1).values(
                'id', 'service_name', 'type'
            )
            service_packages_list = list(service_packages)
            return response.json(0, {'service_packages': service_packages_list})

        except Exception as e:
            print(e)
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def batch_rebind_customer_packages(self, userID, request_dict, response):
        """
        清空并重新绑定代理商服务套餐
        @param userID: 操作用户的ID
        @param request_dict: 请求参数，包含代理商ID和新的服务套餐ID列表
        @param response: 响应对象
        @return:
        """
        ac_id = request_dict.get('ac_id', None)  # 代理客户ID
        new_cs_ids = json.loads(request_dict.get('cs_ids', '[]'))  # 新的服务套餐ID列表

        if not ac_id:
            return response.json(444, 'Missing agent customer ID.')
        if not new_cs_ids:
            return response.json(444, 'Service package IDs are required.')

        try:
            # 删除该代理客户的所有现有绑定
            AgentCustomerPackage.objects.filter(ac_id=ac_id).delete()

            # 过滤出存在且状态为有效的套餐ID
            valid_new_cs_ids = AgentCloudServicePackage.objects.filter(id__in=new_cs_ids, status=1).values_list('id',
                                                                                                                flat=True)
            # 准备批量创建的数据
            packages_to_bind = [
                AgentCustomerPackage(
                    ac_id=ac_id,
                    cs_id=cs_id,
                    created_by=userID,
                    updated_by=userID,
                    created_time=int(time.time()),
                    updated_time=int(time.time())
                )
                for cs_id in valid_new_cs_ids
            ]

            # 批量创建新的绑定关系
            AgentCustomerPackage.objects.bulk_create(packages_to_bind)

            return response.json(0)
        except Exception as e:
            print(e)
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def get_agent_service_package_list(self, response):
        """
        查询云服务套餐列表 id + service_name
        @param response: 响应对象
        @return:
        """
        try:
            # 查询所有有效的代理云服务套餐
            all_packages = AgentCloudServicePackage.objects.filter(status=1).order_by('-created_time').values('id',
                                                                                                              'service_name')
            # 转换查询集为列表形式
            packages_list = list(all_packages)
            # 返回数据
            return response.json(0, {'packages': packages_list})
        except Exception as e:
            # 出错时返回错误信息
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def get_agent_settle_order(self, userID, request_dict, response):
        """
        查询结算明细
        @param userID: userID
        @param request_dict: 请求参数
        @param request_dict status: 结算状态
        @param request_dict time: 年-季度 2023-1
        @param response: 响应对象
        @return:
        """

        status = request_dict.get('status', None)
        time_str = request_dict.get('time', None)
        package_type = request_dict.get('package_type', None)
        startTime = int(request_dict.get('start_time', 1))
        endTime = int(request_dict.get('end_time', 0))

        page = int(request_dict.get('page', 1))  # 默认为第一页
        page_size = int(request_dict.get('page_size', 10))  # 默认每页10条记录

        try:
            agent_customer_info = AgentCustomerInfo.objects.filter(user_id=userID).first()
            if agent_customer_info is None:
                agent_device_orders_qs = AgentDeviceOrder.objects.filter(is_del=False)
            else:
                ac_id = agent_customer_info.id
                agent_device_orders_qs = AgentDeviceOrder.objects.filter(ac_id=ac_id, is_del=False)

            if time_str:
                year, quarter = map(int, time_str.split('-'))
                start_month = 3 * (quarter - 1) + 1
                end_month = start_month + 2

                start_date = datetime(year, start_month, 1)
                end_date = datetime(year + 1, 1, 2) if end_month == 12 else datetime(year, end_month + 1, 1)

                start_time = int(start_date.timestamp())
                end_time = int(end_date.timestamp()) - 1
                agent_device_orders_qs = agent_device_orders_qs.filter(created_time__gte=start_time,
                                                                       created_time__lte=end_time)
            if startTime < endTime:
                agent_device_orders_qs = agent_device_orders_qs.filter(created_time__gte=startTime,
                                                                       created_time__lte=endTime)
            if package_type:
                csp_ids = list(
                    AgentCloudServicePackage.objects.filter(type=int(package_type)).values_list('id', flat=True))
                agent_device_orders_qs = agent_device_orders_qs.filter(csp_id__in=csp_ids)

            if status is None:
                total_profit = agent_device_orders_qs.aggregate(Sum('profit'))['profit__sum'] or 0
            else:
                # 计算特定状态的device_orders总额
                full_profit = agent_device_orders_qs.filter(status=status).aggregate(Sum('profit'))[
                                  'profit__sum'] or 0

                # 初始化total_profit
                total_profit = full_profit

                # 对于状态1和2，计算部分结算的利润
                if status in ["1", "2"]:
                    partial_settled_profit = AgentDeviceOrderInstallment.objects.filter(
                        ado_id__in=agent_device_orders_qs.filter(status=1).values_list('id', flat=True),
                        status=2
                    ).aggregate(total=Sum('amount'))['total'] or 0

                    # 根据状态调整total_profit
                    if status == "1":
                        total_profit -= partial_settled_profit
                    else:  # 当status为"2"时，添加partial_settled_profit（根据业务逻辑，这可能需要调整）
                        total_profit += partial_settled_profit

                agent_device_orders_qs = agent_device_orders_qs.filter(status=status)

            # 应用分页
            agent_device_orders_qs = agent_device_orders_qs.order_by('-created_time')
            paginator = Paginator(agent_device_orders_qs, page_size)
            current_page = paginator.get_page(page)

            orders = []
            for order in current_page:
                csp = AgentCloudServicePackage.objects.filter(id=order.csp_id).first()
                service_name = csp.service_name if csp else order.csp_id
                agent_customer_info = AgentCustomerInfo.objects.filter(id=order.ac_id).first()
                company_name = agent_customer_info.company_name if agent_customer_info else order.ac_id
                ado_id = order.id
                settled_profit = \
                    AgentDeviceOrderInstallment.objects.filter(ado_id=ado_id, status=2).aggregate(Sum('amount'))[
                        'amount__sum'] or 0
                orders.append({
                    'id': order.id,
                    'company_name': company_name,
                    'serial_number': order.serial_number,
                    'status': order.status,
                    'service_name': service_name,
                    'profit_amount': order.profit_amount,
                    'profit': order.profit,
                    'settled_profit': settled_profit,
                    'settlement_time': order.settlement_time,
                    'remark': order.remark,
                    'created_time': order.created_time
                })

            response_data = {
                'list': orders,
                'total_profit': total_profit,
                'total': paginator.count,
                'page': current_page.number,
                'page_size': page_size,
                'num_pages': paginator.num_pages,
            }

            return response.json(0, response_data)

        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def get_agent_account_withdraw(self, userID, request_dict, response):
        """
        查询提现明细
        @param userID: userID
        @param request_dict: 请求参数
        @param response: 响应对象
        @return:
        """
        agent_customer_info = AgentCustomerInfo.objects.filter(user_id=userID).first()
        if not agent_customer_info:
            return response.json(104, 'Agent customer not found')

        ac_id = agent_customer_info.id
        try:
            agent_account_withdraw_qs = AgentAccountWithdraw.objects.filter(ac_id=ac_id).order_by('-updated_time')

            page = int(request_dict.get('page', 1))  # 默认为第一页
            page_size = int(request_dict.get('page_size', 10))  # 默认每页10条记录

            # 应用分页
            paginator = Paginator(agent_account_withdraw_qs, page_size)
            current_page = paginator.get_page(page)

            withdraw_list = []

            for withdraw in current_page:
                agent_customer_card = AgentCustomerCard.objects.filter(ac_id=ac_id).first()
                card_no = agent_customer_card.card_no
                masked_card_no = card_no[:4] + '*' * (len(card_no) - 8) + card_no[-4:]
                withdraw_list.append({
                    'id': withdraw.id,
                    'amount': withdraw.amount,
                    'created_time': withdraw.created_time,
                    'card_no': masked_card_no,
                    'status': withdraw.status,
                    'remark': withdraw.remark,
                    'arrival_time': withdraw.arrival_time
                })

            response_data = {
                'list': withdraw_list,
                'total': paginator.count,
                'page': current_page.number,
                'page_size': page_size,
                'num_pages': paginator.num_pages,
            }
            return response.json(0, response_data)

        except Exception as e:
            # 出错时返回错误信息
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def get_check_balance(self, userID, response):
        """
        查询余额
        @param userID: userID
        @param response: 响应对象
        @return:
        """

        agent_customer_info = AgentCustomerInfo.objects.filter(user_id=userID).first()
        if not agent_customer_info:
            return response.json(104, 'Agent customer not found')

        ac_id = agent_customer_info.id

        try:
            # 计算冻结金额
            frozen_amount_qs = AgentAccountWithdraw.objects.filter(ac_id=ac_id, status__in=[1, 2, 3])
            frozen_amount = frozen_amount_qs.aggregate(total=Sum('amount'))['total'] or Decimal('0.00')

            # 计算余额：已结算 - (退款+已打款)
            incomes_qs = AgentAccount.objects.filter(ac_id=ac_id, status=1)
            expense_qs = AgentAccount.objects.filter(ac_id=ac_id, status__in=[2, 3])
            incomes_all = incomes_qs.aggregate(total=Sum('amount'))['total'] or Decimal('0.00')
            expense_all = expense_qs.aggregate(total=Sum('amount'))['total'] or Decimal('0.00')
            total_profit = incomes_all - expense_all

            # 可用余额 = 总余额 - 冻结金额
            available_balance = total_profit - frozen_amount

            # 构造返回数据
            balance_data = {
                'frozen_amount': frozen_amount,  # 冻结金额
                'total_profit': total_profit,  # 总余额
                'available_balance': available_balance  # 可用余额
            }

            return response.json(0, balance_data)

        except Exception as e:
            # 出错时返回错误信息
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def get_data_statistics(self, userID, response):
        """
        首页总览
        @param userID: userID
        @param response: 响应对象
        @return:
        """
        agent_customer_info = AgentCustomerInfo.objects.filter(user_id=userID).first()
        if agent_customer_info is None:
            agent_device_orders_qs = AgentDeviceOrder.objects.filter(is_del=False)
            agent_device_qs = AgentDevice.objects.filter()
        else:
            ac_id = agent_customer_info.id
            agent_device_orders_qs = AgentDeviceOrder.objects.filter(ac_id=ac_id, is_del=False)
            agent_device_qs = AgentDevice.objects.filter(ac_id=ac_id)

        now = datetime.now()
        today_start = datetime(now.year, now.month, now.day)
        yesterday_start = today_start - timedelta(days=1)
        tomorrow_start = today_start + timedelta(days=1)

        try:
            # 计算AgentDeviceOrderInstallment表 结算金额
            partial_settled_profit = AgentDeviceOrderInstallment.objects.filter(
                ado_id__in=agent_device_orders_qs.filter(status=1).values_list('id', flat=True),
                status=2
            ).aggregate(total=Sum('amount'))['total'] or 0

            # 总利润
            total_profit_all = \
                agent_device_orders_qs.filter(status__in=[1, 2], is_del=False).aggregate(
                    total=Sum('profit'))[
                    'total'] or Decimal('0.00')
            total_profit_no = \
                agent_device_orders_qs.filter(status=1, is_del=False).aggregate(total=Sum('profit'))[
                    'total'] or Decimal('0.00')
            total_profit_no = total_profit_no - partial_settled_profit

            total_profit_yes = \
                agent_device_orders_qs.filter(status=2, is_del=False).aggregate(total=Sum('profit'))[
                    'total'] or Decimal('0.00')
            total_profit_yes = total_profit_yes + partial_settled_profit

            # 总营业额
            profit_amount_all = agent_device_orders_qs.filter(status__in=[1, 2], is_del=False).aggregate(
                total=Sum('profit_amount'))['total'] or Decimal('0.00')

            # 今日总营业额
            profit_amount_today_all = agent_device_orders_qs.filter(
                status__in=[1, 2],
                is_del=False,
                created_time__gte=int(today_start.timestamp()),
                created_time__lt=int(tomorrow_start.timestamp())
            ).aggregate(total=Sum('profit_amount'))['total'] or Decimal('0.00')

            # 昨日总营业额
            profit_amount_yesterday_all = agent_device_orders_qs.filter(
                status__in=[1, 2],
                is_del=False,
                created_time__gte=int(yesterday_start.timestamp()),
                created_time__lt=int(today_start.timestamp())
            ).aggregate(total=Sum('profit_amount'))['total'] or Decimal('0.00')

            # 激活设备数
            active_device_count_today = agent_device_qs.filter(
                status=1,
                created_time__gte=int(today_start.timestamp()),
                created_time__lt=int(tomorrow_start.timestamp())
            ).count()
            active_device_count_yesterday = agent_device_qs.filter(
                status=1,
                created_time__gte=int(yesterday_start.timestamp()),
                created_time__lt=int(today_start.timestamp())
            ).count()
            # 总设备数
            active_device_count = agent_device_qs.filter(status=1).count()
            inactive_device_count = agent_device_qs.filter(status=0).count()

            return response.json(0, {
                'total_profit_all': total_profit_all,
                'total_profit_no': total_profit_no,
                'total_profit_yes': total_profit_yes,
                'profit_amount_all': profit_amount_all,
                'profit_amount_today_all': profit_amount_today_all,
                'profit_amount_yesterday_all': profit_amount_yesterday_all,
                'active_device_count_today': active_device_count_today,
                'active_device_count_yesterday': active_device_count_yesterday,
                'active_device_count': active_device_count,
                'inactive_device_count': inactive_device_count,
                'all_device_count': active_device_count + inactive_device_count
            })
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def agent_apply_withdraw(self, userID, request_dict, response):
        """
        用户提现申请
        @param userID: userID
        @param request_dict: 请求参数
        @param request_dict amount: 金额
        @param response: 响应对象
        @return:
        """
        amount = Decimal(request_dict.get('amount', '0.00'))

        if amount == Decimal('0.00'):
            return response.json(444)

        agent_customer_info = AgentCustomerInfo.objects.filter(user_id=userID).first()
        if not agent_customer_info:
            return response.json(104, 'Agent customer not found')

        ac_id = agent_customer_info.id
        try:
            # 计算余额：已结算 - (退款+已打款)
            incomes_qs = AgentAccount.objects.filter(ac_id=ac_id, status=1)
            expense_qs = AgentAccount.objects.filter(ac_id=ac_id, status__in=[2, 3])
            incomes_all = incomes_qs.aggregate(total=Sum('amount'))['total'] or Decimal('0.00')
            expense_all = expense_qs.aggregate(total=Sum('amount'))['total'] or Decimal('0.00')
            total_profit = incomes_all - expense_all

            # 冻结余额
            frozen_amount_qs = AgentAccountWithdraw.objects.filter(ac_id=ac_id, status__in=[1, 2, 3])
            frozen_amount = frozen_amount_qs.aggregate(total=Sum('amount'))['total'] or Decimal('0.00')

            # 可提现余额 = 总余额 - 冻结金额
            available_balance = total_profit - frozen_amount

            if amount < Decimal('1000.00'):
                return response.json(10, '每次提现最少为1000元')
            if amount > available_balance:
                return response.json(10, '余额不足，无法提现')

            # 提交提现申请
            acc = AgentCustomerCard.objects.filter(ac_id=ac_id).first()
            AgentAccountWithdraw.objects.create(ac_id=ac_id, status=1, card_id=acc.id, amount=amount,
                                                created_time=int(time.time()), updated_time=int(time.time()))
            return response.json(0)

        except Exception as e:
            print(e)
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def get_withdrawal_review(self, request_dict, response):
        """
        后台用户提现申请审核列表
        @param request_dict: 请求参数
        @param request_dict company_name: 公司名称
        @param request_dict status: 审核状态
        @param response: 响应对象
        @return:
        """
        company_name = request_dict.get('company_name', None)
        status = request_dict.get('status', None)
        page = int(request_dict.get('page', 1))  # 默认为第一页
        page_size = int(request_dict.get('page_size', 10))  # 默认每页10条记录
        try:
            agent_account_withdraw_qs = AgentAccountWithdraw.objects.filter().order_by('updated_time')
            if company_name:
                agent_customer_info = AgentCustomerInfo.objects.filter(company_name=company_name).first()
                ac_id = agent_customer_info.id
                agent_account_withdraw_qs = agent_account_withdraw_qs.filter(ac_id=ac_id)

            if status:
                agent_account_withdraw_qs = agent_account_withdraw_qs.filter(status=status)

            # 提现id 公司名 审核状态 提现金额 余额
            paginator = Paginator(agent_account_withdraw_qs, page_size)
            current_page = paginator.get_page(page)

            review_list = []

            for review in current_page:
                ac_id = review.ac_id
                agent_customer_info = AgentCustomerInfo.objects.filter(id=ac_id).first()
                company_name = agent_customer_info.company_name
                review_list.append({
                    "id": review.id,
                    "company_name": company_name,
                    "status": review.status,
                    "amount": review.amount,
                    "remark": review.remark,
                })

            response_data = {
                'list': review_list,
                'total': paginator.count,
                'page': current_page.number,
                'page_size': page_size,
                'num_pages': paginator.num_pages,
            }
            return response.json(0, response_data)
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def update_withdrawal_review(self, userID, request_dict, response):
        """
        后台提现审核
        @param request_dict: 请求参数
        @param request_dict id: 提现单id
        @param request_dict status: 提现单状态
        @param request_dict remark: 提现单备注
        @param response: 响应对象
        @return:
        """
        id = request_dict.get('id', None)
        status = request_dict.get('status', None)
        remark = request_dict.get('remark', "")

        if not all([id, status]):
            return response.json(444)

        try:
            agent_account_withdraw = AgentAccountWithdraw.objects.get(pk=id)
            agent_account_withdraw.status = status
            agent_account_withdraw.remark = remark
            agent_account_withdraw.save()
            if int(status) == 4:
                AgentAccount.objects.create(ac_id=agent_account_withdraw.ac_id,
                                            amount=agent_account_withdraw.amount,
                                            status=3,
                                            created_time=int(time.time()),
                                            updated_time=int(time.time()),
                                            remark=f"{userID}修改为已提现")
            return response.json(0)

        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def apply_agent(self, request_dict, response):
        """
        代理申请
        @param request_dict: 请求参数
        @param request_dict name: 名字
        @param request_dict phone: 电话
        @param request_dict regin: 地区
        @param request_dict remark: 备注
        @param response: 响应对象
        @return:
        """
        name = request_dict.get('name', None)
        phone = request_dict.get('phone', None)
        regin = request_dict.get('regin', None)
        remark = request_dict.get('remark', "")
        if not all([name, phone, regin]):
            return response.json(444)

        # 去除非数字字符
        clean_phone = re.sub(r'\D', '', phone)

        if ApplyAgent.objects.filter(phone=clean_phone).exists():
            return response.json(174, 'Phone number already exists')

        ApplyAgent.objects.create(name=name, phone=clean_phone, regin=regin, remark=remark)

        return response.json(0, '申请已提交')