import os
from urllib import parse
from django.db import models
from django.contrib.auth.models import AbstractUser
from imagekit.models import ImageSpecField
from imagekit.processors import ResizeToFill
from imagekit.exceptions import MissingSource
from mosquito.models import Org


def avatar_path(instance, filename):
    return os.path.join('avatar', instance.username, filename)


class User(AbstractUser):
    ROLES = (
        ('superadmin', 'superadmin'),
        ('admin', 'admin'),
        ('staff', 'staff'),
        ('editor', 'editor')
    )
    phone = models.CharField(max_length=11, null=True, blank=True, default='', verbose_name='电话')
    department = models.CharField(max_length=20, null=True, blank=True, default='', verbose_name='部门')
    position = models.CharField(max_length=50, null=True, blank=True, default='', verbose_name='职位')
    introduction = models.TextField(null=True, blank=True, verbose_name='简介')
    # org = models.CharField(max_length=100, null=True, blank=True, verbose_name='组织')
    org = models.ForeignKey(Org, verbose_name='所在组织', on_delete='PROTECT')
    parent_org = models.PositiveIntegerField(verbose_name="父组织ID", null=True, blank=True)
    role = models.CharField(max_length=20, default='admin', choices=ROLES, verbose_name='角色')
    avatar = models.ImageField(upload_to=avatar_path,
                               default='https://wpimg.wallstcn.com/f778738c-e4f8-4870-b634-56703b4acafe.gif', null=True, blank=True)
    avatar_thumbnail = ImageSpecField(source='avatar',
                                      processors=[ResizeToFill(100, 100)],
                                      format='JPEG',
                                      options={'quality': 100})

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

    def avatar_url(self):
        try:
            avatar_thumbnail_url = self.avatar_thumbnail.url
        except MissingSource:
            avatar_thumbnail_url = ''
        return avatar_thumbnail_url

    def get_image_url(self):
        return parse.unquote(self.avatar.url)

    def __str__(self):
        return self.username

    def get_child(self):
        print(self.org)
        return [u.id for u in User.objects.filter(parent_org=self.org)]