RogerWorkPC
/
dingxin_toolbox_drf
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

去除影片名称前后的书名号

Browse Source
main
rogersun 4 days ago
parent 4f0d873448
commit d5a48332f1
3 changed files with 52 additions and 52 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 6
      ai/utils/show_database.py
  2. 10
      ai/utils/show_func.py
  3. 88
      ai/utils/show_process.py

6
ai/utils/show_database.py
Unescape View File

@ -51,7 +51,7 @@ class GetData:
self.cur.execute(GET_AVAILABLE_MOVIE, (start_datetime, end_datetime)) self.cur.execute(GET_AVAILABLE_MOVIE, (start_datetime, end_datetime))
available_movie = self.cur.fetchall() available_movie = self.cur.fetchall()
available_movie_list = [ 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] for m in available_movie]
return '\n'.join(available_movie_list) return '\n'.join(available_movie_list)
@ -71,7 +71,7 @@ class GetData:
print(self.cur.mogrify(sql, (start_datetime, end_datetime))) print(self.cur.mogrify(sql, (start_datetime, end_datetime)))
available_movie = self.cur.fetchall() available_movie = self.cur.fetchall()
available_movie_list = [ 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] for m in available_movie]
return '\n'.join(available_movie_list) return '\n'.join(available_movie_list)
@ -82,7 +82,7 @@ class GetData:
seconds=59) seconds=59)
self.cur.execute(GET_NEW_MOVIE, (start_datetime, end_datetime)) self.cur.execute(GET_NEW_MOVIE, (start_datetime, end_datetime))
new = self.cur.fetchall() 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 return new_list
# 为获取影片热度功能提供影院数据 # 为获取影片热度功能提供影院数据

10
ai/utils/show_func.py
Unescape View File

@ -126,9 +126,9 @@ def get_all_movie_hot_info(req_date: str = 'today', show_type: str = 'old', get_
else: else:
return [] return []
for hot in hot_data: 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( 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) print(data_list)
return 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() 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 is not None:
if new_movie.suggestion == '0': if new_movie.suggestion == '0':
text = f"\t《{m['cinema_movie_cn_name']}》建议排一些场次;" text = f"\t{m['cinema_movie_cn_name']} 建议排一些场次;"
else: 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) new_movie_list.append(text)
if len(new_movie_list) > 0: if len(new_movie_list) > 0:
return '\n'.join(new_movie_list) return '\n'.join(new_movie_list)
@ -203,7 +203,7 @@ def get_movie_guidance(cinema, show_date):
if movie_guidance: if movie_guidance:
movie_list = [] movie_list = []
for m in movie_guidance: 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 '\n'.join(movie_list)
return '\t无' return '\t无'

88
ai/utils/show_process.py
Unescape View File

@ -44,50 +44,50 @@ def show_main_process(zz_code=None, day_delta=None):
else: else:
target_show = '影厅别名,影厅id,影片别名,本地影片id,语言,放映日期,开始时间,结束时间,片长,场间' target_show = '影厅别名,影厅id,影片别名,本地影片id,语言,放映日期,开始时间,结束时间,片长,场间'
prompt = show_ai.general_prompt() prompt = show_ai.general_prompt()
start = datetime.datetime.now() # 开始计时 # 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)
#
# 获取排片数据 # # 获取排片数据
print('result:', result) # print('result:', result)
if model.config_name in ['lmstudio_qwen3.6_35b', 'company_qwen3.5_35b']: # if model.config_name in ['lmstudio_qwen3.6_35b', 'company_qwen3.5_35b']:
result_obj = json.loads(result.replace('```json', '').replace('}\n```', '}').strip()) # result_obj = json.loads(result.replace('```json', '').replace('}\n```', '}').strip())
else: # else:
result_obj = json.loads(result.strip()) # result_obj = json.loads(result.strip())
#
# 预测排片数据 # # 预测排片数据
_show = result_obj['show'].replace('```', '').replace('csv', '').strip() # _show = result_obj['show'].replace('```', '').replace('csv', '').strip()
#
# 预测销售数据 # # 预测销售数据
_sales = str(result_obj['income']).strip() # _sales = str(result_obj['income']).strip()
#
# 处理返回结果 # # 处理返回结果
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,
is_manual_test=False, # is_manual_test=False,
ai_model=model.config_name, # ai_model=model.config_name,
template=template_show, # template=template_show,
temp_date=template_date, # temp_date=template_date,
target=target_show, # target=target_show,
show=data_cleansing(_show), # show=data_cleansing(_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,
prompt_version=prompt_ver, # prompt_version=prompt_ver,
) # )
except Exception as e: # except Exception as e:
print(e) # print(e)
return True return True

Write Preview
Loading…
Cancel
Save