clean up

generic/models.py

@@ -563,12 +563,25 @@ class UserUserGroup(models.Model):
     def get_absolute_url(self):
         return reverse("generic:user_group_update", kwargs={"pk": self.pk})
     
+class UserGrades(models.Model):
+    name = models.CharField(max_length=100)
+
+    def __str__(self):
+        return self.name
 
 class UserProfile(models.Model):
     user = models.OneToOneField(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
     supervisor_email = models.EmailField(blank=True)
     supervisor_name = models.CharField(max_length=100, blank=True)
     registration_number = models.CharField(max_length=25, blank=True)
+    grade = models.ForeignKey(UserGrades, null=True, blank=True, help_text="User grade", on_delete=models.CASCADE)
+    peninsula_trainee = models.BooleanField(default=False)
+
+
+    def getusername(self):
+        return self.user.username
+
+    username = property(getusername)
     
 
 @receiver(post_save, sender=User)