main.py

# pylint: disable=global-statement,redefined-outer-name
import argparse
import os
from typing import Any, Dict
from urllib.parse import quote_plus

from flask import Flask, jsonify, redirect, render_template, send_from_directory,request
from flask_frozen import Freezer
from flaskext.markdown import Markdown

from miniconf.load_site_data import load_site_data
from miniconf.site_data import Paper, PlenarySession, Tutorial, Workshop

site_data: Dict[str, Any] = {}
by_uid: Dict[str, Any] = {}

# ------------- SERVER CODE -------------------->

app = Flask(__name__)
app.config.from_object(__name__)
freezer = Freezer(app)
markdown = Markdown(app)

app.jinja_env.filters["quote_plus"] = quote_plus

# MAIN PAGES


def _data():
    data = {"config": site_data["config"]}
    return data


@app.route("/")
def index():
    return redirect("/index.html")


# TOP LEVEL PAGES


@app.route("/index.html")
def home():
    data = _data()
    return render_template("index.html", **data)


@app.route("/invited_talks.html")
def invited_talks():
    data = _data()
    return render_template("invited_talks.html", **data)


@app.route("/awards.html")
def awards():
    data = _data()
    data['awards'] = site_data['awards']
    return render_template("awards.html", **data)


@app.route("/iaai.html")
def iaai():
    data = _data()
    data["IAAI_papers"] = site_data["IAAI_papers"]
    data["iaai_poster_schedule"] = site_data["iaai_poster_schedule"]
    data["iaai_poster_schedule_days"] = site_data["iaai_poster_schedule_days"]
    return render_template("iaai.html", **data)


@app.route("/eaai.html")
def eaai():
    data = _data()
    return render_template("eaai.html", **data)


@app.route("/doctoral_consortium.html")
def doctoral_consortium():
    data = _data()
    data["doctoral_consortium"] = site_data["doctoral_consortium"]
    return render_template("doctoral_consortium.html", **data)


@app.route("/undergraduate_consortium.html")
def undergraduate_consortium():
    data = _data()
    data["tutorials_UC"] = site_data["tutorials_UC"]
    data["tutorials_OTHER"] = site_data["tutorials_OTHER"]
    return render_template("undergraduate_consortium.html", **data)

@app.route("/student_abstract_program.html")
def student_abstract_program():
    data = _data()
    data["tutorials_UC"] = site_data["tutorials_UC"]
    data["tutorials_OTHER"] = site_data["tutorials_OTHER"]
    return render_template("student_abstract_program.html", **data)

@app.route("/new_faculty_highlights_program.html")
def new_faculty_highlights_program():
    data = _data()
    data["tutorials_FH"] = site_data["tutorials_FH"]
    return render_template("new_faculty_highlights_program.html", **data)

@app.route("/demonstrations.html")
def demonstrations():
    data = _data()
    data["demonstrations"] = site_data["demonstrations"]
    return render_template("demonstrations.html", **data)

@app.route("/diversity_programs.html")
def diversity_programs():
    data = _data()
    data["diversity_programs"] = site_data["diversity_programs"]
    data["diversity_programs_days"] = site_data["diversity_programs_days"]
    return render_template("diversity_programs.html", **data)


@app.route("/meet_with_a_fellow.html")
def gathers():
    data = _data()
    return render_template("meet_with_a_fellow.html", **data)


@app.route("/ai_job_fair.html")
def ai_job_fail():
    data = _data()
    return render_template("ai_job_fair.html", **data)


#@app.route("/reception.html")
#def reception():
#    data = _data()
#    return render_template("reception.html", **data)


@app.route("/faq.html")
def faq():
    data = _data()
    return render_template("faq.html", **data)


@app.route("/committees.html")
def committees():
    data = _data()
    return render_template("committees.html", **data)


@app.route("/help_desk.html")
def help_desk():
    data = _data()
    return render_template("help_desk.html", **data)


@app.route("/about.html")
def about():
    data = _data()
    data["FAQ"] = site_data["faq"]
    data["CodeOfConduct"] = site_data["code_of_conduct"]
    return render_template("about.html", **data)


@app.route("/papers.html")
def papers():
    data = _data()
    # The data will be loaded from `papers.json`.
    # See the `papers_json()` method and `static/js/papers.js`.
    data["tracks"] = site_data["main_program_tracks"]
    data["workshop_names"] = [wsh.title for wsh in site_data["workshops"]]
    return render_template("papers.html", **data)


@app.route("/papers_vis.html")
def papers_vis():
    data = _data()
    # The data will be loaded from `papers.json`.
    # See the `papers_json()` method and `static/js/papers.js`.
    data["tracks"] = site_data["main_program_tracks"] + ["System Demonstrations"]
    return render_template("papers_vis.html", **data)


@app.route("/papers_keyword_vis.html")
def papers_keyword_vis():
    data = _data()
    # The data will be loaded from `papers.json`.
    # See the `papers_json()` method and `static/js/papers.js`.
    data["tracks"] = site_data["tracks"]
    return render_template("papers_keyword_vis.html", **data)


@app.route("/schedule.html")
def schedule():
    data = _data()
    data["calendar"] = site_data["calendar"]
    data["event_types"] = site_data["event_types"]
    return render_template("schedule.html", **data)


# @app.route("/livestream.html")
# def livestream():
#     data = _data()
#     return render_template("livestream.html", **data)


@app.route("/invited_panels_program.html")
def invited_panels():
    data = _data()
    data["invited_panels"] = site_data["invited_panels"]
    data["invited_panels_days"] = site_data["invited_panels_days"]
    return render_template("invited_panels_program.html", **data)

@app.route("/invited_speaker_program.html")
def invited_speaker():
    data = _data()
    data["invited_speakers"] = site_data["invited_speakers"]
    data["invited_speakers_days"] = site_data["invited_speakers_days"]
    return render_template("invited_speaker_program.html", **data)

@app.route("/ai_in_practice.html")
def ai_in_practice():
    data = _data()
    data["ai_in_practice"] = site_data["ai_in_practice"]
    data["ai_in_practice_days"] = site_data["ai_in_practice_days"]
    return render_template("ai_in_practice.html", **data)

@app.route("/plenary_sessions.html")
def plenary_sessions():
    data = _data()
    data["plenary_sessions"] = site_data["plenary_sessions"]
    data["plenary_session_days"] = site_data["plenary_session_days"]
    return render_template("plenary_sessions.html", **data)


@app.route("/posters.html",defaults={"tab_id":"feb4"})
@app.route("/posters_<tab_id>.html")
def posters(tab_id):
    # tab_id = request.args.get("tab_id","")
    data = _data()
    # data["poster_days"] = site_data["poster_days"]
    poster_days = []
    days = ["Feb 4","Feb 5","Feb 6","Feb 7","Award-Winning Papers"]
    for i, day in enumerate(days):
        tab = day.replace(" ", "").lower()
        if tab_id == "":
            poster_days.append(
                (tab, day, "active" if i==0 else "")
            )
        else:
            poster_days.append(
                (tab, day, "active" if tab == tab_id else "")
            )
    data["poster_days"] = poster_days
    data["poster_info"] = site_data["poster_info_by_day"]
    data["papers"] = by_uid["papers"]
    return render_template("posters.html", **data)


@app.route("/tutorials.html")
def tutorials():
    data = _data()
    data["tutorials"] = site_data["tutorials"]
    data["tutorials_MQ"] = site_data["tutorials_MQ"]
    data["tutorials_MH"] = site_data["tutorials_MH"]
    data["tutorials_AQ"] = site_data["tutorials_AQ"]
    data["tutorials_AH"] = site_data["tutorials_AH"]
    return render_template("tutorials.html", **data)

@app.route("/workshops.html")
def workshops():
    data = _data()
    data["workshops"] = site_data["workshops"]
    data["workshop_days"] = site_data["workshop_days"]
    return render_template("workshops.html", **data)


@app.route("/sponsors.html")
def sponsors():
    data = _data()
    data["sponsors"] = site_data["sponsors_by_level"]
    data["sponsor_levels"] = site_data["sponsor_levels"]
    return render_template("sponsors.html", **data)


@app.route("/socials.html")
def socials():
    data = _data()
    data["socials"] = site_data["socials"]
    return render_template("socials.html", **data)

# @app.route("/diversity_programs.html")
# def diversity_programs():
#     data = _data()
    
#     return render_template("diversity_programs.html, **data")

@app.route("/organizers.html")
def organizers():
    data = _data()

    data["committee"] = site_data["committee"]
    return render_template("organizers.html", **data)


# ITEM PAGES


@app.route("/paper_<uid>.html")
def paper(uid):
    data = _data()

    v: Paper = by_uid["papers"][uid]
    data["id"] = uid
    data["openreview"] = v
    data["paper"] = v
    data["paper_recs"] = [
        by_uid["papers"][ii] for ii in v.content.similar_paper_uids[1:]
    ]

    return render_template("paper.html", **data)


@app.route("/plenary_session_<uid>.html")
def plenary_session(uid):
    data = _data()
    data["plenary_session"] = by_uid["plenary_sessions"][uid]
    return render_template("plenary_session.html", **data)

@app.route("/plenary_session_opening_remarks_and_speaker_by_tuomas_sandholm.html")
def plenary_session_first():
    data = _data()
    data["plenary_session"] = by_uid["plenary_sessions"]['opening_remarks_speaker_by_tuomas_sandholm']
    return render_template("plenary_session.html", **data)

@app.route("/tutorial_<uid>.html")
def tutorial(uid):
    data = _data()
    data["tutorial"] = by_uid["tutorials"][uid]
    return render_template("tutorial.html", **data)

@app.route("/fh_<uid>.html")
def faculty_highlights(uid):
    data = _data()
    data["tutorial"] = by_uid["tutorials"][uid]
    return render_template("new_faculty_highlights_program_single.html", **data)


@app.route("/workshop_<uid>.html")
def workshop(uid):
    data = _data()
    data["workshop"] = by_uid["workshops"][uid]
    return render_template("workshop.html", **data)


@app.route("/sponsor_<uid>.html")
def sponsor(uid):
    data = _data()
    data["sponsor"] = by_uid["sponsors"][uid]
    data["papers"] = by_uid["papers"]
    return render_template("sponsor.html", **data)


@app.route("/chat.html")
def chat():
    data = _data()
    return render_template("chat.html", **data)


# FRONT END SERVING


@app.route("/papers.json")
def papers_json():
    all_papers = site_data["papers"]

    return jsonify(all_papers)


@app.route("/papers_<program>.json")
def papers_program(program):
    paper: Paper
    if program == "workshop":
        papers_for_program = []
        for wsh in site_data["workshops"]:
            papers_for_program.extend(wsh.papers)
    else:
        if program == "Best":
            papers_for_program = [
                paper for paper in site_data["papers"] if paper.content.best_type >= 1
            ]
            papers_for_program.sort(key=lambda paper :paper.content.best_type)

        else:
            papers_for_program = [
                paper for paper in site_data["papers"] if paper.content.program == program
            ]
    return jsonify(papers_for_program)


# @app.route("/track_<program_name>_<track_name>.json")
# def track_json(program_name, track_name):
#     paper: Paper
#     if program_name == "workshop":
#         papers_for_track = None
#         for wsh in site_data["workshops"]:
#             if wsh.title == track_name:
#                 papers_for_track = wsh.papers
#                 break
#     else:
#         papers_for_track = [
#             paper
#             for paper in site_data["papers"]
#             if paper.content.track == track_name
#             and paper.content.program == program_name
#         ]
#     return jsonify(papers_for_track)


# @app.route("/static/<path:path>")
# def send_static(path):
#     return send_from_directory("static", path)


@app.route("/serve_<path>.json")
def serve(path):
    return jsonify(site_data[path])


# --------------- DRIVER CODE -------------------------->
# Code to turn it all static


@freezer.register_generator
def generator():

    paper: Paper
    for paper in site_data["papers"]:
        yield "paper", {"uid": paper.id}
    for program in site_data["programs"]:
        yield "papers_program", {"program": program}
        # for track in site_data["tracks"]:
        #     yield "track_json", {"track_name": track, "program_name": program}

    # yield "papers_program", {"program": "workshop"}
    # for wsh in site_data["workshops"]:
    #     yield "track_json", {"track_name": wsh.title, "program_name": "workshop"}
    plenary_session: PlenarySession
    for _, plenary_sessions_on_date in site_data["plenary_sessions"].items():
        for plenary_session in plenary_sessions_on_date:
            yield "plenary_session", {"uid": plenary_session.id}
    tutorial: Tutorial
    for tutorial in site_data["tutorials"]:
        yield "tutorial", {"uid": tutorial.id}
    for tutorial in site_data["tutorials_MQ"]:
        yield "tutorial", {"uid": tutorial.id}
    for tutorial in site_data["tutorials_MH"]:
        yield "tutorial", {"uid": tutorial.id}
    for tutorial in site_data["tutorials_AQ"]:
        yield "tutorial", {"uid": tutorial.id}
    for tutorial in site_data["tutorials_AH"]:
        yield "tutorial", {"uid": tutorial.id}
    for tutorial in site_data["tutorials_UC"]:
        yield "tutorial", {"uid": tutorial.id}
    for tutorial in site_data["tutorials_OTHER"]:
        yield "tutorial", {"uid": tutorial.id}
    for tutorial in site_data["tutorials_FH"]:
        yield "faculty_highlights", {"uid": tutorial.id}
    workshop: Workshop
    # for workshop in site_data["workshops"]:
    #     yield "workshop", {"uid": workshop.id}
    for _, workshops_on_date in site_data["workshops"].items():
        for workshop in workshops_on_date:
            yield "workshop", {"uid": workshop.id}

    for tab_id in ["feb4", "feb5", "feb6", "feb7"]:
        yield "posters", {"tab_id": tab_id}

    # demonstrations: Demonstrations
    # for demonstration in site_data["demonstrations"]:
    #     yield "demonstration", {"uid": demonstration.id}

    # doctoral_consortium: Doctoral_consortium
    # for doctoral_consortium in site_data["doctoral_consortium"]:
    #     yield "doctoral_consortium", {"uid": doctoral_consortium.id}

    for sponsor in site_data["sponsors"]:
        if "landingpage" in sponsor:
            continue
        yield "sponsor", {"uid": str(sponsor["UID"])}

    for key in site_data:
        yield "serve", {"path": key}


def parse_arguments():
    parser = argparse.ArgumentParser(description="MiniConf Portal Command Line")
    parser.add_argument(
        "--build",
        action="store_true",
        default=False,
        help="Convert the site to static assets",
    )
    parser.add_argument(
        "-b",
        action="store_true",
        default=False,
        dest="build",
        help="Convert the site to static assets",
    )

    return parser.parse_args()


if __name__ == "__main__":
    args = parse_arguments()

    extra_files = load_site_data("sitedata", site_data, by_uid)

    if args.build:
        freezer.freeze()
    else:
        debug_val = False
        if os.getenv("FLASK_DEBUG") == "True":
            debug_val = True

        app.run(port=5000, debug=debug_val, extra_files=extra_files)