generalists.py

"""
Build different lektor files with lists of contributions.

The different files account for lists of posters, talks and other oral
contributions. Each list is composed of different theme blocks, so
contributions are grouped into the different themes. Each item in
the list is a link to its own record in the ADASS program.
"""
import argparse
import logging
import os
import psycopg2
from pathlib import Path

logging.basicConfig(level=logging.INFO)
log = logging.getLogger(__name__)

DEF_BASE = "/Users/jer/git/adassweb/website/content"
program_url = "https://schedule.adass2020.es/adass2020/talk"
folders = ["posters", "talks", "invited-talks", "bofs", "demos", "tutorials"]
themes = {
    "1": "Science Platforms and Data Lakes",
    "2": "Cloud Computing at Different Scales",
    "3": "Cross-Discipline Projects",
    "4": "Multi-Messenger Astronomy",
    "5": "Machine Learning, Statistics, and Algorithms",
    "6": "Time-Domain Ecosystem",
    "7": "Citizen Science Projects in Astronomy",
    "8": "Data Processing Pipelines and Science-Ready Data",
    "9": "Data Interoperability",
    "10": "Open Source Software and Community Development in Astronomy",
    "11": "Other",
}
template = """
 _model: page
---
title:
---
description:
---
_template: page-md.html
---
body:
"""


def add_icons(pdf, video):
    icons = "- "
    if pdf:
        icons += f"<a href='{pdf}'>&#x1F4D1;</a>"
    if video:
        icons += f"<a href='{video}'>&#x1F3A6;</a>"
    return icons


def make_folder_structure(base):
    for item in folders:
        fold = Path(item)
        content = template
        item = item.replace("-", " ")
        content = content.replace("title:", f"title: {item.capitalize()}")
        content = content.replace("description:", f"description: {item.capitalize()}")
        if item == "bofs":
            content = content.replace("Bofs", "BOFs")
        Path.mkdir(base / fold, parents=True, exist_ok=True)
        fn = open(base / fold / "contents.lr", "w")
        fn.write(content)
        fn.close()
    log.info(" making folders successful")


def fill_posters(base):
    fold = Path(folders[0])
    contents = ""
    contents += """
#### NOTE
**There are no specific posters sessions scheduled.**
Instead conference participants will be allowed to download the PDF poster files from this list.
They could also download optional complementary self-recorded lightning talks as MP4 video files.
Access authentication info will be delivered in Discord, where participants could also interact 
with posters authors individually or within the #posters channel. 
    """
    for number, label in themes.items():
        bag = listings[number]["posters"]
        contents += f"\n**{label.upper()}**\n\n"
        for pack in bag:
            title, abstract, code, author, pdf, video = pack
            contents += f"{add_icons(pdf, video)} <a href='{program_url}/{code}' target='_schedule'>{title}</a>, {author}\n"
    fn = open(base / fold / "contents.lr", "a")
    fn.write(contents)
    fn.close()


def fill_talks(base):
    fold = Path(folders[1])
    contents = ""
    for number, label in themes.items():
        bag = listings[number]["talks"]["contributed"]
        contents += f"\n**{label.upper()}**\n\n"
        for pack in bag:
            title, abstract, code, author, pdf, video = pack
            contents += f"{add_icons(pdf, video)} <a href='{program_url}/{code}' target='_schedule'>{title}</a>, {author}\n"
    fn = open(base / fold / "contents.lr", "a")
    fn.write(contents)
    fn.close()


def fill_invited(base):
    fold = Path(folders[2])
    contents = ""
    for number, label in themes.items():
        bag = listings[number]["talks"]["invited"]
        if len(bag):
            contents += f"\n**{label.upper()}**\n\n"
            for pack in bag:
                title, abstract, code, author, pdf, video = pack
                contents += f"{add_icons(pdf, video)} <a href='{program_url}/{code}' target='_schedule'>{title}</a>, {author}\n"
    fn = open(base / fold / "contents.lr", "a")
    fn.write(contents)
    fn.close()


def fill_bofs(base):
    fold = Path(folders[3])
    contents = ""
    for pack in bofs:
        title, abstract, code, author, pdf, video = pack
        contents += (
            f"<b><a href='{program_url}/{code}' "
            + f"target='_schedule'>{title}</a></b>\n\n<b>{author}</b>\n\n"
        )
        contents += f"{abstract}\n\n"
    fn = open(base / fold / "contents.lr", "a")
    fn.write(contents)
    fn.close()


def fill_demos(base):
    fold = Path(folders[4])
    contents = ""
    for pack in demos:
        title, abstract, code, author, pdf, video = pack
        contents += (
            f"<b><a href='{program_url}/{code}' "
            + f"target='_schedule'>{title}</a></b>\n\n<b>{author}</b>\n\n"
        )
        contents += f"{abstract}\n\n"
    fn = open(base / fold / "contents.lr", "a")
    fn.write(contents)
    fn.close()


def fill_tutos():
    fold = Path(folders[5])
    contents = ""
    for pack in tutos:
        title, abstract, code, author, pdf, video = pack
        contents += (
            f"<b><a href='{program_url}/{code}' "
            + f"target='_schedule'>{title}</a></b>\n\n<b>{author}</b>\n\n"
        )
        contents += f"{abstract}\n\n"
    fn = open(base / fold / "contents.lr", "a")
    fn.write(contents)
    fn.close()


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "base",
        metavar="ROOTDIR",
        type=str,
        nargs=1,
        default=DEF_BASE,
        help="root of the lektor site (website/content)",
    )
    args = parser.parse_args()
    base = args.base[0]

    if not os.path.isdir(base):
        parser.error(f"{base}: no such file or directory")

    listings = {}
    demos = []
    tutos = []
    bofs = []

    # connect to posgre pretalx
    with psycopg2.connect(
        database="pretalx",
        user="pretalx",
        password="",
        host="pretalx.adass2020.es",
        port="5432",
    ) as conn:

        # build listings data structure
        sql = """
        SELECT
        title, abstract, submission_submission.code, name, pdf_path, video_path
        FROM
        submission_submission,
        person_user
        WHERE
        submission_submission.main_author_id=person_user.id
        AND submission_submission.state='confirmed'
        """
        for number, label in themes.items():
            listings[number] = {
                "posters": [],
                "talks": {"invited": [], "contributed": [],},
            }
            # add poster for each theme
            sql_posters = sql
            sql_posters += f"""
                AND submission_submission.submission_type_id in (13, 18)
                AND submission_submission.paper_id LIKE 'P{number}-%'
                ORDER BY title
                """
            cur = conn.cursor()
            cur.execute(sql_posters)
            for row in cur:
                listings[number]["posters"].append(row)

            # add invited talks for each theme
            sql_invited_talks = sql
            sql_invited_talks += f"""
                AND submission_submission.submission_type_id=1
                AND submission_submission.paper_id LIKE 'I{number}-%'
                ORDER BY title
                """
            cur = conn.cursor()
            cur.execute(sql_invited_talks)
            for row in cur:
                listings[number]["talks"]["invited"].append(row)

            # add contributed talks for each theme
            sql_talks = sql
            sql_talks += f"""
                AND submission_submission.submission_type_id=3
                AND submission_submission.paper_id LIKE 'O{number}-%'
                ORDER BY title
                """
            cur = conn.cursor()
            cur.execute(sql_talks)
            for row in cur:
                listings[number]["talks"]["contributed"].append(row)

        # fill bofs
        sql_bofs = sql
        sql_bofs += """
        AND submission_submission.submission_type_id=4
        ORDER BY title
        """
        cur = conn.cursor()
        cur.execute(sql_bofs)
        for row in cur:
            bofs.append(row)

        # fill demos
        sql_demos = sql
        sql_demos += """
        AND submission_submission.submission_type_id=15
        ORDER BY title
        """
        cur = conn.cursor()
        cur.execute(sql_demos)
        for row in cur:
            demos.append(row)

        # fill tutos
        sql_tutos = sql
        sql_tutos += """
        AND submission_submission.submission_type_id=14
        ORDER BY title
        """
        cur = conn.cursor()
        cur.execute(sql_tutos)
        for row in cur:
            tutos.append(row)

        # start filling files
        make_folder_structure()
        fill_posters()
        fill_talks()
        fill_invited()
        fill_bofs()
        fill_demos()
        fill_tutos()