''' Manage talks scheduling in a semantic way ''' from __future__ import print_function import os import io from functools import wraps import logging import re import datetime import shutil import time from docutils import nodes from docutils.parsers.rst import directives, Directive from pelican import signals, generators import jinja2 TALKS_PATH = 'talks' def memoize(function): '''decorators to cache''' memo = {} @wraps(function) def wrapper(*args): if args in memo: return memo[args] else: rv = function(*args) memo[args] = rv return rv return wrapper @memoize def get_talk_names(): return [name for name in os.listdir(TALKS_PATH) if not name.startswith('_') and get_talk_data(name) is not None ] def all_talks(): return [get_talk_data(tn) for tn in get_talk_names()] @memoize def get_global_data(): fname = os.path.join(TALKS_PATH, 'meta.yaml') if not os.path.isfile(fname): return None with io.open(fname, encoding='utf8') as buf: try: data = yaml.load(buf) except Exception as exc: logging.exception("Syntax error reading %s; skipping", fname) return None if data is None: return None if 'startdate' not in data: logging.error("Missing startdate in global data") data['startdate'] = datetime.datetime.now() return data @memoize def get_talk_data(talkname): fname = os.path.join(TALKS_PATH, talkname, 'meta.yaml') if not os.path.isfile(fname): return None with io.open(fname, encoding='utf8') as buf: try: data = yaml.load(buf) except Exception as exc: logging.exception("Syntax error reading %s; skipping", fname) return None if data is None: return None if 'title' not in data: logging.warn("Talk <{}> has no `title` field".format(talkname)) data['title'] = talkname if 'text' not in data: logging.warn("Talk <{}> has no `text` field".format(talkname)) data['text'] = '' if 'duration' not in data: logging.info("Talk <{}> has no `duration` field (50min used)" .format(talkname)) data['duration'] = 50 data['duration'] = int(data['duration']) if 'room' not in data: logging.warn("Talk <{}> has no `room` field".format(talkname)) if 'time' not in data or 'day' not in data: logging.warn("Talk <{}> has no `time` or `day`".format(talkname)) if 'day' in data: data['day'] = get_global_data()['startdate'] + datetime.timedelta(days=data['day']) if 'time' in data and 'day' in data: timeparts = re.findall(r'\d+', str(data['time'])) if 4 > len(timeparts) > 0: timeparts = [int(p) for p in timeparts] data['time'] = datetime.datetime.combine(data['day'], datetime.time(*timeparts)) else: logging.error("Talk <{}> has malformed `time`".format(talkname)) data['id'] = talkname resdir = os.path.join(TALKS_PATH, talkname, 'res') if os.path.isdir(resdir) and os.listdir(resdir): data['resources'] = resdir return data jinja_env = jinja2.Environment( loader=jinja2.FileSystemLoader(os.path.join(TALKS_PATH, '_templates')), autoescape=True, ) class TalkListDirective(Directive): required_arguments = 0 optional_arguments = 0 final_argument_whitespace = True has_content = True def run(self): tmpl = jinja_env.get_template('talk.html') return [ nodes.raw('', tmpl.render(**get_talk_data(n)), format='html') for n in get_talk_names() ] class TalkDirective(Directive): required_arguments = 1 optional_arguments = 0 final_argument_whitespace = True has_content = True def run(self): tmpl = jinja_env.get_template('talk.html') data = get_talk_data(self.arguments[0]) if data is None: return [] return [ nodes.raw('', tmpl.render(**data), format='html') ] def talks_to_ics(): content = 'BEGIN:VCALENDAR\nVERSION:2.0\nPRODID:pelican\n' for t in all_talks(): content += talk_to_ics(t) content += 'END:VCALENDAR\n' return content def talk_to_ics(talk): start = talk['time'] end = start + datetime.timedelta(minutes=talk['duration']) content = 'BEGIN:VEVENT\n' content += "UID:%s@%d.hackmeeting.org\n" % (talk['id'], talk['day'].year) content += "SUMMARY:%s\n" % talk['title'] content += "DTSTAMP:%s\n" % time.strftime('%Y%m%dT%H%M%SZ', time.gmtime(float(start.strftime('%s')))) content += "DTSTART:%s\n" % time.strftime('%Y%m%dT%H%M%SZ', time.gmtime(float( start.strftime('%s')))) content += "DTEND:%s\n" % time.strftime('%Y%m%dT%H%M%SZ', time.gmtime(float( end.strftime('%s')))) content += "LOCATION:%s\n" % talk['room'] content += 'END:VEVENT\n' return content class TalksGenerator(generators.Generator): def __init__(self, *args, **kwargs): self.talks = [] super(TalksGenerator, self).__init__(*args, **kwargs) def generate_context(self): self.talks = {n: get_talk_data(n) for n in get_talk_names()} self._update_context(('talks',)) def generate_output(self, writer=None): for talkname in self.talks: if 'resources' in self.talks[talkname]: outdir = os.path.join(self.output_path, 'talks', talkname, 'res') if os.path.isdir(outdir): shutil.rmtree(outdir) shutil.copytree(self.talks[talkname]['resources'], outdir) with open(os.path.join(self.output_path, 'talks.ics'), 'w') as buf: buf.write(talks_to_ics()) def add_talks_option_defaults(pelican): pelican.settings.setdefault('TALKS_PATH', TALKS_PATH) def get_generators(gen): return TalksGenerator try: import yaml except ImportError: print('ERROR: yaml not found. Talks plugins will be disabled') def register(): pass else: def register(): signals.get_generators.connect(get_generators) signals.initialized.connect(add_talks_option_defaults) directives.register_directive('talklist', TalkListDirective) directives.register_directive('talk', TalkDirective)