自动添加目标日期排片

ai/utils/show_func.py

@@ -67,6 +67,15 @@ def get_template_show(cinema, show_date):
     return temp_show, temp_date
 
 
+# 获取目标日期排片数据
+def get_target_show(cinema, show_date):
+    history_start, history_end, target_start, target_end = get_data_datetime(show_date)
+    data = GetData(cinema)
+    target_date_show = data.get_show_data(target_start, target_end)
+    if not target_date_show:
+        target_date_show = '影厅别名,影厅id,影片别名,本地影片id,语言,放映日期,开始时间,结束时间,片长,场间'
+    return target_date_show
+
 def get_all_movie_hot_info(cinema, show_date):
     data = GetData(cinema)
     movie = data.get_movie_info(show_date)

ai/utils/show_process.py

@@ -33,6 +33,7 @@ def show_main_process(zz_code=None, day_delta=3):
         print(cinema.name)
         show_ai = ShowAI(cinema, show_date)
         template_show, template_date = get_template_show(cinema, show_date)
+        target_show = get_target_show(cinema, show_date)
         prompt = show_ai.general_prompt()
         start = datetime.datetime.now()  # 开始计时
         result, message, tokens = show_ai.get_show_result_ai()
@@ -59,6 +60,7 @@ def show_main_process(zz_code=None, day_delta=3):
                 is_ai_show=True,
                 template=template_show,
                 temp_date=template_date,
+                target=target_show,
                 show=_show,
                 sales=_sales,
                 prompt=prompt,