summaryrefslogtreecommitdiff
path: root/workshop_app/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'workshop_app/models.py')
-rw-r--r--workshop_app/models.py26
1 files changed, 23 insertions, 3 deletions
diff --git a/workshop_app/models.py b/workshop_app/models.py
index bd4b2ab..07db83a 100644
--- a/workshop_app/models.py
+++ b/workshop_app/models.py
@@ -1,5 +1,25 @@
-from __future__ import unicode_literals
-
+#from __future__ import unicode_literals
from django.db import models
+from django.contrib.auth.models import User
+
+position_choices = (
+ ("student", "Student"),
+ ("instructor", "Instructor")
+ )
+
+class Profile(models.Model):
+ """Profile for users"""
+
+ user = models.OneToOneField(User)
+ institute = models.CharField(max_length=150)
+ department = models.CharField(max_length=150)
+ position = models.CharField(max_length=32, choices=position_choices)
+
+ def __str__(self):
+ return u"{0} {1} | {2}".format(self.user.first_name, self.user.last_name, self.user.email)
+
+
+def has_profile(user):
+ """ check if user has profile """
+ return True if hasattr(user, 'profile') else False
-# Create your models here.