# cython:language_level=3
from django.db import models
from django.contrib.auth.models import (
    BaseUserManager, AbstractBaseUser, PermissionsMixin
)
from .models_work import *
from django.utils.translation import gettext as _
from django.core.exceptions import ObjectDoesNotExist


class MyUserManager(BaseUserManager):
    def create_user(self, username, password=None):
        if not username:
            raise ValueError('Users must have an username')

        user = self.model(username=username,
                          )

        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_superuser(self, username, password):
        user = self.create_user(username,
                                password=password,

                                )
        user.is_admin = True
        user.save(using=self._db)
        return user


class UserProfile(AbstractBaseUser, PermissionsMixin):
    username = models.CharField(_('用户名'),
                                unique=True,
                                max_length=50)

    rid = models.CharField(verbose_name='RustDesk ID', max_length=16)
    uuid = models.CharField(verbose_name='uuid', max_length=60)
    autoLogin = models.BooleanField(verbose_name='autoLogin', default=True)
    rtype = models.CharField(verbose_name='rtype', max_length=20)
    deviceInfo = models.TextField(verbose_name=_('登录信息:'), blank=True)

    is_active = models.BooleanField(verbose_name=_('是否激活'), default=True)
    is_admin = models.BooleanField(verbose_name=_('是否管理员'), default=False)

    objects = MyUserManager()

    USERNAME_FIELD = 'username'  # 用作用户名的字段
    REQUIRED_FIELDS = ['password']  # 必须填写的字段

    def get_full_name(self):
        # The user is identified by their email address
        return self.username

    def get_short_name(self):
        # The user is identified by their email address
        return self.username

    def __str__(self):              # __unicode__ on Python 2
        return self.username

    def has_perm(self, perm, obj=None):  # 有没有指定的权限
        "Does the user have a specific permission?"
        # Simplest possible answer: Yes, always
        return True

    def has_module_perms(self, app_label):
        "Does the user have permissions to view the app `app_label`?"
        # Simplest possible answer: Yes, always
        return True

    @classmethod
    def get_uid_by_username(cls, username: str) -> int:
        ''' 
        Get the uid by username.

        Args:
            username (required): str, username

        Returns:
            int: The uid of the user.
        '''
        try:
            user = cls.objects.get(username=username)
            return user.pk
        except ObjectDoesNotExist:
            return None

    @classmethod
    def get_username_by_uid(cls, uid: str) -> str:
        ''' 
        Get the username by uid.

        Args:
            uid (required): str, uid

        Returns:
            str: The username of the user.
        '''
        try:
            user = cls.objects.get(pk=uid)
            return user.username
        except ObjectDoesNotExist:
            return None

    @classmethod
    def check_if_user_exists_by_username(cls, username: str) -> bool:
        ''' 
        Check if a user exists.

        Args:
            username (required): str, username

        Returns:
            bool: True if the user exists, False otherwise.
        '''
        return cls.objects.filter(username=username).exists()

    @classmethod
    def check_if_user_exists_by_uid(cls, uid: str) -> bool:
        ''' 
        Check if a user exists.

        Args:
            uid (required): str, uid

        Returns:
            bool: True if the user exists, False otherwise.
        '''
        return cls.objects.filter(pk=uid).exists()

    @property
    def is_staff(self):
        "Is the user a member of staff?"
        # Simplest possible answer: All admins are staff
        return self.is_admin

    class Meta:

        verbose_name = _("用户")
        verbose_name_plural = _("用户列表")
        permissions = (
            ("view_task", "Can see available tasks"),
            ("change_task_status", "Can change the status of tasks"),
            ("close_task", "Can remove a task by setting its status as closed"),
        )