diff --git a/bikeshop_project/core/models.py b/bikeshop_project/core/models.py index 9ffe470..d167290 100644 --- 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(