27 lines
759 B
Python
27 lines
759 B
Python
from django.db import models
|
|
from tp.models import UUIDAsIDModel
|
|
from django.contrib.auth.models import User
|
|
from django.core.exceptions import ValidationError
|
|
|
|
|
|
def twitter_handle_validator(x):
|
|
if x[0] != '@':
|
|
raise ValidationError('twitter_handle must begin with "@"')
|
|
|
|
|
|
class EmailWhiteList(UUIDAsIDModel):
|
|
email = models.EmailField(unique=True)
|
|
|
|
def __str__(self):
|
|
return self.email
|
|
|
|
|
|
class Account(UUIDAsIDModel):
|
|
user = models.OneToOneField(User, on_delete=models.CASCADE, unique=True)
|
|
totp_key = models.CharField(max_length=32, null=True)
|
|
use_totp = models.BooleanField(default=False)
|
|
twitter_handle = models.CharField(max_length=64, default='@Twitter')
|
|
|
|
def __str__(self):
|
|
return str(self.user)
|