fossasia/open-event-orga-server

from urllib.parse import quote

import pytz
from flask import jsonify
from flask_jwt_extended import current_user
from icalendar import Calendar, Event, Timezone
from sqlalchemy import or_
from sqlalchemy.orm import joinedload

from app.api.helpers.utilities import remove_html_tags
from app.models.session import Session


def to_ical(event, include_sessions=False, my_schedule=False, user_id=None):
    cal = Calendar()
    cal.add('version', '2.0')
    cal.add('METHOD', 'PUBLISH')
    cal.add('X-WR-CALNAME', event.name)
    cal.add('X-WR-CALDESC', 'Event Calendar')

    event_component = Event()
    event_component.add('uid', event.identifier)
    event_component.add('summary', event.name)
    event_component.add('url', event.site_link)
    event_component.add('dtstart', event.starts_at_tz)
    event_component.add('dtend', event.ends_at_tz)
    event_component.add('location', event.normalized_location)
    event_component.add('description', remove_html_tags(event.description))
    if event.has_coordinates:
        event_component.add('geo', (event.latitude, event.longitude))
    if event.owner_description:
        event_component.add('organizer', remove_html_tags(event.owner_description))

    cal.add_component(event_component)

    timezone = Timezone()
    timezone.add('TZID', event.timezone)

    cal.add_component(timezone)

    if include_sessions:
        sessions_query = (
            Session.query.filter_by(event_id=event.id)
            .options(joinedload(Session.microlocation))
            .filter_by(deleted_at=None)
            .filter(or_(Session.state == 'accepted', Session.state == 'confirmed'))
            .order_by(Session.starts_at.asc())
        )
        if my_schedule:
            if not (current_user or user_id):
                return jsonify(error='Login or User ID required'), 401

            user_id = user_id or current_user.id
            sessions_query = sessions_query.join(Session.favourites).filter_by(
                user_id=user_id
            )
        sessions = sessions_query.all()

        for session in sessions:

            if not (session and session.starts_at and session.ends_at):
                continue

            session_video_url = (
                " "
                + event.site_link
                + '/video/'
                + quote(session.microlocation.video_stream.name)
                + "/"
                + str(session.microlocation.video_stream.id)
                if session.microlocation.video_stream
                else ""
            )
            session_link_heading = (
                " Join using link: " + session_video_url + " "
                if session_video_url
                else ""
            )
            session_description = (
                " "
                + "Room: "
                + session.microlocation.name
                + " "
                + session_link_heading
                + " "
                + session.short_abstract
            )
            session_component = Event()
            session_component.add('summary', session.title)
            session_component.add('uid', str(session.id) + "-" + event.identifier)
            session_component.add('geo', (event.latitude, event.longitude))
            session_component.add(
                'location',
                session_video_url
                or (session.microlocation and session.microlocation.name)
                or '' + " " + event.location_name,
            )
            session_component.add(
                'dtstart', session.starts_at.astimezone(pytz.timezone(event.timezone))
            )
            session_component.add(
                'dtend', session.ends_at.astimezone(pytz.timezone(event.timezone))
            )
            session_component.add('description', remove_html_tags(session_description))
            session_component.add('url', event.site_link + '/session/' + str(session.id))

            cal.add_component(session_component)

    return cal.to_ical()