Enforce company constraints

Signed-off-by: Sean Smith <ssmith@iol.unh.edu>
Change-Id: Id5726e5c9930e684c23cebafb98d5fbcb95e67bc

1 files changed, 18 insertions, 2 deletions

diff --git a/src/account/models.py b/src/account/models.py
index 6828ee1..2c133bb 100644
--- a/src/account/models.py
+++ b/src/account/models.py
@@ -11,6 +11,8 @@
 from django.contrib.auth.models import User
 from django.db import models
 from django.apps import apps
+from django.core.exceptions import ValidationError
+import re
 import json
 import random
 
@@ -50,13 +52,27 @@ class UserProfile(models.Model):
     oauth_token = models.CharField(max_length=1024, blank=False)
     oauth_secret = models.CharField(max_length=1024, blank=False)
 
-    jira_url = models.CharField(max_length=100, default='')
-    full_name = models.CharField(max_length=100, default='')
+    jira_url = models.CharField(max_length=100, null=True, blank=True, default='')
+    full_name = models.CharField(max_length=100, null=True, blank=True, default='')
     booking_privledge = models.BooleanField(default=False)
 
     class Meta:
         db_table = 'user_profile'
 
+    def clean(self, *args, **kwargs):
+        company = self.company
+        regex = r'[a-z\_\-\.\$]*'
+        pattern = re.compile(regex)
+
+        if not pattern.fullmatch(company):
+            raise ValidationError('Company may only include lowercase letters, _, -, . and $')
+
+        super().clean(*args, **kwargs)
+
+    def save(self, *args, **kwargs):
+        self.full_clean()
+        super().save(*args, **kwargs)
+
     def __str__(self):
         return self.user.username