增加指定生成日期的功能

ai/utils/show_process.py

@@ -9,7 +9,7 @@ from django.db.models import Q
 import re
 
 # ai排片主流程
-def show_main_process(zz_code=None):
+def show_main_process(zz_code=None, day_delta=3):
     # 查看状态
     redis_conn = get_redis_connection()
     redis_key = f'ai_show{datetime.date.today().strftime("%Y%m%d")}'
@@ -26,7 +26,7 @@ def show_main_process(zz_code=None):
         Q(del_flag=False) & Q(prompt_type='version') & Q(prompt_key='ShowPromptVersion')).first().prompt_val
     # 开始处理提示词
     # 生成目标日期
-    show_date = datetime.date.strftime(datetime.date.today() + datetime.timedelta(days=3), '%Y-%m-%d')
+    show_date = datetime.date.strftime(datetime.date.today() + datetime.timedelta(days=day_delta), '%Y-%m-%d')
     for cinema in test_cinema_list:
         print(cinema.name)
         show_ai = ShowAI(cinema, show_date)

ai/views.py

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