优化订单管理功能按钮显示功能

AdminController/ServeManagementController.py

@@ -19,7 +19,7 @@ from Ansjer.config import PAYPAL_CRD
 from Controller.Cron.CronTaskController import CronUpdateDataView
 from Model.models import VodBucketModel, CDKcontextModel, Store_Meal, Order_Model, \
     UID_Bucket, ExperienceContextModel, Lang, CloudLogModel, UidSetModel, Unused_Uid_Meal, \
-    Device_Info, DeviceTypeModel
+    Device_Info, DeviceTypeModel, UnicomComboOrderInfo, AiService
 from Object.ResponseObject import ResponseObject
 from Object.TokenObject import TokenObject
 from Service.CommonService import CommonService
@@ -48,13 +48,14 @@ class serveManagement(View):
         elif operation == 'cloudDataExport':  # 导出流失预警
             return self.cloudDataExport(request, 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
+            # 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
+            userID = ''
             if operation == 'getVodBucketList':
                 return self.getVodBucketList(userID, request_dict, response)
             elif operation == 'addOrEditVodBucket':
@@ -710,14 +711,28 @@ class serveManagement(View):
                     'status': order['status'],
                     'order_type': order['order_type'],
                 }
-                #  添加设备套餐使用状态
-                uid_bucket = UID_Bucket.objects.filter(uid=order['UID']).values('use_status')
-                user_status = uid_bucket[0]['use_status'] if uid_bucket.exists() else ''
-                #  判断订单是否已过期
-                if user_status != '':
-                    data['user_status'] = user_status
-                else:
-                    data['user_status'] = 2
+                #  订单显示(或不显示)停用/退款功能
+                if order['order_type'] == 0: # 云存
+                    uid_bucket = UID_Bucket.objects.filter(uid=order['UID']).values('use_status')
+                    user_status = uid_bucket[0]['use_status'] if uid_bucket.exists() else ''
+                    if user_status != '':
+                        data['user_status'] = user_status
+                    else:
+                        data['user_status'] = 2
+                elif order['order_type'] == 1: # ai
+                    ai_service_qs = AiService.objects.filter(
+                        Q(orders_id=order['orderID']) & ~Q(use_status=2))
+                    if ai_service_qs.exists():
+                        data['user_status'] = 1
+                    else:
+                        data['user_status'] = 2
+                elif order['order_type'] == 2: # 联通4G
+                    unicom_combor_order_qs = UnicomComboOrderInfo.objects.filter(
+                        Q(order_id=order['orderID']) & ~Q(status=2))
+                    if unicom_combor_order_qs.exists():
+                        data['user_status'] = 1
+                    else:
+                        data['user_status'] = 2
                 #  添加PayPal交易号字段
                 data['trade_no'] = 'N/A'
                 if data['payType'] == 1: