from datetime import date, time, datetime, timedelta from pytz import timezone from uuid import uuid4 from os import path from icalendar import Calendar, Event, vCalAddress, vText from colloscope.models import * from .create_calendar import get_calendar LOCAL_TZ = "Europe/Paris" def emailize(nom, prenom=None): if prenom is not None: return "{}.{}@example.com" \ .format( prenom.replace(" ", "_").lower(), nom.replace(" ", "_").lower() ) else: return "{}@example.com" \ .format(nom.replace(" ", "_").lower()) def to_calendar(etudiant, periode, include_EDT: bool = True): p = path.abspath('./static/Base_Calendar.ics') with open(p) as f: cal = Calendar.from_ical(f.read()) rotations = periode.query_rotations_etudiant(etudiant) for rotation in rotations: event = Event() summary = f"Colle {rotation.creneau.matiere} ({rotation.creneau.colleur})" event.add("summary", summary) start = rotation.datetime() fin = start + rotation.creneau.duree event.add("dtstart", start, parameters={"tzid": LOCAL_TZ}) event.add("dtend", fin, parameters={"tzid": LOCAL_TZ}) event.add("dtstamp", datetime.now()) event.add("uid", str(uuid4())) event.add("location", f"{rotation.creneau.salle} ({rotation.creneau.periode.classe.lycee})") event.add("categories", "COLLE-" + str(rotation.creneau.matiere)) description = f"Groupes: {','.join(str(groupe) for groupe in rotation.groupes.all())}" event.add("description", description) organizer = vCalAddress(f"mailto:{emailize(rotation.creneau.colleur.nom)}") organizer.params["cn"] = vText(str(rotation.creneau.colleur)) organizer.params["role"] = vText("Colleur") event.add("organizer", organizer) for e in rotation.groupe_effectif(): attendee = vCalAddress("mailto:{emailize(e.nom, prenom=e.prenom)}") attendee.params["role"] = vText("Etudiant") attendee.params["cn"] = vText(str(e)) event.add("attendee", attendee, encode=0) cal.add_component(event) """ if include_EDT: #TODO: get le groupe de l'étudiant et ses langue edt = get_calendar() for event in edt.walk("VEVENT"): cal.add_component(event) """ return cal