1

timer/get_task_to_db.py

@@ -27,6 +27,13 @@ def getPSid(pid):
     res = json.loads(res.text)
     return str(res['data'])
 
+def get_ps_type(pid):
+    # return 1:圆形影棚 2:方形影棚
+    res = requests.get("https://mp.api.suwa3d.com/api/takephotoOrder/photoStudioInfo", params={'pid': pid})
+    return res.json()['data']['type']
+
+
+
 def readTask(key):    
     print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())+"-读取队列-"+'model:'+key)
     if r.llen('model:'+key) == 0:
@@ -39,7 +46,16 @@ def readTask(key):
         print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())+"-读取的PID为空")
         return
     pid = pid.decode('utf-8')
+
+    #判断影棚类型
+    ps_type = get_ps_type(pid)
+    if ps_type == 2:
+        #方形影棚
+        # r.rpush('ai:ai_repeat_build',pid)
+        r.rpush('ai:ai_repeat_for_align_info',pid)
+        return
     psid = getPSid(pid)
+
     #判断是否走新的建模系统 2023-11-10 全部走新的建模系统
     # if libs_db.is_new_make_psid(psid) == False:
     #     #如果不是走新的建模系统就塞回原来的队列