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)