import psycopg2
import pandas as pd
from query import *
from psycopg2.extras import NamedTupleCursor

# Connect to base
try:
    conn = psycopg2.connect(database="oog", user="oog", password="oog", host="172.17.74.134", port="5432")
    print('Base opened')
except:
    print("Conneсtion failed")
cur = conn.cursor(cursor_factory=NamedTupleCursor)

#create xlsx
def save_xlsx():
    list_one_x = pd.read_sql(list_one, conn)
    df1 = pd.DataFrame(list_one_x)

    list_two_x = pd.read_sql(list_one, conn)
    df2 = pd.DataFrame(list_two_x)

    writer = pd.ExcelWriter('answers.xlsx', engine='xlsxwriter')
    df1.to_excel(writer, sheet_name='list_one')
    df2.to_excel(writer, sheet_name='list_two')
    writer.save()
    print('Changes saved')


if __name__ == "__main__":
    save_xlsx()