from flask import render_template, request, g, redirect, url_for, session
from .config import db
import pymysql

def index():
    if request.method == 'GET':
        if not g.user:
            return redirect(url_for("login"))
        flightID = request.args.get('flightID')

        conn = pymysql.connect(**db)
        cursor = conn.cursor(pymysql.cursors.DictCursor)
        
        search_sql = """ """
        cursor.execute(search_sql, (flightID, ))
        flights = cursor.fetchall()
        cursor.close()
        conn.close()
        
        return render_template(
            'search.html', 
            flights=flights, 
            username=g.user
        )

def logout():
    session.clear()
    session.pop('user_id', None)
    return redirect(url_for('login'))

def modify():
    pass