Browse Source

开通云存 默认开通云盘

peng 2 years ago
parent
commit
75250a6973
1 changed files with 46 additions and 10 deletions
  1. 46 10
      Controller/CloudStorage.py

+ 46 - 10
Controller/CloudStorage.py

@@ -929,8 +929,6 @@ class CloudStorageView(View):
                 if not icloud_meal_qs.exists():
                     return response.json(173)
                 size = icloud_meal_qs[0]['size']
-                icloud_use_qs = IcloudUseDetails.objects.filter(user_id=userid).values('id')
-                use_details_id = icloud_use_qs[0]['id']
                 if order_list[0]['isSelectDiscounts'] == 1:
                     expire = store_qs[0]['expire'] * 2
                 # 是否有促销
@@ -1000,6 +998,17 @@ class CloudStorageView(View):
                             }
                             AiService.objects.create(**ai_service_dict)
                     # 开通云盘服务
+                    icloud_use_qs = IcloudUseDetails.objects.filter(user_id=userid).values('id')
+                    if not icloud_use_qs.exists():
+                        bucket_qs = VodBucketModel.objects.filter(bucket='icloud').values('id')
+                        icloud_use_qs = IcloudUseDetails.objects.create(add_time=now_time, upd_time=now_time,
+                                                                        detect_status=1, user_id=userid,
+                                                                        bucket_id=bucket_qs[0]['id'])
+                        use_details_id = icloud_use_qs.id
+                        IcloudService.objects.create(add_time=now_time, upd_time=now_time, type=0,
+                                                     size=1, use_details_id=use_details_id)
+                    else:
+                        use_details_id = icloud_use_qs[0]['id']
                     IcloudService.objects.create(orders_id=order_id, add_time=now_time, upd_time=now_time, type=1,
                                                  size=size, end_time=end_time, use_details_id=use_details_id)
                     # 如果存在序列号,消息提示用序列号
@@ -1110,9 +1119,6 @@ class CloudStorageView(View):
             if not icloud_meal_qs.exists():
                 return response.json(173)
             size = icloud_meal_qs[0]['size']
-            icloud_use_qs = IcloudUseDetails.objects.filter(user_id=userid).values('id')
-            use_details_id = icloud_use_qs[0]['id']
-
             if order_list[0]['isSelectDiscounts'] == 1:
                 expire = store_qs[0]['expire'] * 2
             # 是否有促销
@@ -1181,6 +1187,17 @@ class CloudStorageView(View):
                         }
                         AiService.objects.create(**ai_service_dict)
                 # 开通云盘服务
+                icloud_use_qs = IcloudUseDetails.objects.filter(user_id=userid).values('id')
+                if not icloud_use_qs.exists():
+                    bucket_qs = VodBucketModel.objects.filter(bucket='icloud').values('id')
+                    icloud_use_qs = IcloudUseDetails.objects.create(add_time=now_time, upd_time=now_time,
+                                                                    detect_status=1, user_id=userid,
+                                                                    bucket_id=bucket_qs[0]['id'])
+                    use_details_id = icloud_use_qs.id
+                    IcloudService.objects.create(add_time=now_time, upd_time=now_time, type=0,
+                                                 size=1, use_details_id=use_details_id)
+                else:
+                    use_details_id = icloud_use_qs[0]['id']
                 IcloudService.objects.create(orders_id=order_id, add_time=now_time, upd_time=now_time, type=1,
                                              size=size, end_time=end_time, use_details_id=use_details_id)
                 # 如果存在序列号,消息提示用序列号
@@ -1273,8 +1290,6 @@ class CloudStorageView(View):
                 if not icloud_meal_qs.exists():
                     return response.json(173)
                 size = icloud_meal_qs[0]['size']
-                icloud_use_qs = IcloudUseDetails.objects.filter(user_id=userid).values('id')
-                use_details_id = icloud_use_qs[0]['id']
                 if order_list[0]['isSelectDiscounts'] == 1:
                     expire = store_qs[0]['expire'] * 2
                 # 是否有促销
@@ -1344,6 +1359,17 @@ class CloudStorageView(View):
                             }
                             AiService.objects.create(**ai_service_dict)
                     # 开通云盘服务
+                    icloud_use_qs = IcloudUseDetails.objects.filter(user_id=userid).values('id')
+                    if not icloud_use_qs.exists():
+                        bucket_qs = VodBucketModel.objects.filter(bucket='icloud').values('id')
+                        icloud_use_qs = IcloudUseDetails.objects.create(add_time=now_time, upd_time=now_time,
+                                                                        detect_status=1, user_id=userid,
+                                                                        bucket_id=bucket_qs[0]['id'])
+                        use_details_id = icloud_use_qs.id
+                        IcloudService.objects.create(add_time=now_time, upd_time=now_time, type=0,
+                                                     size=1, use_details_id=use_details_id)
+                    else:
+                        use_details_id = icloud_use_qs[0]['id']
                     IcloudService.objects.create(orders_id=order_id, add_time=now_time, upd_time=now_time, type=1,
                                                  size=size, end_time=end_time, use_details_id=use_details_id)
                     # 如果存在序列号,消息提示用序列号
@@ -1669,10 +1695,8 @@ class CloudStorageView(View):
         icloud_store_meal_id = store_qs[0]['icloud_store_meal_id']
         icloud_meal_qs = ICloudStoreMeal.objects.filter(id=icloud_store_meal_id).values('size')
         if not icloud_meal_qs.exists():
-            return response.json(911)
+            return response.json(173)
         size = icloud_meal_qs[0]['size']
-        icloud_use_qs = IcloudUseDetails.objects.filter(user_id=user_id).values('id')
-        use_details_id = icloud_use_qs[0]['id']
         try:
             with transaction.atomic():
                 if uid_bucket_qs.exists():
@@ -1708,8 +1732,20 @@ class CloudStorageView(View):
                         AiService.objects.create(uid=uid, channel=channel, detect_status=1, endTime=end_time,
                                                  addTime=now_time, updTime=now_time, use_status=1)
                 #  开通云盘体验
+                icloud_use_qs = IcloudUseDetails.objects.filter(user_id=user_id).values('id')
+                if not icloud_use_qs.exists():
+                    bucket_qs = VodBucketModel.objects.filter(bucket='icloud').values('id')
+                    icloud_use_qs = IcloudUseDetails.objects.create(add_time=now_time, upd_time=now_time,
+                                                                    detect_status=1, user_id=user_id,
+                                                                    bucket_id=bucket_qs[0]['id'])
+                    use_details_id = icloud_use_qs.id
+                    IcloudService.objects.create(add_time=now_time, upd_time=now_time, type=0,
+                                                 size=1, use_details_id=use_details_id)
+                else:
+                    use_details_id = icloud_use_qs[0]['id']
                 IcloudService.objects.create(orders_id=order_id, add_time=now_time, upd_time=now_time, type=1,
                                              size=size, end_time=end_time, use_details_id=use_details_id)
+
                 store_meal_qs = Store_Meal.objects.filter(id=rank, lang__lang='cn', is_show=0).values('lang__title',
                                                                                                       'lang__content')
                 if store_meal_qs.exists():