models.py
2.67 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
import re
from os import path
from django.db import models
from django.core import validators
from django.core.exceptions import ValidationError
from django.utils.translation import ugettext_lazy as _
from django.contrib.auth.models import AbstractBaseUser, UserManager, PermissionsMixin
from django.contrib.staticfiles.templatetags.staticfiles import static
from log.models import Log
def validate_img_extension(value):
valid_formats = ['image/jpeg','image/x-citrix-jpeg','image/png','image/x-citrix-png','image/x-png']
if hasattr(value.file, 'content_type'):
if not value.file.content_type in valid_formats:
raise ValidationError(_('File not supported.'))
class User(AbstractBaseUser, PermissionsMixin):
email = models.EmailField(_('Mail'), unique = True, validators = [
validators.RegexValidator(
re.compile('^[\w.@+-]+$'),
_('Type a valid email. This fields should only contain letters, numbers and the characteres: @/./+/-/_ .')
, 'invalid'
)
], help_text = _('Your email address that will be used to access the platform'))
username = models.CharField(_('Name'), max_length = 100)
last_name = models.CharField(_('Last Name'), max_length = 100)
social_name = models.CharField(_('Social Name'), max_length = 100, blank = True, null = True)
description = models.TextField(_('Description'), blank = True)
image = models.ImageField(verbose_name = _('Photo'), null=True, blank = True, upload_to = 'users/', validators = [validate_img_extension])
date_created = models.DateTimeField(_('Create Date'), auto_now_add = True)
last_update = models.DateTimeField(_('Last Update'), auto_now = True)
show_email = models.IntegerField(_('Show email?'), null = True, blank = True, choices = ((1, _('Allow everyone to see my address')), (2, _('Only classmates can see my address')), (3, _('Nobody can see my address'))), default = 1)
is_staff = models.BooleanField(_('Administrator'), default = False)
is_active = models.BooleanField(_('Active'), default = True)
USERNAME_FIELD = 'email'
REQUIRED_FIELDS = ['username', 'last_name']
objects = UserManager()
class Meta:
verbose_name = _('User')
verbose_name_plural = _('Users')
def __str__(self):
return self.social_name or (self.username + " " + self.last_name)
def get_short_name(self):
return str(self)
@property
def image_url(self):
if self.image and hasattr(self.image, 'url'):
if path.exists(self.image.path):
return self.image.url
return static('img/no_image.jpg')
def get_items(self):
data = Log.objects.filter(user_id = self.id)
return data
def is_admin(self):
if self.is_staff:
return _('Yes')
return _('Is not an admin')