优化代码结构

main
rogersun 2 days ago
parent ffcd320475
commit 6594d98953
  1. 1
      ai/utils/show_process.py
  2. 2
      ai/utils/show_prompt.py
  3. 2
      ai/views.py

@ -21,6 +21,7 @@ def show_main_process(zz_code=None, day_delta=3):
test_cinema_list = TestCinema.objects.filter(is_active=True).all()
else:
test_cinema_list = TestCinema.objects.filter(Q(zz_code=zz_code) & Q(is_active=True)).all()
day_delta = 3 if day_delta is None else int(day_delta)
# 获取提示词版本
prompt_ver = PromptTemplate.objects.filter(
Q(del_flag=False) & Q(prompt_type='version') & Q(prompt_key='ShowPromptVersion')).first().prompt_val

@ -41,7 +41,7 @@ class ShowAI:
pre_data = pre_data.replace('{history_show}', data.get_show_data(history_start, history_end)) # 处理排片数据
pre_data = pre_data.replace('{history_sales}', data.get_sell_data(history_start, history_end)) # 处理销售数据
pre_data = pre_data.replace('{target_already_show}',
'' if data.get_show_data(target_start,
'\t' if data.get_show_data(target_start,
target_end) is False else data.get_show_data(
target_start, target_end)) # 处理目标日期已排场次数据
pre_data = pre_data.replace('{reference_date}', history_date) # 处理参考日期

@ -12,7 +12,7 @@ from ai.utils.basic_func import *
@csrf_exempt
def manual_general_show(request):
zz_code = request.GET.dict().get('cinema_code')
delta = 3 if request.GET.dict().get('delta') is None else int(request.GET.dict().get('delta'))
delta = request.GET.dict().get('delta')
result = show_main_process(zz_code, delta)
result_dict = {
'status': 'success' if result else 'fail',

Loading…
Cancel
Save