浏览代码

Merge branch '2024-10-18-luojunhui-redirect-minigram-port' of Server/title_with_video into master

luojunhui 6 月之前
父节点
当前提交
b0de5e1a7d
共有 1 个文件被更改,包括 20 次插入1 次删除
  1. 20 1
      applications/deal/minigram.py

+ 20 - 1
applications/deal/minigram.py

@@ -1,6 +1,8 @@
 """
 @author: luojunhui
 """
+from applications.functions.forward import forward_requests
+
 minigram_map = {
     1: {
         # 25: {
@@ -57,8 +59,10 @@ class Minigram(object):
     """
     小程序卡片
     """
+    NEW_STRATEGY = "strategy_v2"
 
     def __init__(self, params):
+        self.strategy = None
         self.params = params
         self.business_type = None
         self.mini_code = None
@@ -73,6 +77,9 @@ class Minigram(object):
             self.business_type = self.params['businessType']
             self.mini_code = self.params['miniCode']
             self.trace_id = self.params['traceId']
+            self.strategy = self.params.get('strategy')
+            if not self.strategy:
+                self.strategy = "strategy_v1"
             return None
         except Exception as e:
             response = {
@@ -107,4 +114,16 @@ class Minigram(object):
         """
         :return:
         """
-        return self.check_params() if self.check_params() else self.choose_minigram()
+        params_error = self.check_params()
+        if params_error:
+            return params_error
+        else:
+            if self.strategy == self.NEW_STRATEGY:
+                response = await forward_requests(
+                    params=self.params,
+                    api="choose_minigram"
+                )
+                return response
+            else:
+                return self.choose_minigram()
+