269 lines
12 KiB
Python
269 lines
12 KiB
Python
from operator import mul
|
|
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import User
|
|
from django.core.validators import RegexValidator
|
|
from django.db import models
|
|
from django.db.models import Max, CASCADE
|
|
from django.urls import reverse
|
|
from django.utils.functional import cached_property
|
|
from django.utils.timezone import now
|
|
from django.utils.translation import gettext_lazy as _
|
|
from fernet_fields import EncryptedCharField
|
|
from sortedm2m.fields import SortedManyToManyField
|
|
|
|
from judge.models.choices import ACE_THEMES, MATH_ENGINES_CHOICES, TIMEZONE
|
|
from judge.models.runtime import Language
|
|
from judge.ratings import rating_class
|
|
|
|
__all__ = ['Organization', 'Profile', 'OrganizationRequest', 'Friend']
|
|
|
|
|
|
class EncryptedNullCharField(EncryptedCharField):
|
|
def get_prep_value(self, value):
|
|
if not value:
|
|
return None
|
|
return super(EncryptedNullCharField, self).get_prep_value(value)
|
|
|
|
|
|
class Organization(models.Model):
|
|
name = models.CharField(max_length=128, verbose_name=_('organization title'))
|
|
slug = models.SlugField(max_length=128, verbose_name=_('organization slug'),
|
|
help_text=_('Organization name shown in URL'))
|
|
short_name = models.CharField(max_length=20, verbose_name=_('short name'),
|
|
help_text=_('Displayed beside user name during contests'))
|
|
about = models.TextField(verbose_name=_('organization description'))
|
|
registrant = models.ForeignKey('Profile', verbose_name=_('registrant'), on_delete=models.CASCADE,
|
|
related_name='registrant+', help_text=_('User who registered this organization'))
|
|
admins = models.ManyToManyField('Profile', verbose_name=_('administrators'), related_name='admin_of',
|
|
help_text=_('Those who can edit this organization'))
|
|
creation_date = models.DateTimeField(verbose_name=_('creation date'), auto_now_add=True)
|
|
is_open = models.BooleanField(verbose_name=_('is open organization?'),
|
|
help_text=_('Allow joining organization'), default=True)
|
|
slots = models.IntegerField(verbose_name=_('maximum size'), null=True, blank=True,
|
|
help_text=_('Maximum amount of users in this organization, '
|
|
'only applicable to private organizations'))
|
|
access_code = models.CharField(max_length=7, help_text=_('Student access code'),
|
|
verbose_name=_('access code'), null=True, blank=True)
|
|
logo_override_image = models.CharField(verbose_name=_('Logo override image'), default='', max_length=150,
|
|
blank=True,
|
|
help_text=_('This image will replace the default site logo for users '
|
|
'viewing the organization.'))
|
|
|
|
def __contains__(self, item):
|
|
if isinstance(item, int):
|
|
return self.members.filter(id=item).exists()
|
|
elif isinstance(item, Profile):
|
|
return self.members.filter(id=item.id).exists()
|
|
else:
|
|
raise TypeError('Organization membership test must be Profile or primany key')
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('organization_home', args=(self.id, self.slug))
|
|
|
|
def get_users_url(self):
|
|
return reverse('organization_users', args=(self.id, self.slug))
|
|
|
|
class Meta:
|
|
ordering = ['name']
|
|
permissions = (
|
|
('organization_admin', 'Administer organizations'),
|
|
('edit_all_organization', 'Edit all organizations'),
|
|
)
|
|
verbose_name = _('organization')
|
|
verbose_name_plural = _('organizations')
|
|
|
|
|
|
class Profile(models.Model):
|
|
user = models.OneToOneField(User, verbose_name=_('user associated'), on_delete=models.CASCADE)
|
|
about = models.TextField(verbose_name=_('self-description'), null=True, blank=True)
|
|
timezone = models.CharField(max_length=50, verbose_name=_('location'), choices=TIMEZONE,
|
|
default=settings.DEFAULT_USER_TIME_ZONE)
|
|
language = models.ForeignKey('Language', verbose_name=_('preferred language'), on_delete=models.SET_DEFAULT,
|
|
default=Language.get_default_language_pk)
|
|
points = models.FloatField(default=0, db_index=True)
|
|
performance_points = models.FloatField(default=0, db_index=True)
|
|
problem_count = models.IntegerField(default=0, db_index=True)
|
|
ace_theme = models.CharField(max_length=30, choices=ACE_THEMES, default='github')
|
|
last_access = models.DateTimeField(verbose_name=_('last access time'), default=now)
|
|
ip = models.GenericIPAddressField(verbose_name=_('last IP'), blank=True, null=True)
|
|
organizations = SortedManyToManyField(Organization, verbose_name=_('organization'), blank=True,
|
|
related_name='members', related_query_name='member')
|
|
display_rank = models.CharField(max_length=10, default='user', verbose_name=_('display rank'),
|
|
choices=(('user', 'Normal User'), ('setter', 'Problem Setter'), ('admin', 'Admin')))
|
|
mute = models.BooleanField(verbose_name=_('comment mute'), help_text=_('Some users are at their best when silent.'),
|
|
default=False)
|
|
is_unlisted = models.BooleanField(verbose_name=_('unlisted user'), help_text=_('User will not be ranked.'),
|
|
default=False)
|
|
rating = models.IntegerField(null=True, default=None)
|
|
user_script = models.TextField(verbose_name=_('user script'), default='', blank=True, max_length=65536,
|
|
help_text=_('User-defined JavaScript for site customization.'))
|
|
current_contest = models.OneToOneField('ContestParticipation', verbose_name=_('current contest'),
|
|
null=True, blank=True, related_name='+', on_delete=models.SET_NULL)
|
|
math_engine = models.CharField(verbose_name=_('math engine'), choices=MATH_ENGINES_CHOICES, max_length=4,
|
|
default=settings.MATHOID_DEFAULT_TYPE,
|
|
help_text=_('the rendering engine used to render math'))
|
|
is_totp_enabled = models.BooleanField(verbose_name=_('2FA enabled'), default=False,
|
|
help_text=_('check to enable TOTP-based two factor authentication'))
|
|
totp_key = EncryptedNullCharField(max_length=32, null=True, blank=True, verbose_name=_('TOTP key'),
|
|
help_text=_('32 character base32-encoded key for TOTP'),
|
|
validators=[RegexValidator('^$|^[A-Z2-7]{32}$',
|
|
_('TOTP key must be empty or base32'))])
|
|
notes = models.TextField(verbose_name=_('internal notes'), null=True, blank=True,
|
|
help_text=_('Notes for administrators regarding this user.'))
|
|
|
|
@cached_property
|
|
def organization(self):
|
|
# We do this to take advantage of prefetch_related
|
|
orgs = self.organizations.all()
|
|
return orgs[0] if orgs else None
|
|
|
|
@cached_property
|
|
def username(self):
|
|
return self.user.username
|
|
|
|
@cached_property
|
|
def count_unseen_notifications(self):
|
|
query = {
|
|
'read': False,
|
|
}
|
|
return self.notifications.filter(**query).count()
|
|
|
|
_pp_table = [pow(settings.DMOJ_PP_STEP, i) for i in range(settings.DMOJ_PP_ENTRIES)]
|
|
|
|
def calculate_points(self, table=_pp_table):
|
|
from judge.models import Problem
|
|
public_problems = Problem.get_public_problems()
|
|
data = (
|
|
public_problems.filter(submission__user=self, submission__points__isnull=False)
|
|
.annotate(max_points=Max('submission__points')).order_by('-max_points')
|
|
.values_list('max_points', flat=True).filter(max_points__gt=0)
|
|
)
|
|
extradata = (
|
|
public_problems.filter(submission__user=self, submission__result='AC').values('id').distinct().count()
|
|
)
|
|
bonus_function = settings.DMOJ_PP_BONUS_FUNCTION
|
|
points = sum(data)
|
|
problems = len(data)
|
|
entries = min(len(data), len(table))
|
|
pp = sum(map(mul, table[:entries], data[:entries])) + bonus_function(extradata)
|
|
if self.points != points or problems != self.problem_count or self.performance_points != pp:
|
|
self.points = points
|
|
self.problem_count = problems
|
|
self.performance_points = pp
|
|
self.save(update_fields=['points', 'problem_count', 'performance_points'])
|
|
return points
|
|
|
|
calculate_points.alters_data = True
|
|
|
|
def remove_contest(self):
|
|
self.current_contest = None
|
|
self.save()
|
|
|
|
remove_contest.alters_data = True
|
|
|
|
def update_contest(self):
|
|
contest = self.current_contest
|
|
if contest is not None and (contest.ended or not contest.contest.is_accessible_by(self.user)):
|
|
self.remove_contest()
|
|
|
|
update_contest.alters_data = True
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('user_page', args=(self.user.username,))
|
|
|
|
def __str__(self):
|
|
return self.user.username
|
|
|
|
@classmethod
|
|
def get_user_css_class(cls, display_rank, rating, rating_colors=settings.DMOJ_RATING_COLORS):
|
|
if rating_colors:
|
|
return 'rating %s %s' % (rating_class(rating) if rating is not None else 'rate-none', display_rank)
|
|
return display_rank
|
|
|
|
@cached_property
|
|
def css_class(self):
|
|
return self.get_user_css_class(self.display_rank, self.rating)
|
|
|
|
def get_friends(self): #list of usernames, including you
|
|
friend_obj = self.following_users.all()
|
|
ret = set()
|
|
|
|
if (friend_obj):
|
|
ret = set(friend.username for friend in friend_obj[0].users.all())
|
|
|
|
ret.add(self.username)
|
|
return ret
|
|
|
|
class Meta:
|
|
permissions = (
|
|
('test_site', 'Shows in-progress development stuff'),
|
|
('totp', 'Edit TOTP settings'),
|
|
)
|
|
verbose_name = _('user profile')
|
|
verbose_name_plural = _('user profiles')
|
|
|
|
|
|
class OrganizationRequest(models.Model):
|
|
user = models.ForeignKey(Profile, verbose_name=_('user'), related_name='requests', on_delete=models.CASCADE)
|
|
organization = models.ForeignKey(Organization, verbose_name=_('organization'), related_name='requests',
|
|
on_delete=models.CASCADE)
|
|
time = models.DateTimeField(verbose_name=_('request time'), auto_now_add=True)
|
|
state = models.CharField(max_length=1, verbose_name=_('state'), choices=(
|
|
('P', 'Pending'),
|
|
('A', 'Approved'),
|
|
('R', 'Rejected'),
|
|
))
|
|
reason = models.TextField(verbose_name=_('reason'))
|
|
|
|
class Meta:
|
|
verbose_name = _('organization join request')
|
|
verbose_name_plural = _('organization join requests')
|
|
|
|
|
|
class Friend(models.Model):
|
|
users = models.ManyToManyField(Profile)
|
|
current_user = models.ForeignKey(Profile, related_name="following_users", on_delete=CASCADE)
|
|
|
|
@classmethod
|
|
def is_friend(self, current_user, new_friend):
|
|
try:
|
|
return current_user.following_users.get().users \
|
|
.filter(user=new_friend.user).exists()
|
|
except:
|
|
return False
|
|
|
|
@classmethod
|
|
def make_friend(self, current_user, new_friend):
|
|
friend, created = self.objects.get_or_create(
|
|
current_user = current_user
|
|
)
|
|
friend.users.add(new_friend)
|
|
|
|
@classmethod
|
|
def remove_friend(self, current_user, new_friend):
|
|
friend, created = self.objects.get_or_create(
|
|
current_user = current_user
|
|
)
|
|
friend.users.remove(new_friend)
|
|
|
|
@classmethod
|
|
def toggle_friend(self, current_user, new_friend):
|
|
if (self.is_friend(current_user, new_friend)):
|
|
self.remove_friend(current_user, new_friend)
|
|
else:
|
|
self.make_friend(current_user, new_friend)
|
|
|
|
@classmethod
|
|
def get_friend_profiles(self, current_user):
|
|
try:
|
|
ret = self.objects.get(current_user=current_user).users.all()
|
|
except Friend.DoesNotExist:
|
|
ret = Profile.objects.none()
|
|
return ret
|
|
|
|
def __str__(self):
|
|
return str(self.current_user)
|