from django.contrib.auth.models import AbstractBaseUser, PermissionsMixin
from django.db import models
from django.utils import timezone
from django.contrib.auth.base_user import BaseUserManager

from apps.common.models import BaseModel


class UserManager(BaseUserManager):
    def create_user(self, username, password=None, **extra_fields):
        if not username:
            raise ValueError('用户名不能为空')
        user = self.model(username=username, **extra_fields)
        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_superuser(self, username, password=None, **extra_fields):
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_superuser', True)
        extra_fields.setdefault('status', 1)
        return self.create_user(username, password, **extra_fields)


class User(AbstractBaseUser, PermissionsMixin, BaseModel):
    """用户表"""
    GENDER_CHOICES = [
        (0, '未知'),
        (1, '男'),
        (2, '女'),
    ]
    STATUS_CHOICES = [
        (0, '禁用'),
        (1, '正常'),
    ]

    id = models.BigAutoField(primary_key=True)
    username = models.CharField('用户名', max_length=50, unique=True)
    password = models.CharField('密码', max_length=255)
    real_name = models.CharField('真实姓名', max_length=50, blank=True)
    phone = models.CharField('手机号', max_length=20, unique=True, blank=True)
    avatar = models.CharField('头像', max_length=255, blank=True)
    gender = models.SmallIntegerField('性别', choices=GENDER_CHOICES, default=0)
    role_type = models.CharField('主角色', max_length=30, blank=True)
    institution = models.ForeignKey(
        'user.Institution', on_delete=models.SET_NULL,
        null=True, blank=True, verbose_name='所属机构'
    )
    department = models.ForeignKey(
        'user.Department', on_delete=models.SET_NULL,
        null=True, blank=True, verbose_name='所属科室'
    )
    title_name = models.CharField('职称', max_length=50, blank=True)
    major = models.CharField('专业', max_length=100, blank=True)
    training_stage = models.CharField('培训阶段', max_length=50, blank=True)
    learning_target = models.CharField('学习目标', max_length=255, blank=True)
    competency_profile = models.JSONField('能力画像', default=dict, blank=True)
    weak_dimensions = models.JSONField('薄弱项', default=list, blank=True)
    strong_dimensions = models.JSONField('优势项', default=list, blank=True)
    ai_preference = models.JSONField('AI训练偏好', default=dict, blank=True)
    total_training_count = models.IntegerField('总训练次数', default=0)
    total_case_count = models.IntegerField('完成病例数', default=0)
    current_level = models.CharField('当前能力等级', max_length=30, blank=True)
    status = models.SmallIntegerField('状态', choices=STATUS_CHOICES, default=1)
    last_login_time = models.DateTimeField('最后登录', null=True, blank=True)

    # Django required fields
    is_staff = models.BooleanField('staff status', default=False)
    is_active = models.BooleanField('active', default=True)
    date_joined = models.DateTimeField('date joined', default=timezone.now)

    objects = UserManager()

    USERNAME_FIELD = 'username'
    REQUIRED_FIELDS = []

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

    def __str__(self):
        return self.username


class Role(BaseModel):
    """角色表"""
    id = models.BigAutoField(primary_key=True)
    role_code = models.CharField('角色编码', max_length=50, unique=True)
    role_name = models.CharField('角色名称', max_length=50)

    class Meta:
        db_table = 'role'
        verbose_name = '角色'
        verbose_name_plural = '角色'

    def __str__(self):
        return self.role_name


class TeacherStudentRelation(BaseModel):
    """师生关系表"""
    STATUS_CHOICES = [
        (0, '已结束'),
        (1, '进行中'),
    ]

    id = models.BigAutoField(primary_key=True)
    teacher = models.ForeignKey(
        User, on_delete=models.CASCADE,
        related_name='teacher_relations', verbose_name='带教老师'
    )
    student = models.ForeignKey(
        User, on_delete=models.CASCADE,
        related_name='student_relations', verbose_name='学员'
    )
    relation_type = models.CharField('关系类型', max_length=30, blank=True)
    start_time = models.DateTimeField('开始时间', null=True, blank=True)
    end_time = models.DateTimeField('结束时间', null=True, blank=True)
    status = models.SmallIntegerField('状态', choices=STATUS_CHOICES, default=1)

    class Meta:
        db_table = 'teacher_student_relation'
        verbose_name = '师生关系'
        verbose_name_plural = '师生关系'

    def __str__(self):
        return f"{self.teacher.real_name or self.teacher.username} -> {self.student.real_name or self.student.username}"


class Institution(BaseModel):
    """医疗机构表"""

    id = models.BigAutoField(primary_key=True)
    code = models.CharField('机构编码', max_length=100, unique=True)
    name = models.CharField('名称', max_length=255)
    type = models.CharField('类型', max_length=30, default='hospital', blank=True)
    level = models.CharField('等级', max_length=30, blank=True)
    province = models.CharField('省份', max_length=50, blank=True)
    city = models.CharField('城市', max_length=50, blank=True)

    class Meta:
        db_table = 'institution'
        verbose_name = '机构'
        verbose_name_plural = '机构'

    def __str__(self):
        return self.name


class Department(BaseModel):
    """科室表"""
    id = models.BigAutoField(primary_key=True)
    institution = models.ForeignKey(
        Institution, on_delete=models.CASCADE,
        verbose_name='所属机构'
    )
    name = models.CharField('科室名称', max_length=100)
    category = models.CharField('科室分类', max_length=50, blank=True)

    class Meta:
        db_table = 'department'
        verbose_name = '科室'
        verbose_name_plural = '科室'

    def __str__(self):
        return self.name