diff --git a/calendar/teachingdates/calendars/basecalendar.py b/calendar/teachingdates/calendars/basecalendar.py index 9f08493..d000739 100644 --- a/calendar/teachingdates/calendars/basecalendar.py +++ b/calendar/teachingdates/calendars/basecalendar.py @@ -155,6 +155,22 @@ autoescape=False, loader=jinja2.PackageLoader("teachingdates", "templates")) env.filters.update(filters) + template = env.get_template( + "{style}.tex.j2".format(style=self.config.style)) + if self.config.style == "lecture": + return template.render( + weeks=self.lecture_dates(), + paper=self.config.paper, + period=self.config.period, + year=self.config.year, + eow_offset=datetime.timedelta(self.config.end_of_week)) + else: + return template.render( + weeks=self.calendars[self.config.period], + paper=self.config.paper, + period=self.config.period, + year=self.config.year, + eow_offset=datetime.timedelta(self.config.end_of_week)) def render_text(self): env = jinja2.Environment(