From 5cd4e2d61c4d3e28fd6544ad5963270943354fbf Mon Sep 17 00:00:00 2001
From: Chif Gergo <chifgeri97@gmail.com>
Date: Tue, 7 Nov 2017 20:20:28 +0100
Subject: [PATCH] Modellek aprobb valtoztatasai

---
 src/stats/models.py | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/src/stats/models.py b/src/stats/models.py
index 37b40a6..8bbc7af 100644
--- a/src/stats/models.py
+++ b/src/stats/models.py
@@ -5,8 +5,16 @@ from django.contrib.auth.models import User
 
 class Profile(models.Model):
     year_of_join = models.IntegerField()
-    user = models.OneToOneField(User)
-    # Homeworks=models.ForeignKey(Homework);
+    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
@@ -14,5 +22,6 @@ class Profile(models.Model):
 
 class KszkEvent(models.Model):
     date = models.DateField()
+    visitors = models.ManyToManyField(User, related_name = 'visitor')
+
     num_of_pers = models.IntegerField()
-    visitors = models.ManyToManyField(User)
-- 
GitLab