From 543d4321544c3e4a570a7154e1d2ecc5e889deba Mon Sep 17 00:00:00 2001
From: Joel Lavikainen <joel.lavikainen@aalto.fi>
Date: Wed, 7 Jun 2017 19:14:57 +0300
Subject: [PATCH] Add kaehmyform and message

---
 webapp/models.py | 31 ++++++++++++++++++++++++++++++-
 1 file changed, 30 insertions(+), 1 deletion(-)

diff --git a/webapp/models.py b/webapp/models.py
index aae924e..d314415 100644
--- a/webapp/models.py
+++ b/webapp/models.py
@@ -51,13 +51,42 @@ class PresetRole(BaseRole):
     summary = models.TextField(_('Summary'))
 
 
-class CustomRole(BaseRole):
+class PresetKaehmyRole(PresetRole):
+    form = models.ForeignKey('KaehmyForm', related_name='preset_roles')
+
+
+class CustomKaehmyRole(BaseRole):
     '''
     Model representing a user-specified custom occupation
     '''
+    form = models.ForeignKey('KaehmyForm', related_name='custom_roles')
+
+
+class MessageParent(models.Model):
     pass
 
 
+class KaehmyMessage(MessageParent):
+    '''
+    Model representing a kaehmymessage.
+    Every message relates to certain kaehmyform or parent message.
+    '''
+    name = models.TextField(_('Name'))
+    email = models.EmailField(_('Email'))
+    message = models.TextField(_('Message'))
+    parent = models.ForeignKey('MessageParent', related_name='messages')
+
+
+class KaehmyForm(MessageParent):
+    '''
+    Model representing a form for kaehmy.
+    Allows user to choose from existing roles or to create custom ones.
+    '''
+    name = models.TextField(_('Name'))
+    email = models.EmailField(_('Email'))
+    year = models.IntegerField(_('Year'))
+
+
 class Role(PresetRole):
     '''
     Model representing an active or historical occupation