Move to class props.

8 years ago · 26c009a537
1 changed files with 13 additions and 6 deletions
--- a/bikeshop_project/core/models.py
+++ b/bikeshop_project/core/models.py
@ -47,13 +47,20 @@ class Payment(models.Model):


 class Visit(models.Model):
+    VOLUNTEER = 'VOLUNTEER'
+    FIX = 'FIX'
+    WORKSHOP = 'WORKSHOP'
+    VISIT = 'VISIT'
+    DONATE = 'DONATE'
+    STAFF = 'STAFF'
+
    visit_choices = (
-        ('VOLUNTEER', 'volunteer'),
-        ('FIX', 'fix bike'), # fix
-        ('WORKSHOP', 'workshop'),
-        ('VISIT', 'visit'),
-        ('DONATE', 'donate'),
-        ('STAFF', 'staff'),
+        (VOLUNTEER, 'volunteer'),
+        (FIX, 'fix bike'),  # fix
+        (WORKSHOP, 'workshop'),
+        (VISIT, 'visit'),
+        (DONATE, 'donate'),
+        (STAFF, 'staff'),
    )
    
    member = models.ForeignKey(