From 59096ef9c090ce2677b58bc864c4bc38905864dd Mon Sep 17 00:00:00 2001 From: rogersun Date: Fri, 10 Jul 2026 18:01:05 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=EF=BC=8C=E6=96=B0=E7=89=87?= =?UTF-8?q?=E7=9A=84=E4=BF=A1=E6=81=AF=E4=BB=8E=E6=8E=92=E7=89=87=E5=BD=93?= =?UTF-8?q?=E5=A4=A9=E6=94=B9=E4=B8=BA=E7=9B=AE=E6=A0=87=E6=97=A5=E6=9C=9F?= =?UTF-8?q?=E5=88=B0=E6=8E=92=E7=89=87=E6=97=A5=E6=9C=9F=E4=B8=AD=E7=9A=84?= =?UTF-8?q?=E5=85=A8=E9=83=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ai/utils/show_func.py | 30 +++++++++++++++++++++--------- 1 file changed, 21 insertions(+), 9 deletions(-) diff --git a/ai/utils/show_func.py b/ai/utils/show_func.py index 33cac6c..302ad30 100644 --- a/ai/utils/show_func.py +++ b/ai/utils/show_func.py @@ -136,14 +136,25 @@ def get_all_movie_hot_info(req_date: str = 'today', show_type: str = 'old', get_ # 模拟用户设置的新片推荐数量 def get_new_suggestion(cinema, show_date): data = GetData(cinema) - movie = data.get_movie_info(show_date, 'new') + tomorrow = datetime.datetime.today() + datetime.timedelta(days=-1) + show_date_obj = datetime.datetime.strptime(show_date, '%Y-%m-%d') + delta = (show_date_obj - tomorrow).days + date_list = [datetime.datetime.strftime(tomorrow + datetime.timedelta(days=i), "%Y-%m-%d") for i in + range(0, delta + 2)] + print('get_new_suggestion-date_list', date_list) new_movie_list = [] - for m in movie: - new_movie = NewMovieSuggestion.objects.filter( - 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: - new_movie_list.append( - f"\t《{m['cinema_movie_cn_name']}》的不同制式版本总计建议排片占比为{new_movie.suggestion}%;") + for date in date_list: + movie = data.get_movie_info(date, 'new') + for m in movie: + print(m['cinema_movie_cn_name']) + new_movie = NewMovieSuggestion.objects.filter( + 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']}》建议排一些场次;" + else: + 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) return None @@ -187,10 +198,11 @@ def get_new_movie(start_date): # 模拟用户设置的新片推荐数量 def get_movie_guidance(cinema, show_date): - movie_guidance = MovieGuidance.objects.filter(Q(zz_code=cinema.zz_code) & Q(show_date=show_date) & Q(delete_flag=False)).all() + movie_guidance = MovieGuidance.objects.filter( + Q(zz_code=cinema.zz_code) & Q(show_date=show_date) & Q(delete_flag=False)).all() if movie_guidance: movie_list = [] for m in movie_guidance: movie_list.append(f"\t《{m.movie_name}》强制排片占比为{m.guidance}%;") return '\n'.join(movie_list) - return '\t无' \ No newline at end of file + return '\t无'