from django.db import models
from django.contrib.auth.models import AbstractUser, Group


# Create your models here.
class User(AbstractUser):
    real_name = models.CharField(max_length=150, verbose_name='姓名')
    staff_id = models.CharField(max_length=150, verbose_name='工号')
    department = models.ForeignKey('Department', on_delete=models.SET_NULL, null=True, blank=True)

    class Meta:
        verbose_name = '用户'
        verbose_name_plural = '用户'

    def __str__(self):
        return self.real_name


class MyGroup(Group):
    class Meta:
        verbose_name = '权限组'
        verbose_name_plural = '权限组'


class Department(models.Model):
    name = models.CharField(max_length=50, verbose_name='部门名称')

    def __str__(self):
        return self.name

    class Meta:
        verbose_name = '部门'
        verbose_name_plural = '部门'


class MarkValue(models.Model):
    mark_value = models.DecimalField('评价等级考核系数', max_digits=3, decimal_places=2)

    def __str__(self):
        return str(self.mark_value)
    class Meta:
        verbose_name = '评价等级考核系数'
        verbose_name_plural = verbose_name


class QualityMark(models.Model):
    mark_name = models.CharField('评价等级定义', max_length=10)
    mark_value = models.ForeignKey('MarkValue', on_delete=models.CASCADE)

    def __str__(self):
        return self.mark_name
    class Meta:
        verbose_name = '评价等级定义'
        verbose_name_plural = verbose_name


class TaskProperty(models.Model):
    own_department = models.ForeignKey('Department', on_delete=models.SET_NULL, null=True, blank=True, verbose_name='所属部门')
    task_property = models.CharField(max_length=50, verbose_name='任务属性')

    class Meta:
        verbose_name = '任务属性'
        verbose_name_plural = '任务属性'

    def __str__(self):
        return self.task_property