"""Example web application for IMDb."""
# Run with: flask --app imdb run --debug

import psycopg
from flask import Flask, render_template, request

app = Flask(__name__)
con = psycopg.connect(host="localhost", user="demo", password="demo", dbname="imdb")
cur = con.cursor()


@app.route("/")
def index():
    pid = request.args.get("pid", "")
    if pid:
        # Get the Person information
        cur.execute("SELECT * FROM person WHERE pid = %s", [pid])
        person = cur.fetchone()
        if person:
            # Get the "known for" titles
            ids = [int(tid) for tid in person[5].split(",")]
            cur.execute("SELECT * FROM title WHERE tid = ANY(%s)", [ids])
            known = cur.fetchall()
    else:
        person = None
        known = None
    return render_template("imdb.jinja",
        pid=pid, person=person, known=known)