OnlineJudge/account/models.py
virusdefender 592720dd01 增加用户登录
包括用户model、用户登录模板、登录api和测试。
2015-06-29 12:43:17 +08:00

28 lines
627 B
Python

# coding=utf-8
from django.db import models
from django.contrib.auth.models import AbstractBaseUser
class AdminGroup(models.Model):
pass
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)
admin_group = models.ForeignKey(AdminGroup, null=True, on_delete=models.SET_NULL)
USERNAME_FIELD = 'username'
REQUIRED_FIELDS = []
objects = UserManager()
class Meta:
db_table = "user"