from django.contrib.auth.models import AbstractUser
from django.db import models
from department.models import Department

# Create your models here.


# class Department(models.Model):
#     department_name = models.CharField(max_length=128, verbose_name='部门名称')
#     # parent_id = models.IntegerField(verbose_name='上级部门id', null=True, blank=True)
#     # status = models.IntegerField(verbose_name='状态', default=1)
#
#     class Meta:
#         db_table = 'user_department'


class UserProfile(AbstractUser):
    GENDER = ((0, '女'), (1, '男'))
    organization = models.CharField(max_length=256, verbose_name='组织', null=True, blank=True)
    gender = models.IntegerField(choices=GENDER, default=1, verbose_name='性别')
    phone_number = models.CharField(max_length=12, verbose_name='手机号', null=True, blank=True)
    head_sculpture = models.CharField(max_length=512, verbose_name='头像', blank=True, null=True)
    status = models.IntegerField(verbose_name='状态', default=1)
    recorder_number = models.CharField(max_length=256, verbose_name='执法记录仪编号', null=True, blank=True)
    department = models.ForeignKey(Department, on_delete=models.CASCADE, null=True)

    class Meta:
        db_table = 'tp_user'