Merge pull request #320 – Přidání času do modelu

#320

Author: encukou

naucse/models.py

@@ -1,6 +1,8 @@
 from collections import OrderedDict
 import copy
+import datetime
 
+import dateutil.tz
 import jinja2
 
 from naucse.modelutils import Model, YamlProperty, DataProperty, DirProperty
@@ -10,6 +12,8 @@
 from naucse.notebook_util import convert_notebook
 from pathlib import Path
 
+_TIMEZONE = 'Europe/Prague'
+
 
 class Lesson(Model):
     """An individual lesson stored on naucse"""
@@ -294,10 +298,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:
@@ -314,6 +319,23 @@ def __str__(self):
     slug = DataProperty(info)
     date = DataProperty(info, default=None)
 
+    def _time(self, key, default_time):
+        if self.date and default_time:
+            return datetime.datetime.combine(self.date, default_time)
+        return None
+
+    @reify
+    def start_time(self):
+        if self.course:
+            return self._time('start', self.course.default_start_time)
+        return None
+
+    @reify
+    def end_time(self):
+        if self.course:
+            return self._time('end', self.course.default_end_time)
+        return None
+
     @reify
     def materials(self):
         materials = [material(self.root, self.path, s)
@@ -352,7 +374,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())
@@ -424,6 +446,25 @@ def end_date(self):
             return None
         return max(dates)
 
+    def _default_time(self, key):
+        default_time = self.info.get('default_time')
+        if default_time:
+            time_string = default_time[key]
+            hour, minute = time_string.split(':')
+            hour = int(hour)
+            minute = int(minute)
+            tzinfo = dateutil.tz.gettz(_TIMEZONE)
+            return datetime.time(hour, minute, tzinfo=tzinfo)
+        return None
+
+    @reify
+    def default_start_time(self):
+        return self._default_time('start')
+
+    @reify
+    def default_end_time(self):
+        return self._default_time('end')
+
 
 class RunYear(Model):
     """A year of runs"""

naucse/routes.py

@@ -334,15 +334,19 @@ def course_calendar_ics(course):
         abort(404)
     calendar = ics.Calendar()
     for session in course.sessions.values():
-        combined = datetime.datetime.combine(session.date, datetime.time())
+        if session.start_time:
+            start_time = session.start_time
+            end_time = session.end_time
+        else:
+            abort(404)
         cal_event = ics.Event(
             name = session.title,
-            begin = combined,
+            begin = start_time,
+            end = end_time,
             uid = url_for("session_coverpage",
                            course=course,
                            session=session.slug,
                            _external=True),
         )
-        cal_event.make_all_day()
         calendar.events.append(cal_event)
     return Response(str(calendar), mimetype="text/calendar")

naucse/templates.py

@@ -170,6 +170,13 @@ def __str__(self):
     }
 
 
+@template_filter()
+def format_time(time):
+    if time.second:
+        return '{d.hour}:{d.minute:02}:{d.second:02}'.format(d=time)
+    return '{d.hour}:{d.minute:02}'.format(d=time)
+
+
 @template_filter()
 def format_date(date, relative_to=None):
     return '{d.day}. {d.month}. {d.year}'.format(d=date)

naucse/templates/course.html

@@ -57,6 +57,13 @@ <h4>
                             <div class="calendar-dates">
                                 {{ (course.start_date, course.end_date) | format_date_range }}
                             </div>
+                            {% if course.default_start_time %}
+                                <div class="default-times">
+                                    {{ course.default_start_time | format_time }}
+                                    –
+                                    {{ course.default_end_time | format_time }}
+                                </div>
+                            {% endif %}
                         </div>
                         <div class="card-footer small text-left">
                             <ul class="list-unstyled" style="margin-bottom:0;">

naucse/templates/course_calendar.html

@@ -28,10 +28,14 @@ <h1>{{ course.title }}</h1>
     {% if course.subtitle is defined and course.subtitle != None %}
         <h2>{{ course.subtitle }}</h2>
     {% endif%}
-    <div>
-      Přidej si kurz do svého kalendáře:
-      <a href="{{ url_for('course_calendar_ics', course=course) }}">iCal</a>
-    </div>
+    {% if course.default_start_time %}
+        <div>
+            Přidej si kurz do svého kalendáře:
+            <a href="{{ url_for('course_calendar_ics', course=course) }}">iCal</a>
+        </div>
+    {% else %}
+        <!-- iCal link missing because course doesn't have info about start/end times. -->
+    {% endif%}
     {% if course.start_date %}
         {% for year, month in months %}
             <h3 class="calendar-heading">

requirements.txt

@@ -10,5 +10,6 @@ jupyter_client
 mistune
 nbconvert
 pygments
+python-dateutil
 traitlets
 werkzeug

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: |

test_naucse/fixtures/test_content/courses/normal-course/info.yml

@@ -0,0 +1,5 @@
+title: A plain vanilla course
+
+plan:
+- title: A normal lesson
+  slug: normal-lesson

test_naucse/fixtures/test_content/runs/2000/run-with-times/info.yml

@@ -0,0 +1,9 @@
+title: Test run with scheduled times
+default_time:
+  start: '19:00'
+  end: '21:00'
+
+plan:
+- title: A normal lesson
+  slug: normal-lesson
+  date: 2000-01-01

test_naucse/fixtures/test_content/runs/2000/run-without-times/info.yml

@@ -0,0 +1,6 @@
+title: Test run without scheduled times
+
+plan:
+- title: A normal lesson
+  slug: normal-lesson
+  date: 2000-01-01