from MonetDB import *
from PostgreSQL import *
import time

#Monet DB
requeteSelect1MonetDB = """SELECT * FROM flights WHERE "AIRLINE" = 'AA';"""
requeteSelect2MonetDB = """SELECT * FROM flights;"""
requeteSelect3MonetDB = """SELECT * FROM flights WHERE "DAY" > 10;"""
requeteSelect4MonetDB = """SELECT * FROM flights WHERE "DAY" > 30;"""

requeteCreateTableMonetDB = """CREATE TABLE flights10 AS SELECT * FROM FLIGHTS WHERE "DAY" > 10;"""
requeteDeleteMonetDB = """DELETE FROM flights WHERE "DAY" > 10;"""
requeteInsertMonetDB = """INSERT INTO flights SELECT * FROM flights10;"""

#Postgre SQL
requeteSelect1PostgreSQL = """SELECT * FROM public.flights WHERE airline = 'AA'"""
requeteSelect2PostgreSQL = """SELECT * FROM public.flights"""
requeteSelect3PostgreSQL = """SELECT * FROM public.flight WHERE "DAY" > 10"""
requeteSelect4PostgreSQL = """SELECT * FROM public.flight WHERE "DAY" > 30"""

requeteCreateTablePostgreSQL = """CREATE TABLE flights10 AS SELECT * FROM FLIGHTS WHERE "DAY" > 10"""
requeteDeletePostgreSQL = """DELETE FROM flights WHERE "DAY" > 10"""
requeteInsertPostgreSQL = """INSERT INTO flights SELECT * FROM flights10"""

def display_table(data, headers):
     max_len = max(len(header) for header in headers)
     print(" | ".join(header.ljust(max_len) for header in headers))
     sep = "-" * max_len
     print("-|-".join(sep for _ in headers))
     for row in data:
        print(" | ".join(header.ljust(max_len) for header in row))


#start_time = time.time()
#end_time = time.time()
#print(end_time - start_time)


def initMonetDB(conn,cursor):
    reset_data_monetdb(conn,cursor)

    start_time = time.time()
    create_table_monetdb(conn,cursor)
    end_time = time.time()
    print("[MonetDB] creat table: time = ", (end_time - start_time), "s")

    start_time = time.time()
    load_data_monetdb(conn,cursor)
    end_time = time.time()
    print("[MonetDB] load data : time = ", (end_time - start_time), "s")

def execMonetDB(conn,cursor,str):
    start_time = time.time()
    requete_monetdb(conn,cursor, str)
    end_time = time.time()
    print("[MonetDB] requete :", str , "\n time = ", (end_time - start_time), "s")

def runMonetDB():
    conn, cursor =connect_monetdb()
    start = time.time()
    initMonetDB(conn,cursor)
    print(time.time() - start)
    execMonetDB(conn,cursor,requeteSelect1MonetDB)
    execMonetDB(conn,cursor,requeteSelect2MonetDB)
    execMonetDB(conn,cursor,requeteSelect3MonetDB)
    execMonetDB(conn,cursor,requeteSelect4MonetDB)

    disconnect_monetdb(conn, cursor)

def runPostgreSQL():
    reset_database_postgres()
    create_tables_postgres()
    create_data_postgres()
    #requete_postgres(str)

if __name__ == '__main__':
    #PostgreSQL
    runMonetDB()
    #runPostgreSQL()
    #MonetDB