添加直播中课程支付

app/service/order_service.py

@@ -37,6 +37,8 @@ course_rollback_api = "http://s-nginx/fatools/pay/course/rollback"
 
 ad_course_rollback_api = "http://s-nginx/course/advertise/detail?id="
 
+zhibo_course_rollback_api = "http://s-nginx/course/new/res/price?userId="
+
 class TopUpOrderService:
     """"""
     def __init__(self):
@@ -268,6 +270,33 @@ class CurriculumOrderService:
                 except:
                     raise CustomFlaskErr(Errors.PROD_NOPRICE)
 
+            elif args['ab_type'] == 3:
+                data = [
+                            {
+                                "resId": args['ab_proid'],
+                                "columnId": None,
+                                "tranResId": None
+                            }
+                        ]
+                headers = {
+                    "Content-Type": "application/json",
+                    # "tampToken": args['tampToken'],
+                    # "env": args['env']
+                }
+                amount = requests.post(zhibo_course_rollback_api + args['user_id'],
+                                    headers = headers,
+                                    json = data,
+                                    timeout = 5)
+                try:
+                    amount = amount.json()
+                    amount = amount[args['ab_proid']]
+                    if amount["finalChargeMode"] == 2:
+                        amount = [int(amount["finalPrice"])]
+                    else:
+                        raise CustomFlaskErr(Errors.PROD_NOPRICE)
+                except:
+                    raise CustomFlaskErr(Errors.PROD_NOPRICE)
+
             elif args['ab_type'] == 300:
                 data = {
                             "payReqs": [{"resId": args['ab_proid']}],