dingxin_toolbox_drf/ai/models.py

from django.db import models


# Create your models here.
class TestCinema(models.Model):
    id = models.AutoField(primary_key=True)
    name = models.CharField(max_length=50)
    zz_code = models.CharField(max_length=50)
    db_config = models.TextField()
    user_config = models.TextField()
    is_active = models.BooleanField(default=True)

    def __str__(self):
        return self.name

    class Meta:
        verbose_name = '测试影院'
        verbose_name_plural = '测试影院'
        db_table = 'ai_show_cinema'


class PromptTemplate(models.Model):
    id = models.AutoField(primary_key=True)
    prompt_type = models.CharField(max_length=50)
    prompt_key = models.CharField(max_length=50)
    prompt_val = models.TextField()
    del_flag = models.BooleanField(default=False)

    def __str__(self):
        return self.prompt_key

    class Meta:
        verbose_name = '提示词'
        verbose_name_plural = '提示词'
        db_table = 'ai_prompt_template'


class AiShow(models.Model):
    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='')
    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)
    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='')

    def __str__(self):
        return self.cinema

    class Meta:
        verbose_name = 'AI排片数据'
        verbose_name_plural = 'AI排片数据'
        db_table = 'ai_show_result'


class CinemaHall(models.Model):
    id = models.AutoField(primary_key=True)
    cinema_code = models.CharField(max_length=50)
    hall_id = models.CharField(max_length=50)
    hall_name = models.CharField(max_length=50)
    hall_type = models.CharField(max_length=500 , default='')

    def __str__(self):
        return self.hall_name

    class Meta:
        verbose_name = '影厅数据'
        verbose_name_plural = '影厅数据'
        db_table = 'ai_show_hall'