Přidání času do modelu

Author: vahalova

naucse/models.py

@@ -1,5 +1,6 @@
 from collections import OrderedDict
 import copy
+import datetime
 
 import jinja2
 
@@ -279,10 +280,11 @@ def merge_dict(base, patch):
 
 class Session(Model):
     """An ordered collection of materials"""
-    def __init__(self, root, path, base_course, info, index):
+    def __init__(self, root, path, base_course, info, index, course=None):
         super().__init__(root, path)
         base_name = info.get('base')
         self.index = index
+        self.course = course
         if base_name is None:
             self.info = info
         else:
@@ -299,6 +301,33 @@ def __str__(self):
     slug = DataProperty(info)
     date = DataProperty(info, default=None)
 
+    @reify
+    def start(self):
+        if self.date != None and self.course != None:
+            default_time = self.course.info.get('default_time')
+            if default_time != None:
+                start_time = default_time['start']
+                hour, minute = start_time.split(':')
+                hour = int(hour)
+                minute = int(minute)
+                start_time = datetime.time(hour, minute)
+                return datetime.datetime.combine(self.date, start_time)
+        return None
+
+    @reify
+    def end(self):
+        if self.date != None and self.course != None:
+            default_time = self.course.info.get('default_time')
+            if default_time != None:
+                end_time = default_time['end']
+                hour, minute = end_time.split(':')
+                hour = int(hour)
+                minute = int(minute)
+                end_time = datetime.time(hour, minute)
+                return datetime.datetime.combine(self.date, end_time)
+        return None
+
+
     @reify
     def materials(self):
         materials = [material(self.root, self.path, s)
@@ -337,7 +366,7 @@ def _get_sessions(course, plan):
     result = OrderedDict()
     for index, sess_info in enumerate(plan):
         session = Session(course.root, course.path, course.base_course,
-                          sess_info, index=index)
+                          sess_info, index=index, course=course)
         result[session.slug] = session
 
     sessions = list(result.values())

runs/2018/pyladies-brno-jaro/info.yml

@@ -1,6 +1,9 @@
 title: Začátečnický kurz PyLadies
 subtitle: Brno - jaro 2018 - pondělí
 time: 18:00–20:00
+default_time:
+  start: '18:00'
+  end: '20:00'
 place: Experis, BC Titanuim, Nové sady 996/25
 description: Naučte se Python vážně od začátku. Žádné předchozí znalosti nejsou potřeba.
 long_description: |