from django.db import models
from django.contrib.auth.models import User
# Create your models here.


class Profile(models.Model):
    year_of_join = models.IntegerField()
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    TEAMS = (
     ('DT', 'DevTeam'),
     ('NET', 'NeTeam'),
     ('ST', 'SecurITeam'),
     ('SYS', 'SysAdmin'),
     ('N', 'None'),
    )
    pref_group = models.CharField(max_length=10, choices=TEAMS, default='None')
    # Homeworks=models.ForeignKey(Homework)

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


class KszkEvent(models.Model):
    date = models.DateField()
    visitors = models.ManyToManyField(User, related_name='visitor')

    num_of_pers = models.IntegerField()