# -*- encoding: utf-8 -*- """ @File : AgentCustomerController.py @Time : 2024/3/7 16:56 @Author : stephen @Email : zhangdongming@asj6.wecom.work @Software: PyCharm """ import time import json 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) elif operation == 'getAgentInfo': return self.get_agent_info(userID, response) else: return response.json(444, 'operation') def get_unicom_and_icloud(self, response): """ 查询云存储套餐和物联网卡套餐列表 @param response: 响应对象 @return: """ try: # 云存储套餐查询，只包括is_show=1的记录 store_meals = Store_Meal.objects.filter(is_show=1).values('id', 'bucket__bucket').distinct() # 联通套餐查询，只包括is_show=1且未被删除的记录 unicom_combos = UnicomCombo.objects.filter(is_show=1, is_del=False).values('id', 'combo_name').distinct() # 将查询结果转换为列表 store_meal_list = [{'id': meal['id'], 'name': meal['bucket__bucket']} for meal in store_meals] unicom_combo_list = [{'id': combo['id'], 'name': combo['combo_name']} for combo in unicom_combos] # 合并结果并返回 return response.json(0, { 'storeMeals': store_meal_list, 'unicomCombos': unicom_combo_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_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: result['agent_info'].update({ 'card_name': card_details.name, 'card_no': card_details.card_no, 'card_address': card_details.card_address, }) return response.json(0, result) except Exception as e: return response.json({'error': str(e)}, status=500) 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)