增加打印信息

ai/utils/show_process.py

@@ -21,35 +21,35 @@ def show_main_process():
         show_ai = ShowAI(cinema, show_date)
         prompt = show_ai.general_prompt()
         start = datetime.datetime.now()
-        result, message, tokens = show_ai.get_show_result_ai()
+        # result, message, tokens = show_ai.get_show_result_ai()
-        end = datetime.datetime.now()
+        # end = datetime.datetime.now()
-        print('prompt:', prompt)
+        # print('prompt:', prompt)
-        print('result:', result)
+        # print('result:', result)
-        print('message:', message)
+        # print('message:', message)
-        print('tokens:', tokens)
+        # print('tokens:', tokens)
-        # 获取排片数据
+        # # 获取排片数据
-        _show = next((s for s in result.split('------') if s.startswith('\n影厅别名,影厅id')), '')
+        # _show = next((s for s in result.split('------') if s.startswith('\n影厅别名,影厅id')), '')
-        _show = _show.strip()
+        # _show = _show.strip()
-        print('_show:', _show)
+        # print('_show:', _show)
-        # 获取销售额
+        # # 获取销售额
-        _sales = next((s for s in result.split('\n') if re.search('预估销售数据', s)), '')
+        # _sales = next((s for s in result.split('\n') if re.search('预估销售数据', s)), '')
-        _sales = str(re.findall(r'\d+\.?\d*', _sales)[0])
+        # _sales = str(re.findall(r'\d+\.?\d*', _sales)[0])
-        print('_sales:', _sales)
+        # print('_sales:', _sales)
-        # 处理返回结果
+        # # 处理返回结果
-        try:
+        # try:
-            AiShow.objects.create(
+        #     AiShow.objects.create(
-                cinema=cinema.name,
+        #         cinema=cinema.name,
-                zz_code=cinema.zz_code,
+        #         zz_code=cinema.zz_code,
-                show_date=show_date,
+        #         show_date=show_date,
-                is_ai_show=True,
+        #         is_ai_show=True,
-                show=_show,
+        #         show=_show,
-                sales=_sales,
+        #         sales=_sales,
-                prompt=prompt,
+        #         prompt=prompt,
-                result=result,
+        #         result=result,
-                message=message,
+        #         message=message,
-                take_times=int((end - start).seconds),
+        #         take_times=int((end - start).seconds),
-                take_tokens=tokens
+        #         take_tokens=tokens
-            )
+        #     )
-        except Exception as e:
+        # except Exception as e:
-            print(e)
+        #     print(e)
     return True