trentpalmerdotorg/accounts/models.py

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)