99 lines
3.5 KiB
Python
99 lines
3.5 KiB
Python
from functools import cached_property
|
|
|
|
from django.contrib.auth.models import User
|
|
from django.db import models
|
|
from django.db.models import Q
|
|
from django.utils import timezone
|
|
|
|
from Main.models.group import Group
|
|
from Main.models.friendship import Friendship
|
|
from Main.models.set import Set
|
|
from Main.models.task import Task
|
|
from Sprint.settings import CONSTS
|
|
from SprintLib.language import languages
|
|
|
|
|
|
class UserInfo(models.Model):
|
|
surname = models.TextField(null=True, blank=True)
|
|
name = models.TextField(null=True, blank=True)
|
|
last_request = models.DateTimeField(default=timezone.now)
|
|
profile_picture_fs_id = models.IntegerField(null=True, blank=True)
|
|
favourite_language_id = models.IntegerField(null=True, blank=True)
|
|
rating = models.IntegerField(default=0)
|
|
user = models.OneToOneField(User, on_delete=models.CASCADE, null=True, blank=True)
|
|
telegram_chat_id = models.TextField(default="", null=True, blank=True)
|
|
vk_user_id = models.IntegerField(null=True, blank=True)
|
|
notification_solution_result = models.BooleanField(default=False)
|
|
notification_friends = models.BooleanField(default=False)
|
|
notification_messages = models.BooleanField(default=False)
|
|
notification_telegram = models.BooleanField(default=False)
|
|
notification_email = models.BooleanField(default=False)
|
|
code = models.IntegerField(null=True, blank=True)
|
|
verified = models.BooleanField(default=False)
|
|
teacher = models.BooleanField(default=False)
|
|
|
|
def __str__(self):
|
|
return self.surname + ' ' + self.name
|
|
|
|
@property
|
|
def groups(self):
|
|
return Group.objects.filter(Q(creator=self.user) | Q(editors__in=self.user.username))
|
|
|
|
@property
|
|
def tasks_solved(self):
|
|
fltr = Task.objects.filter(solution__result=CONSTS["ok_status"], solution__user=self.user).distinct()
|
|
return len(fltr)
|
|
|
|
@property
|
|
def has_favourite_language(self):
|
|
return self.favourite_language_id is not None
|
|
|
|
@property
|
|
def verified_friends(self):
|
|
return User.objects.filter(Q(to_friendship__from_user=self.user) | Q(from_friendship__to_user=self.user))
|
|
|
|
@cached_property
|
|
def friends(self):
|
|
return Friendship.objects.filter(Q(to_user=self.user) | Q(from_user=self.user)).order_by("verified")
|
|
|
|
@property
|
|
def favourite_language(self):
|
|
if not self.has_favourite_language:
|
|
return None
|
|
return languages[self.favourite_language_id]
|
|
|
|
@cached_property
|
|
def available_tasks(self):
|
|
return Task.objects.filter(Q(public=True) | Q(creator=self.user) | Q(editors__contains=[self.user.username]))
|
|
|
|
@property
|
|
def available_sets(self):
|
|
return Set.objects.filter(Q(public=True) | Q(creator=self.user) | Q(editors__contains=[self.user.username])).order_by('name')
|
|
|
|
@property
|
|
def place(self):
|
|
return len(UserInfo.objects.filter(rating__gt=self.rating)) + 1
|
|
|
|
@property
|
|
def activity_status(self):
|
|
if timezone.now() - self.last_request <= timezone.timedelta(minutes=5):
|
|
return CONSTS["online_status"]
|
|
return timezone.datetime.strftime(self.last_request, "%d-%m-%Y %H:%M")
|
|
|
|
@property
|
|
def can_create(self):
|
|
return self.user.is_superuser or self.teacher
|
|
|
|
@property
|
|
def has_profile_pic(self):
|
|
try:
|
|
return self.profile_picture_fs_id is not None
|
|
except ValueError:
|
|
return False
|
|
|
|
@property
|
|
def profile_pic_url(self):
|
|
if self.has_profile_pic:
|
|
return "/image?id=" + str(self.profile_picture_fs_id)
|
|
return "/static/img/user.png"
|