diff --git a/ai/utils/show_func.py b/ai/utils/show_func.py index da52dcf..607bf05 100644 --- a/ai/utils/show_func.py +++ b/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) diff --git a/ai/utils/show_process.py b/ai/utils/show_process.py index ac77ded..840cb2c 100644 --- a/ai/utils/show_process.py +++ b/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,