fixed course variable in calendar

widget_OhMyBash/widget_calendar/models.py

@@ -2,13 +2,13 @@ from django.db import models
 
 
 class Location(models.Model):
-    onsite = 'Onsite'
-    online = 'Online'
-    hybrid = 'Hybrid'
+    onsite = 'onsite'
+    online = 'online'
+    hybrid = 'hybrid'
     mode_choices = [
-        (onsite, 'Onsite'),
-        (online, 'Online'),
-        (hybrid, 'Hybrid'),
+        (onsite, 'onsite'),
+        (online, 'online'),
+        (hybrid, 'hybrid'),
     ]
     mode = models.CharField(
         max_length=6, choices=mode_choices, default=onsite,
@@ -24,7 +24,11 @@ class Event(models.Model):
     activity = models.CharField(max_length=255)
     estimated_hours = models.FloatField()
     location = models.ForeignKey(Location, on_delete=models.CASCADE)
-    course = models.CharField(max_length=100)
+    course = course = models.ForeignKey(
+        'assignments.Course',
+        on_delete=models.CASCADE,
+        related_name="event_course"
+    )
 
     def __str__(self):
         return '{}, {}'.format(self.activity, self.target_datetime)

widget_OhMyBash/widget_calendar/views.py

@@ -13,7 +13,11 @@ def index(request):
         )
         return_string += 'Activity: {}<br>'.format(events.activity)
         return_string += 'Estimated Hours: {}<br>'.format(round_hours)
-        return_string += 'Course/Section: {}<br>'.format(events.course)
+        return_string += 'Course/Section: {} {}-{}<br>'.format(
+            events.course.code,
+            events.course.title,
+            events.course.section,
+        )
         return_string += 'Mode: {}<br>'.format(events.location.mode)
         return_string += 'Venue: {}<br>'.format(events.location.venue)
         return_string += '<br>'