适配提示词1.2版本，优化影片热度和销售数据的提示词格式，影片热度支持本地缓存，支持模型切换，部分设置改为运行时配置。

ai/migrations/0011_aimodel_aishow_ai_model.py

@@ -0,0 +1,33 @@
+# Generated by Django 4.2.7 on 2026-07-03 14:23
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ai', '0010_aishow_target'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='AiModel',
+            fields=[
+                ('id', models.AutoField(primary_key=True, serialize=False)),
+                ('config_name', models.CharField(max_length=50)),
+                ('base_url', models.CharField(max_length=100)),
+                ('api_key', models.CharField(max_length=50)),
+                ('model_name', models.CharField(default='', max_length=500)),
+            ],
+            options={
+                'verbose_name': '模型配置',
+                'verbose_name_plural': '模型配置',
+                'db_table': 'ai_model_configs',
+            },
+        ),
+        migrations.AddField(
+            model_name='aishow',
+            name='ai_model',
+            field=models.CharField(default='', max_length=100),
+        ),
+    ]

ai/migrations/0012_aimodel_delete_flag.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.7 on 2026-07-04 13:33
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ai', '0011_aimodel_aishow_ai_model'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='aimodel',
+            name='delete_flag',
+            field=models.BooleanField(default=False),
+        ),
+    ]

ai/migrations/0013_aimodel_current_used.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.7 on 2026-07-04 13:47
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ai', '0012_aimodel_delete_flag'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='aimodel',
+            name='current_used',
+            field=models.BooleanField(default=False),
+        ),
+    ]

ai/migrations/0014_airuntime.py

@@ -0,0 +1,26 @@
+# Generated by Django 4.2.7 on 2026-07-04 13:58
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ai', '0013_aimodel_current_used'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='AiRuntime',
+            fields=[
+                ('id', models.AutoField(primary_key=True, serialize=False)),
+                ('runtime_key', models.CharField(max_length=50)),
+                ('runtime_val', models.TextField()),
+            ],
+            options={
+                'verbose_name': 'AI运行时',
+                'verbose_name_plural': 'AI运行时',
+                'db_table': 'ai_runtime',
+            },
+        ),
+    ]

ai/migrations/0015_moviehotinfo.py

@@ -0,0 +1,35 @@
+# Generated by Django 4.2.7 on 2026-07-04 15:06
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ai', '0014_airuntime'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='MovieHotInfo',
+            fields=[
+                ('id', models.AutoField(primary_key=True, serialize=False)),
+                ('movie_zz_code', models.CharField(max_length=50)),
+                ('movie_name', models.CharField(max_length=50)),
+                ('req_date', models.DateField()),
+                ('api_response', models.TextField()),
+                ('wants', models.CharField(max_length=50)),
+                ('hots', models.CharField(max_length=50)),
+                ('day_1', models.CharField(max_length=50)),
+                ('day_2', models.CharField(max_length=50)),
+                ('day_3', models.CharField(max_length=50)),
+                ('day_4', models.CharField(max_length=50)),
+                ('day_5', models.CharField(max_length=50)),
+            ],
+            options={
+                'verbose_name': 'AI影片热度信息',
+                'verbose_name_plural': 'AI影片热度信息',
+                'db_table': 'ai_movie_hot',
+            },
+        ),
+    ]

ai/migrations/0016_remove_moviehotinfo_day_1_remove_moviehotinfo_day_2_and_more.py

@@ -0,0 +1,83 @@
+# Generated by Django 4.2.7 on 2026-07-04 15:16
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ai', '0015_moviehotinfo'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_1',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_2',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_3',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_4',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_5',
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day_1_date',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day_1_val',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day_2_date',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day_2_val',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day_3_date',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day_3_val',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day_4_date',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day_4_val',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day_5_date',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day_5_val',
+            field=models.CharField(default='', max_length=50),
+        ),
+    ]

ai/migrations/0017_remove_moviehotinfo_day_1_date_and_more.py

@@ -0,0 +1,58 @@
+# Generated by Django 4.2.7 on 2026-07-04 15:18
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ai', '0016_remove_moviehotinfo_day_1_remove_moviehotinfo_day_2_and_more'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_1_date',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_1_val',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_2_date',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_2_val',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_3_date',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_3_val',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_4_date',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_4_val',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_5_date',
+        ),
+        migrations.RemoveField(
+            model_name='moviehotinfo',
+            name='day_5_val',
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day_data',
+            field=models.TextField(default=''),
+        ),
+    ]

ai/migrations/0018_moviehotinfo_day1_val_moviehotinfo_day2_val_and_more.py

@@ -0,0 +1,38 @@
+# Generated by Django 4.2.7 on 2026-07-04 22:57
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ai', '0017_remove_moviehotinfo_day_1_date_and_more'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day1_val',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day2_val',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day3_val',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day4_val',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='day5_val',
+            field=models.CharField(default='', max_length=50),
+        ),
+    ]

ai/migrations/0019_moviehotinfo_title_alter_moviehotinfo_hots_and_more.py

@@ -0,0 +1,28 @@
+# Generated by Django 4.2.7 on 2026-07-04 23:00
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ai', '0018_moviehotinfo_day1_val_moviehotinfo_day2_val_and_more'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='title',
+            field=models.CharField(default='', max_length=1000),
+        ),
+        migrations.AlterField(
+            model_name='moviehotinfo',
+            name='hots',
+            field=models.CharField(default='', max_length=50),
+        ),
+        migrations.AlterField(
+            model_name='moviehotinfo',
+            name='wants',
+            field=models.CharField(default='', max_length=50),
+        ),
+    ]

ai/migrations/0020_moviehotinfo_show_type.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.7 on 2026-07-05 09:37
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ai', '0019_moviehotinfo_title_alter_moviehotinfo_hots_and_more'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='moviehotinfo',
+            name='show_type',
+            field=models.CharField(default='', max_length=500),
+        ),
+    ]

ai/models.py

@@ -3,6 +3,11 @@ from django.db import models
 
 # Create your models here.
 class TestCinema(models.Model):
+    """
+    用于存储测试影院基本信息、数据库配置信息
+    user_config用于模拟用户设置情况
+    is_active设置为False时不会自动生成影院的排片，需要手动生成
+    """
     id = models.AutoField(primary_key=True)
     name = models.CharField(max_length=50)
     zz_code = models.CharField(max_length=50)
@@ -20,6 +25,10 @@ class TestCinema(models.Model):
 
 
 class PromptTemplate(models.Model):
+    """
+    分板块的提示词模板，用于动态调整提示词的内容和结构。
+    ShowPromptVersion为当前提示词版本
+    """
     id = models.AutoField(primary_key=True)
     prompt_type = models.CharField(max_length=50)
     prompt_key = models.CharField(max_length=50)
@@ -36,24 +45,33 @@ class PromptTemplate(models.Model):
 
 
 class AiShow(models.Model):
+    """
+    用于存储ai生成的结果已经影院真实排片
+    temp_date： 模板数据日期
+    target： 目标日期已排影片数据，csv格式
+    show： 目标日期ai生成的排片数据，csv格式
+    sales： ai排片预期收入
+    prompt
+    """
     id = models.AutoField(primary_key=True)
     cinema = models.CharField(max_length=50)
     zz_code = models.CharField(max_length=50)
-    show_date = models.DateField()
-    is_ai_show = models.BooleanField(default=True)
-    template = models.TextField(default='')
-    temp_date = models.CharField(max_length=50, default='')
-    target = models.TextField(default='')
-    show = models.TextField()
-    sales = models.CharField(max_length=50)
-    prompt = models.TextField()
-    result = models.TextField()
-    message = models.TextField()
-    created_at = models.DateTimeField(auto_now_add=True)
+    show_date = models.DateField()  # 排片目标日期
+    is_ai_show = models.BooleanField(default=True)  # 当is_ai_show为True时，记录为ai生成的排片，否则为影院真实排片
+    ai_model = models.CharField(default='', max_length=100)  # ai生成排片时使用的模型
+    template = models.TextField(default='')  # ai生成排片时的模板日期排片数据，csv格式
+    temp_date = models.CharField(max_length=50, default='')  # 模板数据日期
+    target = models.TextField(default='')  # 目标日期已排影片数据，csv格式
+    show = models.TextField()  # 目标日期ai生成的排片数据，csv格式
+    sales = models.CharField(max_length=50)  # ai排片预期收入
+    prompt = models.TextField()  # 生成的用户提示词部分
+    result = models.TextField()  # ai返回结果中最终输出部分
+    message = models.TextField()  # ai返回结果的完整信息
+    created_at = models.DateTimeField(auto_now_add=True)  # ai返回结果的时间
     updated_at = models.DateTimeField(auto_now=True)
-    take_times = models.IntegerField(default=0)
-    take_tokens = models.CharField(default='', max_length=2000)
-    prompt_version = models.CharField(max_length=50, default='')
+    take_times = models.IntegerField(default=0)   # 从提示词发给ai到取得返回结果的时间，单位秒
+    take_tokens = models.CharField(default='', max_length=2000)   # ai返回结果中token消耗部分的信息
+    prompt_version = models.CharField(max_length=50, default='')  # 使用的提示词版本
 
     def __str__(self):
         return self.cinema
@@ -78,3 +96,61 @@ class CinemaHall(models.Model):
         verbose_name = '影厅数据'
         verbose_name_plural = '影厅数据'
         db_table = 'ai_show_hall'
+
+
+class AiModel(models.Model):
+    id = models.AutoField(primary_key=True)
+    config_name = models.CharField(max_length=50)
+    base_url = models.CharField(max_length=100)
+    api_key = models.CharField(max_length=50)
+    model_name = models.CharField(max_length=500, default='')
+    delete_flag = models.BooleanField(default=False)
+    current_used = models.BooleanField(default=False)
+
+    def __str__(self):
+        return self.config_name
+
+    class Meta:
+        verbose_name = '模型配置'
+        verbose_name_plural = '模型配置'
+        db_table = 'ai_model_configs'
+
+
+class AiRuntime(models.Model):
+    id = models.AutoField(primary_key=True)
+    runtime_key = models.CharField(max_length=50)
+    runtime_val = models.TextField()
+
+    def __str__(self):
+        return self.runtime_key
+
+    class Meta:
+        verbose_name = 'AI运行时'
+        verbose_name_plural = 'AI运行时'
+        db_table = 'ai_runtime'
+
+
+class MovieHotInfo(models.Model):
+    id = models.AutoField(primary_key=True)
+    movie_zz_code = models.CharField(max_length=50)
+    movie_name = models.CharField(max_length=50)
+    req_date = models.DateField()
+    api_response = models.TextField()
+    show_type = models.CharField(max_length=500, default='')
+    title = models.CharField(max_length=1000, default='')
+    wants = models.CharField(max_length=50, default='')
+    hots = models.CharField(max_length=50, default='')
+    day1_val = models.CharField(max_length=50, default='')
+    day2_val = models.CharField(max_length=50, default='')
+    day3_val = models.CharField(max_length=50, default='')
+    day4_val = models.CharField(max_length=50, default='')
+    day5_val = models.CharField(max_length=50, default='')
+    day_data = models.TextField(default='')
+
+    def __str__(self):
+        return self.movie_name
+
+    class Meta:
+        verbose_name = 'AI影片热度信息'
+        verbose_name_plural = 'AI影片热度信息'
+        db_table = 'ai_movie_hot'

ai/serializers.py

@@ -25,3 +25,21 @@ class CinemaHallSerializer(serializers.ModelSerializer):
     class Meta:
         model = CinemaHall
         fields = '__all__'
+
+
+class AiModelSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = AiModel
+        fields = '__all__'
+
+
+class AiRuntimeSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = AiRuntime
+        fields = '__all__'
+
+
+class MovieHotInfoSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = MovieHotInfo
+        fields = '__all__'

ai/utils/basic_func.py

@@ -49,10 +49,11 @@ def format_show_data(cinema_code: str, show_date: str, data_left: dict, data_rig
     for h in hall:
         hall_dict = {'hall_name': h.hall_name, 'hall_id': h.hall_id, 'ai': [], 'real': []}
         for ai in data_left:
-            if ai['影厅id'] == str(h.hall_id):
+            if dict_get(ai, ['影厅id', '影厅 id']) == str(h.hall_id):
+                print(ai)
                 hall_dict['ai'].append({
                     'movie_name': ai['影片别名'],
-                    'movie_id': ai['本地影片id'],
+                    'movie_id': dict_get(ai, ['本地影片id' '本地影片 id']),
                     'language': ai['语言'],
                     'show_date': ai['放映日期'],
                     'start': ai['开始时间'],
@@ -61,10 +62,10 @@ def format_show_data(cinema_code: str, show_date: str, data_left: dict, data_rig
                     'duration': ai['场间']
                 })
         for real in data_right:
-            if real['影厅id'] == str(h.hall_id):
+            if dict_get(real, ['影厅id', '影厅 id']) == str(h.hall_id):
                 hall_dict['real'].append({
                     'movie_name': real['影片别名'],
-                    'movie_id': real['本地影片id'],
+                    'movie_id': dict_get(real, ['本地影片id' '本地影片 id']),
                     'language': real['语言'],
                     'show_date': real['放映日期'],
                     'start': real['开始时间'],
@@ -143,6 +144,7 @@ def get_hall_type(cinema_code):
         hall_dict[h.hall_name] = h.hall_type
     return hall_dict
 
+
 # 获取排片占比
 def count_show(show_data):
     print('count_show', show_data)
@@ -158,12 +160,35 @@ def count_show(show_data):
         show_count.append(f"{show}：{str(round(count / len(show_data) * 100, 2))}%，共排{str(count)}场")
     return '\n'.join(show_count)
 
+
 # 处理黑名单
 def get_black_list(cinema_code: str):
     black_list = json.loads(TestCinema.objects.filter(zz_code=cinema_code).first().user_config)['movie_black_list']
     black_list_bool = False
-    black_list_str = '\' \''
+    black_list_str = "' '"
     if len(black_list) > 0:
         black_list_bool = True
         black_list_str = ','.join(['%s'] * len(black_list))
     return black_list_bool, black_list_str, black_list
+
+
+# 获取运行时参数
+def get_runtime_val(key: str):
+    return AiRuntime.objects.filter(runtime_key=key).first().runtime_val
+
+# 获取当前开启的模型
+def get_current_model():
+    return AiModel.objects.filter(current_used=True).first()
+
+
+# 通不过不同key获取数据
+def dict_get(_dict: dict, keys: list):
+    for k in keys:
+        if k in _dict.keys():
+            return _dict[k]
+    return False
+
+
+# AI返回数据清洗
+def data_cleansing(data:str):
+    return data.replace('，', ',').replace('影厅 id', '影厅id').replace('本地影片 id', '本地影片id').replace(', ', ',')

ai/utils/datetime_format.py

@@ -14,6 +14,19 @@ def get_data_datetime(show_date):
     target_end = datetime.datetime.strftime(target_end_datetime_obj, '%Y-%m-%d %H:%M:%S')
     return history_start, history_end, target_start, target_end
 
+def get_data_datetime_template(show_date, template_date):
+    show_date_obj = datetime.datetime.strptime(show_date, '%Y-%m-%d')
+    target_start_datetime_obj = show_date_obj + datetime.timedelta(hours=6)
+    target_end_datetime_obj = show_date_obj + datetime.timedelta(hours=29, minutes=59, seconds=59)
+    template_date_obj = datetime.datetime.strptime(template_date, '%Y-%m-%d')
+    data_start_datetime_obj = template_date_obj + datetime.timedelta(hours=6)
+    data_end_datetime_obj = template_date_obj + datetime.timedelta(hours=29, minutes=59, seconds=59)
+    history_start = datetime.datetime.strftime(data_start_datetime_obj, '%Y-%m-%d %H:%M:%S')
+    history_end = datetime.datetime.strftime(data_end_datetime_obj, '%Y-%m-%d %H:%M:%S')
+    target_start = datetime.datetime.strftime(target_start_datetime_obj, '%Y-%m-%d %H:%M:%S')
+    target_end = datetime.datetime.strftime(target_end_datetime_obj, '%Y-%m-%d %H:%M:%S')
+    return history_start, history_end, target_start, target_end
+
 
 def get_date_type(show_date, template_date):
     show_date_obj = datetime.datetime.strptime(show_date, '%Y-%m-%d')

ai/utils/movie_data.py

@@ -1,17 +1,28 @@
 from hashlib import md5
 import requests
+from ai.models import *
 import json
-from datetime import datetime as dt
+from django.db.models import Q
+import datetime
 
 
 # 从BI的日常影片数据接口中获取信息
 class MovieData:
     def __init__(self):
         self.salt = '53b22eda5938706a34fd9d44f097bec2'
-        self.ts = str(int(dt.timestamp(dt.now())))
+        self.ts = str(int(datetime.datetime.timestamp(datetime.datetime.now())))
         self.url = 'https://nation.bi.piao51.cn/nationDataApi/restful/boxoffice/nation/movieSupportData'
 
-    def get_data(self, movie_code):
+    # 从BI获取热度数据
+    def get_data_from_bi(self, movie_code, show_type):
+        # 基本变量定义
+        today_str = datetime.datetime.today().strftime('%Y-%m-%d')  # 当天日期字符串
+        movie_name = ''  # 返回的影片名称，以BI返回为准
+        hot_title = ''  # CSV表头
+        hot_data = ''  # 数据部分
+        # 如果本地有则读取本地数据
+        today_obj = datetime.datetime.strptime(today_str, '%Y-%m-%d')
+        if not MovieHotInfo.objects.filter(Q(movie_zz_code=movie_code) & Q(req_date=today_obj)).exists():
             params = {
                 'movieCode': movie_code,
                 'timeStamp': self.ts,
@@ -19,27 +30,62 @@ class MovieData:
             }
             req = requests.get(self.url, params=params)
             # print(json.dumps(req.json(), indent=4))
-        r = ''
+
             req = req.json()
+
             if (req['success'] is True) and req['data']['movieInfo'] != '':
+                # 基础数据
                 info = req['data']['movieInfo']
-            hot = f"《{info['movie_name']}》 - 想看：{info['wish_num']}，热度：小红书{info['xiaohongshu']}、视频号{info['shipinhao']}、抖音{info['douyin']}、微博{info['weibo']}；"
-            # print(hot)
+                movie_name = info['movie_name']
                 daily = req['data']['movieDailyBox']
-            sales = [f"{daily[n]['businessDate']} {daily[n]['boxRate']}%" for n in range(0, 6)]
-            # print(sales)
-            pre_sales = [f"{daily[n]['businessDate']} {daily[n]['boxRate']}%" for n in range(7, len(daily))]
-            # print(pre_sales)
-            r = f"\t{hot}过去六天的票房占比：{'、'.join(sales)}；今天到当前时间的票房占比：{daily[6]['boxRate']}%；未来7天的预售票房占比：{'、'.join(pre_sales)}；"
-        print('MovieData.get_data', movie_code, r)
-        return r
+                if MovieHotInfo.objects.filter(Q(movie_name=movie_name) & Q(req_date=today_str)).exists():
+                    return '', '', ''
+                # 生成标题
+                hot_info_list = ['影片名称', '想看数量', '鼎新热度指数']
+                hot_info_list.extend([f"{daily[n]['businessDate']}(销售)" for n in range(4, 6)])
+                hot_info_list.append(f"{daily[6]['businessDate']}(今日)")
+                hot_info_list.extend([f"{daily[n]['businessDate']}(预售)" for n in range(7, 9)])
+                hot_title = f"{','.join(hot_info_list)}"
+                # 生成热度指数
+                hot = int((self.to_number(info['xiaohongshu']) + self.to_number(info['shipinhao']) + self.to_number(
+                    info['douyin']) + self.to_number(info['weibo'])) / 10000)
+                # 准备数据部分
+                data_list = []
+                for n in range(4, 6):
+                    data_list.append({'data': f"{daily[n]['businessDate']}(销售)", 'value': f"{daily[n]['boxRate']}%"})
+                data_list.append({'data': f"{daily[6]['businessDate']}(今日)", 'value': f"{daily[6]['boxRate']}%"})
+                for n in range(7, 9):
+                    data_list.append({'date': f"{daily[n]['businessDate']}(预售)", 'value': f"{daily[n]['boxRate']}%"})
+                # 写入数据
+                MovieHotInfo.objects.create(
+                    movie_zz_code=movie_code,
+                    movie_name=info['movie_name'],
+                    req_date=today_str,
+                    api_response=req,
+                    show_type=show_type,
+                    title=hot_title,
+                    wants=info['wish_num'],
+                    hots=hot,
+                    day1_val=daily[4]['boxRate'],
+                    day2_val=daily[5]['boxRate'],
+                    day3_val=daily[6]['boxRate'],
+                    day4_val=daily[7]['boxRate'],
+                    day5_val=daily[8]['boxRate'],
+                    day_data=json.dumps(data_list),
+                )
+        return movie_name, hot_title, hot_data
 
     def sign(self, code):
         _pre = md5((self.salt + f"movieCode={code}&timeStamp={self.ts}").encode()).hexdigest()
         _sign = md5((_pre + self.salt).encode()).hexdigest()
         return _sign
 
-
-if __name__ == '__main__':
-    md = MovieData()
-    md.get_data('082000852026')
+    @staticmethod
+    def to_number(num_str):
+        if num_str == '--':
+            return 0
+        if '万' in num_str:
+            return int(float(num_str.replace('万', '')) * 10000)
+        if '亿' in num_str:
+            return int(float(num_str.replace('亿', '')) * 100000000)
+        return 0

ai/utils/show_database.py

@@ -43,32 +43,6 @@ class GetData:
         }
         return self.format_to_csv(show_data, show_data_mapping)
 
-    # 获取指定范围的影片销售数据
-    def get_sell_data(self, start_datetime, end_datetime):
-        # print(self.cur.mogrify(GET_SELL_DATA, (start_datetime, end_datetime)))
-        self.cur.execute(GET_SELL_DATA, (start_datetime, end_datetime, start_datetime, end_datetime))
-        sell_data = self.cur.fetchall()
-        sell_data_mapping = {
-            'movie_name': '影片别名',
-            'movie_id': '本地影片id',
-            'show_date': '放映日期',
-            'start_time': '开始时间',
-            'hall_name': '影厅别名',
-            'hall_id': '影厅id',
-            'total_count': '场次观影人数',
-            'total_income': '场次收入',
-            'seat_num': '影厅座位数',
-            'average_price': '平均价',
-            'rate': '上座率（单位%）',
-        }
-        return self.format_to_csv(sell_data, sell_data_mapping)
-
-    # 获取指定范围的影片销售数据
-    def get_total_income(self, start_datetime, end_datetime):
-        print(self.cur.mogrify(GET_TOTAL_INCOME, (start_datetime, end_datetime,start_datetime, end_datetime)))
-        self.cur.execute(GET_TOTAL_INCOME, (start_datetime, end_datetime, start_datetime, end_datetime))
-        return self.cur.fetchone()['total_income']
-
     # 获取指定日期的可放映影片
     def get_available_movie(self, date):
         start_datetime = datetime.datetime.strptime(date, '%Y-%m-%d') + datetime.timedelta(hours=29, minutes=59,
@@ -77,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']}，影片时长：{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']}；"
             for m in available_movie]
         return '\n'.join(available_movie_list)
 
@@ -111,6 +85,86 @@ class GetData:
         new_list = [f"《{m['cinema_movie_alias']}》" for m in new]
         return new_list
 
+    # 为获取影片热度功能提供影院数据
+    def get_movie_info(self, date, show_type='all'):
+        start_datetime = datetime.datetime.strptime(date, '%Y-%m-%d') + datetime.timedelta(hours=29, minutes=59,
+                                                                                           seconds=59)
+        end_datetime = datetime.datetime.strptime(date, '%Y-%m-%d') + datetime.timedelta(hours=6)
+        bl_bool, bl_str, bl_list = get_black_list(self.zz_code)
+        if bl_bool:
+            if show_type == "new":
+                sql = GET_NEW_MOVIE_INFO + bl_str + GET_MOVIE_INFO_END
+                self.cur.execute(sql, (start_datetime, end_datetime, *bl_list))
+                print(self.cur.mogrify(sql, (start_datetime, end_datetime, *bl_list)))
+            elif show_type == "old":
+                sql = GET_OLD_MOVIE_INFO + bl_str + GET_MOVIE_INFO_END
+                self.cur.execute(sql, (end_datetime, start_datetime, *bl_list))
+                print(self.cur.mogrify(sql, (start_datetime, end_datetime, *bl_list)))
+            else:
+                sql = GET_MOVIE_INFO + bl_str + GET_MOVIE_INFO_END
+                self.cur.execute(sql, (start_datetime, end_datetime, *bl_list))
+                print(self.cur.mogrify(sql, (start_datetime, end_datetime, *bl_list)))
+
+        else:
+            if show_type == "new":
+                sql = GET_NEW_MOVIE_INFO + bl_str + GET_MOVIE_INFO_END
+                self.cur.execute(sql, (start_datetime, end_datetime))
+                print(self.cur.mogrify(sql, (start_datetime, end_datetime)))
+            elif show_type == "old":
+                sql = GET_OLD_MOVIE_INFO + bl_str + GET_MOVIE_INFO_END
+                self.cur.execute(sql, (end_datetime, start_datetime))
+                print(self.cur.mogrify(sql, (start_datetime, end_datetime)))
+            else:
+                sql = GET_MOVIE_INFO + bl_str + GET_MOVIE_INFO
+                self.cur.execute(sql, (start_datetime, end_datetime))
+                print(self.cur.mogrify(sql, (start_datetime, end_datetime)))
+        return self.cur.fetchall()
+
+    # 按照场次获取销售数据
+    def get_sell_data(self, start_datetime, end_datetime):
+        # print(self.cur.mogrify(GET_SELL_DATA, (start_datetime, end_datetime)))
+        self.cur.execute(GET_SELL_DATA, (start_datetime, end_datetime, start_datetime, end_datetime))
+        sell_data = self.cur.fetchall()
+        sell_data_mapping = {
+            'movie_name': '影片别名',
+            'movie_id': '本地影片id',
+            'show_date': '放映日期',
+            'start_time': '开始时间',
+            'hall_name': '影厅别名',
+            'hall_id': '影厅id',
+            'total_count': '场次观影人数',
+            'total_income': '场次收入',
+            'seat_num': '影厅座位数',
+            'average_price': '平均价',
+            'rate': '上座率（单位%）',
+        }
+        return self.format_to_csv(sell_data, sell_data_mapping)
+
+    # 按照影片获取销售数据
+    def get_sell_data_by_movie(self, start_datetime, end_datetime):
+        self.cur.execute(GET_SELL_DATA_BY_MOVIE, (start_datetime, end_datetime, start_datetime, end_datetime))
+        sell_data = self.cur.fetchall()
+        print('get_sell_data_by_movie', sell_data)
+        sell_data_mapping = {
+            'movie_alias_name': '影片别名',
+            'movie_id': '本地影片id',
+            'total_count': '观影总人数',
+            'average_show_count': '场均人次',
+            'average_seat_rate': '平均上座率（单位%）',
+            'total_income': '票房总收入',
+            'average_ticket_price': '平均票价',
+            'average_show_ticket_price': '场均收入',
+            'show_count': '总场次数',
+            'show_rate': '排片占比（单位%）',
+        }
+        return self.format_to_csv(sell_data, sell_data_mapping)
+
+    # 获取指定范围的影片销售总额
+    def get_total_income(self, start_datetime, end_datetime):
+        print(self.cur.mogrify(GET_TOTAL_INCOME, (start_datetime, end_datetime, start_datetime, end_datetime)))
+        self.cur.execute(GET_TOTAL_INCOME, (start_datetime, end_datetime, start_datetime, end_datetime))
+        return self.cur.fetchone()['total_income']
+
     # 获取影厅支持的制式
     def get_hall_media_type(self):
         self.cur.execute(GET_HALL_MEDIA_TYPE)
@@ -149,19 +203,3 @@ class GetData:
 
     def get_media_type(self):
         self.cur.execute(GET_HALL_MEDIA_TYPE)
-
-    def get_movie_info(self, date):
-        start_datetime = datetime.datetime.strptime(date, '%Y-%m-%d') + datetime.timedelta(hours=29, minutes=59,
-                                                                                           seconds=59)
-        end_datetime = datetime.datetime.strptime(date, '%Y-%m-%d') + datetime.timedelta(hours=6)
-        bl_bool, bl_str, bl_list = get_black_list(self.zz_code)
-        if bl_bool:
-            sql = GET_MOVIE_INFO_1 + bl_str + GET_MOVIE_INFO_2
-            self.cur.execute(sql, (start_datetime, end_datetime, *bl_list))
-            print(self.cur.mogrify(sql, (start_datetime, end_datetime, *bl_list)))
-        else:
-            sql = GET_MOVIE_INFO_1 + bl_str + GET_MOVIE_INFO_2
-            self.cur.execute(sql, (start_datetime, end_datetime))
-            print(self.cur.mogrify(sql, (start_datetime, end_datetime)))
-        self.cur.execute(sql, (start_datetime, end_datetime, *bl_list))
-        return self.cur.fetchall()

ai/utils/show_func.py

@@ -4,10 +4,10 @@ from ai.models import *
 from ai.utils.show_database import GetData
 from ai.utils.datetime_format import *
 from ai.utils.movie_data import *
+from ai.utils.basic_func import *
 
 
 def get_cinema_show_result_func(_zz_code, _show_date):
-    print("get_cinema_show_result_func")
     cinema = TestCinema.objects.filter(zz_code=_zz_code).first()
     start = datetime.datetime.strftime(datetime.datetime.strptime(_show_date, '%Y-%m-%d') + datetime.timedelta(hours=6),
                                        '%Y-%m-%d %H:%M:%S')
@@ -53,6 +53,7 @@ def get_cinema_show_result_func(_zz_code, _show_date):
                 print(e)
     return show, income
 
+
 # 获取模板日期排片数据
 def get_template_show(cinema, show_date):
     history_start, history_end, target_start, target_end = get_data_datetime(show_date)
@@ -63,10 +64,28 @@ def get_template_show(cinema, show_date):
     temp_end = datetime.datetime.strftime(temp_date_obj + datetime.timedelta(hours=29, minutes=59, seconds=59),
                                           '%Y-%m-%d %H:%M:%S')
     temp_show = data.get_show_data(temp_start, temp_end)
-    print('get_template_show', temp_date, temp_show)
     return temp_show, temp_date
 
 
+# 获取模板日期排片数据, 从昨天开始找一天有排片的数据做为模板，如果没有则返回None
+def get_template_show_tomorrow(cinema):
+    i = -1
+    temp_show = False
+    while temp_show is False or i >= -3:
+        tomorrow = datetime.datetime.combine(datetime.date.today() + datetime.timedelta(days=i), datetime.time())
+        tomorrow_start = datetime.datetime.strftime(tomorrow + datetime.timedelta(hours=6), '%Y-%m-%d %H:%M:%S')
+        tomorrow_end = datetime.datetime.strftime(tomorrow + datetime.timedelta(hours=29, minutes=59, seconds=59),
+                                                  '%Y-%m-%d %H:%M:%S')
+        tomorrow_str = datetime.datetime.strftime(tomorrow, '%Y-%m-%d')
+        data = GetData(cinema)
+        temp_show = data.get_show_data(tomorrow_start, tomorrow_end)
+        if temp_show is not False:
+            return temp_show, tomorrow_str
+        else:
+            i -= 1
+    return temp_show, ''
+
+
 # 获取目标日期排片数据
 def get_target_show(cinema, show_date):
     history_start, history_end, target_start, target_end = get_data_datetime(show_date)
@@ -76,15 +95,40 @@ def get_target_show(cinema, show_date):
         target_date_show = '影厅别名,影厅id,影片别名,本地影片id,语言,放映日期,开始时间,结束时间,片长,场间'
     return target_date_show
 
-def get_all_movie_hot_info(cinema, show_date):
+# 从BI接口获取影片热度信息
+def get_movie_hot_info_from_bi(cinema, show_date):
     data = GetData(cinema)
-    movie = data.get_movie_info(show_date)
+    movie_old = data.get_movie_info(show_date, 'old')
+    movie_new = data.get_movie_info(show_date, 'new')
+
     movie_data = MovieData()
-    movie_info_list = []
-    for m in movie:
-        print(m)
-        if (hot:=movie_data.get_data(m['cinema_movie_num'])) != '':
-            # print('hot', hot)
-            movie_info_list.append(hot)
-    # print(movie_info_list)
-    return movie_info_list
+    movie_name_list = []
+    for m in movie_old:
+        name, title, hot = movie_data.get_data_from_bi(m['cinema_movie_num'], 'old')
+        print('name, title, hot', name, title, hot)
+        if hot != '' and name not in movie_name_list:
+            movie_name_list.append(name)
+    for m in movie_new:
+        name, title, hot = movie_data.get_data_from_bi(m['cinema_movie_num'], 'new')
+        print('name, title, hot', name, title, hot)
+        if hot != '' and name not in movie_name_list:
+            movie_name_list.append(name)
+    return True
+
+# 从本地数据库中获取影片热度信息
+def get_all_movie_hot_info(req_date: str = 'today', show_type:str='old', get_num: int = 10):
+    if req_date == 'today':
+        date_sr = datetime.datetime.today().strftime('%Y-%m-%d')
+    else:
+        date_sr = req_date
+    hot_order_day = get_runtime_val('hot_order_day')
+    hot_data = MovieHotInfo.objects.filter(Q(req_date=date_sr) & Q(show_type=show_type)).order_by(hot_order_day).all()[:get_num]
+    if hot_data:
+        title = f"\t{hot_data[0].title}"
+        data_list = [title]
+    else:
+        return []
+    for hot in hot_data:
+        data_list.append(
+            f"\t《{hot.movie_name}》,{hot.wants},{hot.hots},{','.join([d['value'] for d in json.loads(hot.day_data)])}")
+    return data_list

ai/utils/show_process.py

@@ -1,8 +1,8 @@
 import json
-
 from ai.models import *
 from ai.utils.show_prompt import *
 from ai.utils.show_func import *
+from ai.utils.basic_func import *
 import datetime
 from django_redis import get_redis_connection
 from django.db.models import Q
@@ -10,7 +10,7 @@ import re
 
 
 # ai排片主流程
-def show_main_process(zz_code=None, day_delta=3):
+def show_main_process(zz_code=None, day_delta=None):
     # 查看状态
     redis_conn = get_redis_connection()
     redis_key = f'ai_show{datetime.date.today().strftime("%Y%m%d")}'
@@ -22,18 +22,26 @@ 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)
+    day_delta = int(get_runtime_val('day_delta')) 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
     # 开始处理提示词
     # 生成目标日期
     show_date = datetime.date.strftime(datetime.date.today() + datetime.timedelta(days=day_delta), '%Y-%m-%d')
+    # 获取当前模型
+    model = AiModel.objects.filter(current_used=True).first()
+
+    # 按照影院生成排片
     for cinema in test_cinema_list:
         print(cinema.name)
         show_ai = ShowAI(cinema, show_date)
-        template_show, template_date = get_template_show(cinema, show_date)
+        # template_show, template_date = get_template_show(cinema, show_date)
+        template_show, template_date = get_template_show_tomorrow(cinema)
+        if get_runtime_val('real_target') == 1:
             target_show = get_target_show(cinema, show_date)
+        else:
+            target_show = '影厅别名,影厅id,影片别名,本地影片id,语言,放映日期,开始时间,结束时间,片长,场间'
         prompt = show_ai.general_prompt()
         start = datetime.datetime.now()  # 开始计时
         result, message, tokens = show_ai.get_show_result_ai()
@@ -42,12 +50,17 @@ def show_main_process(zz_code=None, day_delta=3):
         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())
-        print('result_obj:', result_obj, type(result_obj))
+
         # 预测排片数据
         _show = result_obj['show'].replace('```', '').replace('csv', '').strip()
+
         # 预测销售数据
         _sales = str(result_obj['income']).strip()
 
@@ -58,10 +71,11 @@ def show_main_process(zz_code=None, day_delta=3):
                 zz_code=cinema.zz_code,
                 show_date=show_date,
                 is_ai_show=True,
+                ai_model=model.config_name,
                 template=template_show,
                 temp_date=template_date,
                 target=target_show,
-                show=_show,
+                show=data_cleansing(_show),
                 sales=_sales,
                 prompt=prompt,
                 result=result,

ai/utils/show_prompt.py

@@ -1,8 +1,9 @@
 from django.db.models import Q
 from ai.models import PromptTemplate
 from ai.utils.show_database import GetData
-from ai.utils.datetime_format import (get_data_datetime, get_date_type)
-from ai.utils.show_func import get_all_movie_hot_info
+from ai.utils.datetime_format import *
+from ai.utils.show_func import *
+from ai.utils.basic_func import *
 import datetime
 import random
 import json
@@ -12,25 +13,128 @@ from openai import OpenAI
 class ShowAI:
     def __init__(self, cinema_info, show_date):
         self.cinema = cinema_info
-        # self.cinema_name = cinema_info.name
-        # self.cinema_zz_code = cinema_info.zz_code
-        # self.cinema_db_config = json.loads(cinema_info.db_config)
         self.user_config = json.loads(cinema_info.user_config)
         self.show_date = show_date
         self.prompt = ""
+        self.model = get_current_model()
+        self.ai_model_name = self.model.model_name
         self.client = OpenAI(
-            api_key='sk-b7993bf4c8844e79bf84f08f07ff86d9',
-            base_url='https://api.deepseek.com',
+            api_key=self.model.api_key,
+            base_url=self.model.base_url,
         )
 
     # 生成提示词
+    # def general_prompt(self):
+    #     data = GetData(self.cinema)
+    #     # 准备各种日期时间
+    #     history_start, history_end, target_start, target_end = get_data_datetime(self.show_date)
+    #     temp = datetime.datetime.strftime(data.get_template_date(history_start, history_end), '%Y-%m-%d')
+    #     history_date, target_date, target_date_is_holiday, template_date, template_date_is_holiday = get_date_type(
+    #         self.show_date, temp)
+    #     # print(history_start, history_end, target_start, target_end, history_date, show_date, template_date)
+    #
+    #     # 开始处理提示词
+    #     prompt_list = []
+    #     # 处理排片数据
+    #     pre_data = PromptTemplate.objects.filter(
+    #         Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='PreData')).first().prompt_val  # 获取提示词模板
+    #     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}',
+    #                                 '\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)  # 处理参考日期
+    #     pre_data = pre_data.replace('{template_date}', template_date)  # 处理模板日期
+    #     pre_data = pre_data.replace('{target_date}', target_date)  # 处理目标日期
+    #     prompt_list.append(pre_data)
+    #
+    #     # 处理角色说明
+    #     role = PromptTemplate.objects.filter(
+    #         Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='Role')).first().prompt_val  # 获取提示词模板
+    #     role = role.replace('{template_date}', template_date)
+    #     prompt_list.append(role)
+    #
+    #     # 处理影片和影厅信息
+    #     movie_hall = PromptTemplate.objects.filter(
+    #         Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='MovieHall')).first().prompt_val  # 获取提示词模板
+    #     movie_hall = movie_hall.replace('{available_movie}', data.get_available_movie(self.show_date))  # 处理上映影片
+    #     movie_hall = movie_hall.replace('{handle_movie}', data.get_handle_movie(self.show_date))  # 处理可排影片
+    #     movie_hall = movie_hall.replace('{hall_info}', data.get_hall_media_type())  # 处理影厅信息
+    #     free_hall = self.user_config['holiday_date_free_hall'] if target_date_is_holiday else self.user_config[
+    #         'holiday_date_free_hall']
+    #     movie_hall = movie_hall.replace('{free_hall}',
+    #                                     '\t无' if len(free_hall) == 0 else '\n'.join(
+    #                                         f'\t{h}；' for h in free_hall))  # 处理不可排片影厅
+    #     prompt_list.append(movie_hall)
+    #
+    #     # 影片热度和排片占比
+    #     movie_hot = PromptTemplate.objects.filter(
+    #         Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='Hot')).first().prompt_val  # 获取提示词模板
+    #     movie_hot = movie_hot.replace('{movie_hot}', '\n'.join(get_all_movie_hot_info(self.cinema, self.show_date)))
+    #     prompt_list.append(movie_hot)
+    #
+    #     # 处理具体要求
+    #     rules = PromptTemplate.objects.filter(
+    #         Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='Rules')).first().prompt_val  # 获取提示词模板
+    #     rules = rules.replace('{template_date}', template_date)  # 处理模板日期
+    #     # 获取新片
+    #     # new_movie_str_list = []
+    #     # new_movie = data.get_new_movie(self.show_date)
+    #     # for new in new_movie:
+    #     #     new_movie_str_list.append(
+    #     #         f"\t增加{new}的场次，建议排{str(random.randint(2, 4))}场，可根据影片热度和口碑适当增加；")
+    #     # rules = rules.replace('{new_movie}',
+    #     #                       '\t无新片上映' if len(new_movie_str_list) == 0 else '\n'.join(
+    #     #                           new_movie_str_list))  # 处理新片数据
+    #     prompt_list.append(rules)
+    #
+    #     # 处理基本规则
+    #     basic = PromptTemplate.objects.filter(
+    #         Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='Basic')).first().prompt_val  # 获取提示词模板
+    #     start = self.user_config['holiday_date_start'] if target_date_is_holiday else self.user_config[
+    #         'work_date_start']
+    #     end = self.user_config['holiday_date_end'] if target_date_is_holiday else self.user_config['work_date_end']
+    #     basic = basic.replace('{start}', start)
+    #     basic = basic.replace('{end}', end)
+    #     prompt_list.append(basic)
+    #
+    #     # 处理增减场次规则
+    #     add_reduce = PromptTemplate.objects.filter(
+    #         Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='AddReduce')).first().prompt_val  # 获取提示词模板
+    #     prompt_list.append(add_reduce)
+    #
+    #     # 处理黄金时段
+    #     prime_time = PromptTemplate.objects.filter(
+    #         Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='PrimeTime')).first().prompt_val  # 获取提示词模板
+    #     prompt_list.append(prime_time)
+    #
+    #     # 处理暑期档
+    #     summer = PromptTemplate.objects.filter(
+    #         Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='Summer')).first().prompt_val  # 获取提示词模板
+    #     prompt_list.append(summer)
+    #
+    #     # 处理输出目标
+    #     output = PromptTemplate.objects.filter(
+    #         Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='Output')).first().prompt_val  # 获取提示词模板
+    #     prompt_list.append(output)
+    #
+    #     # 打印提示词
+    #     self.prompt = '\n'.join(prompt_list)
+    #     print(self.prompt)
+    #     # 关闭数据库连接
+    #     data.exit()
+    #     return self.prompt
+
+    # 生成提示词（对应V1.2版本）
     def general_prompt(self):
         data = GetData(self.cinema)
         # 准备各种日期时间
-        history_start, history_end, target_start, target_end = get_data_datetime(self.show_date)
-        temp = datetime.datetime.strftime(data.get_template_date(history_start, history_end), '%Y-%m-%d')
+        template_show, template_date = get_template_show_tomorrow(self.cinema)
+        history_start, history_end, target_start, target_end = get_data_datetime_template(self.show_date, template_date)
+        # temp = datetime.datetime.strftime(data.get_template_date(history_start, history_end), '%Y-%m-%d')
         history_date, target_date, target_date_is_holiday, template_date, template_date_is_holiday = get_date_type(
-            self.show_date, temp)
+            self.show_date, template_date)
         # print(history_start, history_end, target_start, target_end, history_date, show_date, template_date)
 
         # 开始处理提示词
@@ -38,13 +142,25 @@ class ShowAI:
         # 处理排片数据
         pre_data = PromptTemplate.objects.filter(
             Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='PreData')).first().prompt_val  # 获取提示词模板
-        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('{template_show}', template_show)  # 处理排片数据
+        pre_data = pre_data.replace('{history_sales}',
+                                    data.get_sell_data_by_movie(history_start, history_end))  # 处理销售数据
+        get_movie_hot_info_from_bi(self.cinema, self.show_date)  # 获取全部影片的热度信息
+        movie_data_old = get_all_movie_hot_info('today', 'old')
+        movie_data_new = get_all_movie_hot_info('today', 'new')
+
+        pre_data = pre_data.replace('{hot_movie_old}',
+                                    '\n'.join(movie_data_old) if len(movie_data_old) == 0 else '\t无')
+        pre_data = pre_data.replace('{hot_movie_new}',
+                                    '\n'.join(movie_data_new) if len(movie_data_new) == 0 else '\t无')
+        if get_runtime_val('real_target') == '1':
             pre_data = pre_data.replace('{target_already_show}',
                                         '\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)  # 处理参考日期
+        else:
+            pre_data = pre_data.replace('{target_already_show}', '无')  # 处理目标日期已排场次数据
+        # pre_data = pre_data.replace('{reference_date}', history_date)  # 处理参考日期
         pre_data = pre_data.replace('{template_date}', template_date)  # 处理模板日期
         pre_data = pre_data.replace('{target_date}', target_date)  # 处理目标日期
         prompt_list.append(pre_data)
@@ -58,7 +174,7 @@ class ShowAI:
         # 处理影片和影厅信息
         movie_hall = PromptTemplate.objects.filter(
             Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='MovieHall')).first().prompt_val  # 获取提示词模板
-        movie_hall = movie_hall.replace('{available_movie}', data.get_available_movie(self.show_date))  # 处理上映影片
+        # movie_hall = movie_hall.replace('{available_movie}', data.get_available_movie(self.show_date))  # 处理上映影片
         movie_hall = movie_hall.replace('{handle_movie}', data.get_handle_movie(self.show_date))  # 处理可排影片
         movie_hall = movie_hall.replace('{hall_info}', data.get_hall_media_type())  # 处理影厅信息
         free_hall = self.user_config['holiday_date_free_hall'] if target_date_is_holiday else self.user_config[
@@ -68,25 +184,11 @@ class ShowAI:
                                             f'\t{h}；' for h in free_hall))  # 处理不可排片影厅
         prompt_list.append(movie_hall)
 
-        # 影片热度和排片占比
-        movie_hot = PromptTemplate.objects.filter(
-            Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='Hot')).first().prompt_val  # 获取提示词模板
-        movie_hot = movie_hot.replace('{movie_hot}', '\n'.join(get_all_movie_hot_info(self.cinema, self.show_date)))
-        prompt_list.append(movie_hot)
-
         # 处理具体要求
         rules = PromptTemplate.objects.filter(
             Q(del_flag=False) & Q(prompt_type='show') & Q(prompt_key='Rules')).first().prompt_val  # 获取提示词模板
         rules = rules.replace('{template_date}', template_date)  # 处理模板日期
-        # 获取新片
-        # new_movie_str_list = []
-        # new_movie = data.get_new_movie(self.show_date)
-        # for new in new_movie:
-        #     new_movie_str_list.append(
-        #         f"\t增加{new}的场次，建议排{str(random.randint(2, 4))}场，可根据影片热度和口碑适当增加；")
-        # rules = rules.replace('{new_movie}',
-        #                       '\t无新片上映' if len(new_movie_str_list) == 0 else '\n'.join(
-        #                           new_movie_str_list))  # 处理新片数据
+        rules = rules.replace('{available_movie}', data.get_available_movie(self.show_date))  # 处理上映影片
         prompt_list.append(rules)
 
         # 处理基本规则
@@ -121,7 +223,7 @@ class ShowAI:
 
         # 打印提示词
         self.prompt = '\n'.join(prompt_list)
-        print(self.prompt)
+        print('开始打印提示词\n', self.prompt)
         # 关闭数据库连接
         data.exit()
         return self.prompt
@@ -130,20 +232,18 @@ class ShowAI:
     def get_show_result_ai(self):
         # 生成提示词
         # self.general_prompt()
-        # 与DeepSeek开始对话
-        print("与DeepSeek开始对话")
+        # 与AI开始对话
+        print(f"与{self.ai_model_name}开始对话")
         response = self.client.chat.completions.create(
-            model='deepseek-v4-pro',
+            model=self.ai_model_name,
             messages=[
-                {'role': 'system', 'content': 'You are a helpful assistant'},
+                {'role': 'system', 'content': '你是中文AI助手。要求：1.所有思考过程必须使用中文。2.所有tool reasoning必须使用中文。3.所有规划必须使用中文。4.即使用户使用英文提问，也必须中文推理。5.最终输出使用中文。'},
                 {'role': 'user', 'content': self.prompt},
             ],
             stream=False,
             reasoning_effort="high",
             extra_body={"thinking": {"type": "enabled"}}
         )
-        # print(dict(response))
-        # print(response.choices[0].message.content)
         res_json = response.model_dump_json()
         res_obj = json.loads(res_json)
         print(res_obj)

ai/utils/sql.py

@@ -91,6 +91,50 @@ GROUP BY md.cinema_movie_show_id, CAST(md.cinema_movie_show_id AS CHAR)
 ORDER BY md.show_date, md.hall_id, md.start_time;
 """
 
+GET_SELL_DATA_BY_MOVIE = """
+SELECT sd.movie_alias_name            movie_alias_name,
+       sd.movie_id                    movie_id,
+       SUM(sd.total_count)            total_count,
+       ROUND(IF(SUM(sd.total_count) = 0, 0.0, SUM(sd.total_count) / COUNT(sd.cinema_movie_show_id)),
+             2)                       average_show_count,
+       ROUND(IF(SUM(sd.total_count) = 0, 0.0, SUM(sd.total_count) / SUM(sd.total_seat)) * 100,
+             2)                       average_seat_rate,
+       SUM(sd.total_income)           total_income,
+       ROUND(IF(SUM(sd.total_income) = 0, 0.0, SUM(sd.total_income) / SUM(sd.total_count)),
+             2)                       average_ticket_price,
+       ROUND(IF(SUM(sd.total_income) = 0, 0.0, SUM(sd.total_income) / COUNT(sd.cinema_movie_show_id)),
+             2)                       average_show_ticket_price,
+       COUNT(sd.cinema_movie_show_id) show_count,
+       ROUND(IF(COUNT(sd.cinema_movie_show_id) = 0, 0.0,
+                COUNT(sd.cinema_movie_show_id) / (SELECT COUNT(ms.cinema_movie_show_id)
+                                                  FROM cinema_movie_show ms
+                                                  WHERE ms.cinema_movie_show_joinflg = 0
+                                                    AND ms.cinema_movie_show_start_time >= % s
+                                                    AND ms.cinema_movie_show_start_time <= % s
+                                                    AND ms.cinema_del_flag = 1
+                                                    AND NOT (ms.cinema_movie_show_sold_num = 0 AND ms.cinema_movie_show_status <> 1))) *
+             100,
+             2)                       show_rate
+FROM (SELECT CAST(ms.cinema_movie_show_id AS CHAR)                             cinema_movie_show_id,
+             mi.cinema_movie_id                                                movie_id,
+             mi.cinema_movie_cn_name                                           cinema_movie_name,
+             mi.cinema_movie_alias                                             movie_alias_name,
+             SUM(IF(sl.cinema_ticket_status <> 3, 1, 0))                       total_count,
+             SUM(IF(sl.cinema_ticket_status <> 3, sl.cinema_ticket_income, 0)) total_income,
+             ms.cinema_hall_seat_num                                           total_seat
+      FROM cinema_movie_show ms
+               LEFT JOIN cinema_sell_log sl ON ms.cinema_movie_show_id = sl.cinema_movie_show_id
+               JOIN cinema_movie_info mi ON ms.cinema_movie_id = mi.cinema_movie_id
+               JOIN cinema_hall_info hi ON ms.cinema_hall_id = hi.cinema_hall_id
+      WHERE ms.cinema_movie_show_joinflg = 0
+        AND ms.cinema_movie_show_start_time >= % s
+        AND ms.cinema_movie_show_start_time <= % s
+        AND ms.cinema_del_flag = 1
+        AND NOT (ms.cinema_movie_show_sold_num = 0 AND ms.cinema_movie_show_status <> 1)
+      GROUP BY ms.cinema_movie_show_id, CAST(ms.cinema_movie_show_id AS CHAR)) sd
+GROUP BY sd.movie_id ORDER BY average_show_ticket_price DESC;
+"""
+
 # 获取指定日期的可放映影片
 GET_AVAILABLE_MOVIE = """
 SELECT cmi.cinema_movie_alias,
@@ -202,7 +246,7 @@ FROM (
 
 
 # 获取指定日期的可用于排片的影片
-GET_MOVIE_INFO_1 = """
+GET_MOVIE_INFO = """
 SELECT cmi.cinema_movie_cn_name,
        cmi.cinema_movie_num
 FROM cinema_movie_info cmi
@@ -210,6 +254,23 @@ WHERE cmi.cinema_movie_start_datetime <= %s
   AND cmi.cinema_movie_end_datetime > %s
   AND cmi.cinema_movie_num NOT IN ("""
 
-GET_MOVIE_INFO_2 = """)
+GET_MOVIE_INFO_END = """)
 GROUP BY cmi.cinema_movie_num;
 """
+
+# 获取指定日期的可用于排片的影片
+GET_NEW_MOVIE_INFO = """
+SELECT cmi.cinema_movie_cn_name,
+       cmi.cinema_movie_num
+FROM cinema_movie_info cmi
+WHERE cmi.cinema_movie_start_datetime <= %s
+  AND cmi.cinema_movie_start_datetime > %s
+  AND cmi.cinema_movie_num NOT IN ("""
+
+GET_OLD_MOVIE_INFO = """
+SELECT cmi.cinema_movie_cn_name,
+       cmi.cinema_movie_num
+FROM cinema_movie_info cmi
+WHERE cmi.cinema_movie_start_datetime <= %s
+AND cmi.cinema_movie_end_datetime > %s
+  AND cmi.cinema_movie_num NOT IN ("""

ai/views.py

@@ -76,7 +76,7 @@ def report(request):
     last_real_data = AiShow.objects.filter(Q(is_ai_show=False) &
                                            Q(zz_code=zz_code) &
                                            Q(show_date=show_date)).order_by('-id').first()
-    ai_dict = csv_to_obj(last_ai_data.show)
+    ai_dict = csv_to_obj(data_cleansing(last_ai_data.show))
     real_dict = csv_to_obj(last_real_data.show)
     temp_dict = csv_to_obj(last_ai_data.template)
     target_dict = csv_to_obj(last_ai_data.target)
@@ -92,6 +92,10 @@ def report(request):
     # 生成时间
     create_time = datetime.datetime.strftime((last_ai_data.created_at + datetime.timedelta(hours=8)),
                                              '%Y-%m-%d %H:%M:%S')
+    if last_ai_data.ai_model in ['lmstudio_qwen3.6_35b', 'company_qwen3.5_35b']:
+        think = json.loads(last_ai_data.message)['choices'][0]['message']['reasoning']
+    else:
+        think = json.loads(last_ai_data.message)['choices'][0]['message']['reasoning_content']
     # 返回
     result_dict = {
         'status': 'success',
@@ -113,6 +117,8 @@ def report(request):
             'ai_count': ai_count,
             'real_count': real_count,
             'result': last_ai_data.result,
+            'think': think,
+            'ai_model': last_ai_data.ai_model,
             'create_time': create_time
         }
     }