From d5a48332f1a68927baa80562f71178a4c185cc08 Mon Sep 17 00:00:00 2001 From: rogersun Date: Mon, 13 Jul 2026 16:57:37 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8E=BB=E9=99=A4=E5=BD=B1=E7=89=87=E5=90=8D?= =?UTF-8?q?=E7=A7=B0=E5=89=8D=E5=90=8E=E7=9A=84=E4=B9=A6=E5=90=8D=E5=8F=B7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ai/utils/show_database.py | 6 +-- ai/utils/show_func.py | 10 ++--- ai/utils/show_process.py | 88 +++++++++++++++++++-------------------- 3 files changed, 52 insertions(+), 52 deletions(-) diff --git a/ai/utils/show_database.py b/ai/utils/show_database.py index 46e0ec8..9eb978e 100644 --- a/ai/utils/show_database.py +++ b/ai/utils/show_database.py @@ -51,7 +51,7 @@ class GetData: self.cur.execute(GET_AVAILABLE_MOVIE, (start_datetime, end_datetime)) available_movie = self.cur.fetchall() available_movie_list = [ - f"\t《{m['cinema_movie_alias']}》 - 影片id：{m['cinema_movie_id']}；" + f"\t{m['cinema_movie_alias']} - 影片id：{m['cinema_movie_id']}；" for m in available_movie] return '\n'.join(available_movie_list) @@ -71,7 +71,7 @@ class GetData: print(self.cur.mogrify(sql, (start_datetime, end_datetime))) available_movie = self.cur.fetchall() available_movie_list = [ - f"\t《{m['cinema_movie_alias']}》 - 影片id：{m['cinema_movie_id']}，影片时长：{m['cinema_movie_time']}，语言：{m['language']}，制式：{m['media_type']}， 最早排片开始时间：{m['cinema_movie_start_datetime']}， 最晚排片截止时间：{m['cinema_movie_end_datetime']}；" + f"\t{m['cinema_movie_alias']} - 影片id：{m['cinema_movie_id']}，影片时长：{m['cinema_movie_time']}，语言：{m['language']}，制式：{m['media_type']}， 最早排片开始时间：{m['cinema_movie_start_datetime']}， 最晚排片截止时间：{m['cinema_movie_end_datetime']}；" for m in available_movie] return '\n'.join(available_movie_list) @@ -82,7 +82,7 @@ class GetData: seconds=59) self.cur.execute(GET_NEW_MOVIE, (start_datetime, end_datetime)) new = self.cur.fetchall() - new_list = [f"《{m['cinema_movie_alias']}》" for m in new] + new_list = [f"{m['cinema_movie_alias']}" for m in new] return new_list # 为获取影片热度功能提供影院数据 diff --git a/ai/utils/show_func.py b/ai/utils/show_func.py index eb01b58..c96210d 100644 --- a/ai/utils/show_func.py +++ b/ai/utils/show_func.py @@ -126,9 +126,9 @@ def get_all_movie_hot_info(req_date: str = 'today', show_type: str = 'old', get_ else: return [] for hot in hot_data: - print(f"\t《{hot.movie_name}》,{hot.wants},{hot.hots},{','.join([d['value'] for d in json.loads(hot.day_data)])}") + print(f"\t{hot.movie_name},{hot.wants},{hot.hots},{','.join([d['value'] for d in json.loads(hot.day_data)])}") data_list.append( - f"\t《{hot.movie_name}》,{hot.wants},{hot.hots},{','.join([d['value'] for d in json.loads(hot.day_data)])}") + f"\t{hot.movie_name},{hot.wants},{hot.hots},{','.join([d['value'] for d in json.loads(hot.day_data)])}") print(data_list) return data_list @@ -151,9 +151,9 @@ def get_new_suggestion(cinema, show_date): Q(movie_name=m['cinema_movie_cn_name']) & Q(zz_code=cinema.zz_code) & Q(delete_flag=False)).first() if new_movie is not None: if new_movie.suggestion == '0': - text = f"\t《{m['cinema_movie_cn_name']}》建议排一些场次;" + text = f"\t{m['cinema_movie_cn_name']} 建议排一些场次;" else: - text = f"\t《{m['cinema_movie_cn_name']}》的不同制式版本总计建议排片占比为{new_movie.suggestion}%;" + text = f"\t{m['cinema_movie_cn_name']} 的不同制式版本总计建议排片占比为{new_movie.suggestion}%;" new_movie_list.append(text) if len(new_movie_list) > 0: return '\n'.join(new_movie_list) @@ -203,7 +203,7 @@ def get_movie_guidance(cinema, show_date): if movie_guidance: movie_list = [] for m in movie_guidance: - movie_list.append(f"\t《{m.movie_name}》强制排片占比为{m.guidance}%;") + movie_list.append(f"\t{m.movie_name} 强制排片占比为{m.guidance}%;") return '\n'.join(movie_list) return '\t无' diff --git a/ai/utils/show_process.py b/ai/utils/show_process.py index 24708fc..0de3e74 100644 --- a/ai/utils/show_process.py +++ b/ai/utils/show_process.py @@ -44,50 +44,50 @@ def show_main_process(zz_code=None, day_delta=None): else: target_show = '影厅别名,影厅id,影片别名,本地影片id,语言,放映日期,开始时间,结束时间,片长,场间' prompt = show_ai.general_prompt() - start = datetime.datetime.now() # 开始计时 - result, message, tokens = show_ai.get_show_result_ai() - end = datetime.datetime.now() # 结束计时 - print('prompt:', prompt) - print('result:', result) - print('message:', message) - print('tokens:', tokens) - - # 获取排片数据 - print('result:', result) - if model.config_name in ['lmstudio_qwen3.6_35b', 'company_qwen3.5_35b']: - result_obj = json.loads(result.replace('```json', '').replace('}\n```', '}').strip()) - else: - result_obj = json.loads(result.strip()) - - # 预测排片数据 - _show = result_obj['show'].replace('```', '').replace('csv', '').strip() - - # 预测销售数据 - _sales = str(result_obj['income']).strip() - - # 处理返回结果 - try: - AiShow.objects.create( - cinema=cinema.name, - zz_code=cinema.zz_code, - show_date=show_date, - is_ai_show=True, - is_manual_test=False, - ai_model=model.config_name, - template=template_show, - temp_date=template_date, - target=target_show, - show=data_cleansing(_show), - sales=_sales, - prompt=prompt, - result=result, - message=message, - take_times=int((end - start).seconds), - take_tokens=tokens, - prompt_version=prompt_ver, - ) - except Exception as e: - print(e) + # start = datetime.datetime.now() # 开始计时 + # result, message, tokens = show_ai.get_show_result_ai() + # end = datetime.datetime.now() # 结束计时 + # print('prompt:', prompt) + # print('result:', result) + # print('message:', message) + # print('tokens:', tokens) + # + # # 获取排片数据 + # print('result:', result) + # if model.config_name in ['lmstudio_qwen3.6_35b', 'company_qwen3.5_35b']: + # result_obj = json.loads(result.replace('```json', '').replace('}\n```', '}').strip()) + # else: + # result_obj = json.loads(result.strip()) + # + # # 预测排片数据 + # _show = result_obj['show'].replace('```', '').replace('csv', '').strip() + # + # # 预测销售数据 + # _sales = str(result_obj['income']).strip() + # + # # 处理返回结果 + # try: + # AiShow.objects.create( + # cinema=cinema.name, + # zz_code=cinema.zz_code, + # show_date=show_date, + # is_ai_show=True, + # is_manual_test=False, + # ai_model=model.config_name, + # template=template_show, + # temp_date=template_date, + # target=target_show, + # show=data_cleansing(_show), + # sales=_sales, + # prompt=prompt, + # result=result, + # message=message, + # take_times=int((end - start).seconds), + # take_tokens=tokens, + # prompt_version=prompt_ver, + # ) + # except Exception as e: + # print(e) return True