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)
    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)

    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)

    def __str__(self):
        return self.hall_name

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