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 time

from django.http import QueryDict
from django.views import View
from django.core.paginator import Paginator
from datetime import datetime

from AgentModel.models import AgentCustomerInfo, AgentCustomerCard, AgentCustomerPackage, AgentCloudServicePackage, \
    AgentDeviceOrder
from Model.models import UnicomCombo, Store_Meal, Device_User

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(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 == 'getAgentServicePackage':
                return self.get_agent_service_package(request_dict, response)
            elif operation == 'addAgentServicePackage':
                return self.add_agent_service_package(request_dict, response)
            else:
                return response.json(444, 'operation')

    def get_unicom_and_icloud(self, response):
        """
        查询云存储套餐和物联网卡套餐列表
        @param response: 响应对象
        @return:
        """
        try:
            # 云存储套餐名称查询，只包括is_show=1的记录
            store_meal_names = Store_Meal.objects.filter(is_show=1).values_list('bucket__bucket', flat=True).distinct()

            # 联通套餐名称查询，只包括is_show=1且未被删除的记录
            unicom_combo_names = UnicomCombo.objects.filter(is_show=1, is_del=False).values_list('combo_name',
                                                                                                 flat=True).distinct()
            # 将查询结果转换为列表
            store_meal_name_list = list(store_meal_names)
            unicom_combo_name_list = list(unicom_combo_names)

            # 合并结果并返回
            return response.json(0, {
                'storeMealNames': store_meal_name_list,
                'unicomComboNames': unicom_combo_name_list,
            })
        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(self, request_dict, response):
        """
        查询所有代理云服务套餐
        @param request_dict: 请求参数
        @request_dict page: 页码
        @request_dict page_size: 查询分页数
        @param userID: userID
        @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('-created_time')
            # 创建分页对象
            paginator = Paginator(all_packages, page_size)
            # 获取请求页的数据
            packages_page = paginator.page(page)
            # 准备响应数据，转换查询集为列表形式
            packages_list = list(packages_page.object_list.values(
                'id', 'service_name', 'package_id', 'type',
                'profit_type', 'cost', 'profit', 'status',
                'created_time', 'updated_time'
            ))
            # 返回分页数据
            return response.json(0, {
                'page': page,
                'page_size': page_size,
                'total': paginator.count,
                'num_pages': paginator.num_pages,
                'packages': packages_list
            })
        except Exception as e:
            # 出错时返回错误信息
            return response.json({'error': str(e)}, status=500)

    def add_agent_service_package(self, request_dict, response):
        """
        添加代理云服务套餐
        @param request_dict: 请求参数
        @request_dict service_name: 代理服务名称
        @request_dict package_type: 套餐类型 1:云存,2:4G
        @response_dict profit_type: 利润分配类型 1:固定值,2:百分比
        @response_dict cost: 成本
        @response_dict profit: 利润值
        @param userID: userID
        @param response: 响应对象
        @return:
        """
        service_name = request_dict.get('service_name')
        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)

        try:
            # 1:云存, 2:4G
            print(package_type)
            if package_type == 1:
                # 从Store_Meal查询
                package = Store_Meal.objects.filter(bucket__bucket=service_name, is_show=1).first()
            elif package_type == 2:
                # 从UnicomCombo查询
                package = UnicomCombo.objects.filter(combo_name=service_name, is_show=1, is_del=False).first()
            else:
                return response.json(444)
            if not package:
                return response.json(173)
            # 创建AgentCloudServicePackage实例并保存
            agent_service_package = AgentCloudServicePackage(
                service_name=service_name,
                package_id=package.id if package else None,
                type=package_type,
                profit_type=profit_type,
                cost=package.price,
                profit=profit,
                status=1,
                created_time=int(time.time()),
                updated_time=int(time.time())
            )
            agent_service_package.save()
            return response.json(0)
        except Exception as e:
            return response.json({'error': str(e)}, status=500)