fix bug

app/controller/order.py

@@ -97,6 +97,8 @@ class ConsumeOrder(Resource):
         self.parser.add_argument('ab_pro_siid', type=str, required=False)
         self.parser.add_argument('openid', type=str, required=False)
         self.parser.add_argument('register_phone', type=str, required=False)
+        self.parser.add_argument('ad', type=str, required=False)
+        self.parser.add_argument('ad_id', type=str, required=False)
         args = self.parser.parse_args()
 
         if request.headers.get('env') in ['ios', 'android']:

app/service/order_service.py

@@ -35,6 +35,8 @@ course_price_api = "http://s-nginx/course/res/money/price"
 
 course_rollback_api = "http://s-nginx/fatools/pay/course/rollback"
 
+ad_course_rollback_api = "http://s-nginx/course/advertise/detail?id="
+
 class TopUpOrderService:
     """"""
     def __init__(self):
@@ -257,7 +259,19 @@ class CurriculumOrderService:
                                                                       OrderFlow.ab_status == 'SUCCESS').all()
             if repeat_buy:
                 raise CustomFlaskErr(Errors.REPEAT_BUY)
-            if args['ab_type'] == 300:
+            if args['ab_type'] == 300 and args.get('ad') == 'April':
+                amount = requests.get(ad_course_price_api + args.get('ad_id'),
+                                    timeout = 5)
+                try:
+                    amount = amount.json()
+                    amount = [float(amount["attributes"]["bargainPrice"]) / 100]
+                except:
+                    raise CustomFlaskErr(Errors.PROD_NOPRICE)
+
+                args.pop('ad', '')
+                args.pop('ad_id', '')
+
+            elif args['ab_type'] == 300:
                 data = {
                             "payReqs": [{"resId": args['ab_proid']}],
                             "userId": args['user_id']