fzjg_local/fzjgact/huodong/models.py

from django.db import models
from django.utils import timezone


class Branch(models.Model):
    CATEGORY_CHOICES = (
        ('A型', 'A型'),
        ('B型', 'B型'),
        ('C型', 'C型'),
        ('不适用', '不适用'),
    )
    name = models.CharField(max_length=255, unique=True, verbose_name='分支机构名称')
    location = models.CharField(max_length=255, verbose_name='所在省份')
    contact_info = models.CharField(max_length=255, verbose_name='主要联系人')
    description = models.TextField(blank=True, verbose_name='备注')
    background_color = models.CharField(max_length=7, default='#EFF6FF', verbose_name='背景色',
                                        help_text='使用#RRGGBB格式的颜色代码')
    category = models.CharField(max_length=10, choices=CATEGORY_CHOICES, default='C型', verbose_name='分类')
    is_mature = models.BooleanField(default=False, verbose_name='是否成熟')
    

    def __str__(self):
        return f'{self.name} 💼' if self.is_mature else self.name

    class Meta:
        verbose_name = '分支机构'
        verbose_name_plural = '分支机构（基础信息）'


class Contact(models.Model):
    branch = models.ForeignKey(Branch, on_delete=models.CASCADE, verbose_name='分支机构')
    CATEGORY_CHOICES = [
        ('机房/设备间巡检人', '机房/设备间巡检人'),
        ('信息安全联系人', '信息安全联系人'),
        ('兼岗', '兼岗'),
        ('安全员', '安全员')
        # 可以添加更多类别
    ]
    # 修改为支持多选的 CharField
    category = models.CharField(
        max_length=255,  # 增大长度（原50可能不足）
        choices=CATEGORY_CHOICES,
        verbose_name='联系人分类',
        help_text='按住 Ctrl/Command 键多选（值将以逗号分隔存储）'
    )
    name = models.CharField(max_length=255, verbose_name='姓名')
    phone = models.CharField(max_length=20, verbose_name='电话')
    email = models.EmailField(blank=True, verbose_name='邮箱，可不填')
    description = models.TextField(blank=True, verbose_name='描述，可不填')

    def __str__(self):
        return self.name

    class Meta:
        verbose_name = '联系人群'
        verbose_name_plural = '联系人群'


class Activity(models.Model):
    branch = models.ForeignKey(Branch, on_delete=models.CASCADE, verbose_name='分支机构')
    name = models.CharField(max_length=255, verbose_name='活动名称')
    scope = models.CharField(max_length=255, choices=(
        ('新建', '新建'),
        ('搬迁', '搬迁'),
        ('原址装修', '原址装修'),
        ('撤销', '撤销'),
		('其他技术问题', '其他技术问题')
		
    ), verbose_name='活动类型')
    start_time = models.DateField(verbose_name='开始日期')
    end_time = models.DateField(blank=True, null=True, verbose_name='结束日期')  # 可以为空，表示活动尚未结束
    location = models.CharField(max_length=255, verbose_name='所在地点')
    description = models.TextField(verbose_name='其它内容')

    def __str__(self):
        return self.name

    class Meta:
        verbose_name = '运营活动内容'
        verbose_name_plural = '运营活动内容（新建搬迁装修和技术）'


class EquipmentImage(models.Model):
    branch = models.ForeignKey(Branch, related_name='equipment_images', on_delete=models.CASCADE)
    image = models.ImageField(upload_to='equipment_room_images/')
    uploaded_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return f"设备间图片 {self.id} - {self.branch.name}"
        
    class Meta:
        verbose_name = '设备间图'
        verbose_name_plural = '设备间图'


# 图纸的类
class Drawing(models.Model):
    branch = models.ForeignKey(Branch, related_name='drawings', on_delete=models.CASCADE)
    image = models.ImageField(upload_to='drawings/')
    uploaded_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return f"图纸 {self.id} - {self.branch.name}"
        
    class Meta:
        verbose_name = '图纸'
        verbose_name_plural = '图纸'


# 公共电子屏
class PublicScreen(models.Model):
    SCREEN_TYPES = (
        ('marquee', '跑马灯'),
        ('advertisement', '广告屏'),
        ('information', '信息发布屏'),
    )
    branch = models.ForeignKey(Branch, on_delete=models.CASCADE, related_name='public_screens')
    image = models.ImageField(upload_to='public_screen_images/', null=True, blank=True)
    screen_type = models.CharField(max_length=20, choices=SCREEN_TYPES, verbose_name='功能类型', null=True, blank=True)
    description = models.TextField(blank=True, null=True, verbose_name='功能描述')
    last_drill = models.ForeignKey('Event', on_delete=models.SET_NULL, blank=True, null=True, related_name='public_screens', verbose_name='最后演练事件')
    created_at = models.DateTimeField(default=timezone.now)
    updated_at = models.DateTimeField(auto_now=True)

    def __str__(self):
        return f'{self.branch.name} - {self.get_screen_type_display()} {self.id}'
    
    class Meta:
        verbose_name = '公共电子屏'
        verbose_name_plural = '公共电子屏'


class Event(models.Model):
    branches = models.ManyToManyField(Branch, related_name='events', verbose_name='分支机构')
    name = models.CharField(max_length=255, verbose_name='事件名称')
    start_time = models.DateField(verbose_name='开始时间')
    end_time = models.DateField(blank=True, null=True, verbose_name='结束时间')  # 可以为空，表示活动尚未结束
    description = models.TextField(verbose_name='事件描述')

    def __str__(self):
        return self.name

    class Meta:
        verbose_name = '运营事件'
        verbose_name_plural = '运营事件（其它）'


class VideoTerminal(models.Model):
    TERMINAL_TYPES = (
        ('polycom', '宝利通终端'),
        ('zte', '中兴终端'),
        ('logitech', '罗技摄像头'),
        ('laptop_tv', '笔记本加电视'),
        ('laptop_projector', '笔记本加投影仪'),
        ('other', '其它'),
    )
    branch = models.ForeignKey(Branch, on_delete=models.CASCADE, related_name='video_terminals', verbose_name='分支机构')
    terminal_type = models.CharField(max_length=20, choices=TERMINAL_TYPES, verbose_name='设备类型')
    description = models.TextField(blank=True, verbose_name='设备描述')
    created_at = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
    updated_at = models.DateTimeField(auto_now=True, verbose_name='更新时间')

    def __str__(self):
        return f"{self.branch.name} - {self.get_terminal_type_display()}"

    class Meta:
        verbose_name = '视频设备终端'
        verbose_name_plural = '视频设备终端'


class Evaluation(models.Model):
    activity = models.ForeignKey(Activity, on_delete=models.CASCADE)
    branch = models.ForeignKey(Branch, on_delete=models.CASCADE)
    score = models.DecimalField(max_digits=4, decimal_places=2)
    comment = models.TextField()
    file_path = models.CharField(max_length=255, blank=True, null=True)
    status = models.CharField(max_length=20,
                              choices=(('pending', '待审核'), ('approved', '已通过'), ('rejected', '已拒绝')),
                              default='pending')

    def __str__(self):
        return f"{self.activity.name} - {self.branch.name}"