from django.db import models
from django.contrib.auth.models import User
from django.utils import timezone
from django.core.exceptions import ValidationError
import datetime


class Profile(models.Model):
    join_date = models.DateField()
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    TEAMS = (
        ('DT', 'DevTeam'),
        ('NET', 'NeTeam'),
        ('ST', 'SecurITeam'),
        ('SYS', 'SysAdmin'),
        ('HAT', 'Hallgatói Tudásbázis'),
        ('N', 'None'),
    )
    pref_group = models.CharField(max_length=10, choices=TEAMS, default='None')
    # Homeworks=models.ForeignKey(Homework)

    def __str__(self):
        return self.user.username

    def clean(self):
        if self.join_date > datetime.date.today() or self.join_date < datetime.date(2015, 1,1):
            raise ValidationError('Invalid date')