浏览代码

创建代理账号时绑定所有代理套餐

linhaohong 1 年之前
父节点
当前提交
a466a31fbf
共有 1 个文件被更改,包括 12 次插入1 次删除
  1. 12 1
      AdminController/UserManageController.py

+ 12 - 1
AdminController/UserManageController.py

@@ -10,7 +10,7 @@ from django.utils.timezone import utc
 from django.views.decorators.csrf import csrf_exempt
 from django.views.generic import TemplateView
 
-from AgentModel.models import AgentCustomerInfo,AgentCustomerCard
+from AgentModel.models import AgentCustomerInfo, AgentCustomerCard, AgentCloudServicePackage, AgentCustomerPackage
 from Ansjer.config import SERVER_DOMAIN, OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET, AWS_ACCESS_KEY_ID, \
     AWS_SECRET_ACCESS_KEY, AWS_SES_ACCESS_REGION, DETECT_PUSH_DOMAINS
 from Controller.CheckUserData import DataValid, RandomStr
@@ -476,6 +476,17 @@ class UserManagement(View):
                                                      updated_time=int(time.time()), created_by=userID, updated_by=userID)
                     AgentCustomerCard.objects.create(ac_id=agent_customer_info.id, name=card_name, card_no=card_no, card_address=card_address, status=1,
                                                      created_time=int(time.time()), updated_time=int(time.time()), created_by=userID, updated_by=userID)
+                    # 绑定所有代理套餐
+                    all_service_packages = AgentCloudServicePackage.objects.filter(status=1)
+                    for package in all_service_packages:
+                        AgentCustomerPackage.objects.create(
+                            ac_id=agent_customer_info.id,
+                            cs_id=package.id,
+                            created_time=int(time.time()),
+                            updated_time=int(time.time()),
+                            created_by=userID,  # 假设 userID 是创建这个记录的用户ID
+                            updated_by=userID
+                        )
             return response.json(0)
         except Exception as e:
             print(e)