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


class MyUser(AbstractUser):
    """
    使用AbstractUser可以对User进行扩展使用，添加用户自定义的属性
    """

    phone = models.CharField(
        verbose_name="手机号码",
        unique=True,
        null=True,
        max_length=11,
        help_text="手机号码",
    )
    show_hosts = models.BooleanField(
        verbose_name="是否显示Hosts相关的信息",
        default=False,
        help_text="是否显示Hosts相关的信息",
    )
    name = models.CharField(
        verbose_name="姓名",
        max_length=40,
        blank=True,
        null=True,
        help_text="姓名",
    )
    STATUS_CHOICES = (
        ('pending', '待审核'),
        ('approved', '已通过'),
        ('rejected', '未通过'),
    )
    status = models.CharField(
        verbose_name="状态",
        max_length=20,
        choices=STATUS_CHOICES,
        default='approved',
        help_text="用户状态",
    )

    class Meta(AbstractUser.Meta):
        pass