OnlineJudge/account/models.py

# coding=utf-8
from django.db import models
from django.contrib.auth.models import AbstractBaseUser

from jsonfield import JSONField

# TODO remove these
REGULAR_USER = 0
ADMIN = 1
SUPER_ADMIN = 2


class AdminType(object):
    REGULAR_USER = 0
    ADMIN = 1
    SUPER_ADMIN = 2


class ProblemSolutionStatus(object):
    ACCEPTED = 1
    PENDING = 2


class AdminExtraPermission(object):
    CREATE_PUBLIC_CONTEST = 1
    MANAGE_ALL_CONTEST = 2
    # 3 and 4 are mutually exclusive
    MANAGE_ALL_PROBLEM = 3
    # Manage public problem user created
    MANAGE_OWN_PROBLEM = 4


class UserManager(models.Manager):
    use_in_migrations = True

    def get_by_natural_key(self, username):
        return self.get(**{self.model.USERNAME_FIELD: username})


class User(AbstractBaseUser):
    username = models.CharField(max_length=30, unique=True)
    real_name = models.CharField(max_length=30, blank=True, null=True)
    email = models.EmailField(max_length=254, blank=True, null=True)
    create_time = models.DateTimeField(auto_now_add=True, null=True)
    # One of UserType
    admin_type = models.IntegerField(default=0)
    # List of items in AdminExtraPermission
    admin_extra_permission = JSONField(default=[])
    # Store user problem solution status with json string format
    # {"problems": {1: ProblemSolutionStatus.ACCEPTED}, "contest_problems": {20: ProblemSolutionStatus.PENDING)}
    problems_status = JSONField(default={})
    reset_password_token = models.CharField(max_length=40, blank=True, null=True)
    reset_password_token_create_time = models.DateTimeField(blank=True, null=True)
    # SSO auth token
    auth_token = models.CharField(max_length=40, blank=True, null=True)
    two_factor_auth = models.BooleanField(default=False)
    tfa_token = models.CharField(max_length=40, blank=True, null=True)
    # open api key
    open_api = models.BooleanField(default=False)
    open_api_appkey = models.CharField(max_length=35, blank=True, null=True)
    is_disabled = models.BooleanField(default=False)

    USERNAME_FIELD = 'username'
    REQUIRED_FIELDS = []

    objects = UserManager()

    class Meta:
        db_table = "user"


def _random_avatar():
    import random
    return "/static/img/avatar/avatar-" + str(random.randint(1, 20)) + ".png"


class UserProfile(models.Model):
    user = models.OneToOneField(User)
    avatar = models.CharField(max_length=50, default=_random_avatar)
    blog = models.URLField(blank=True, null=True)
    mood = models.CharField(max_length=200, blank=True, null=True)
    hduoj_username = models.CharField(max_length=30, blank=True, null=True)
    bestcoder_username = models.CharField(max_length=30, blank=True, null=True)
    codeforces_username = models.CharField(max_length=30, blank=True, null=True)
    accepted_problem_number = models.IntegerField(default=0)
    submission_number = models.IntegerField(default=0)
    phone_number = models.CharField(max_length=15, blank=True, null=True)
    school = models.CharField(max_length=200, blank=True, null=True)
    student_id = models.CharField(max_length=15, blank=True, null=True)

    def add_accepted_problem_number(self):
        self.accepted_problem_number = models.F("accepted_problem_number") + 1
        self.save()

    def add_submission_number(self):
        self.submission_number = models.F("submission_number") + 1
        self.save()

    def minus_accepted_problem_number(self):
        self.accepted_problem_number = models.F("accepted_problem_number") - 1
        self.save()

    class Meta:
        db_table = "user_profile"