# -*- coding: utf-8 -*-
import psycopg2.extras
from datetime import datetime


def check_db_status(time_db_cursor, time_db_dbname):

    # if the db is not found, then try to create it
    # try:
    #     time_db_cursor.execute("create database " + time_db_dbname);
    # except Exception as e:
    #     print("Database possibly exists: " + e)

    try:
        sql = """CREATE TABLE IF NOT EXISTS mltd(
           uuid serial PRIMARY KEY,
           asset_id VARCHAR (100) NOT NULL,
           probability REAL NOT NULL,
           timeframe VARCHAR (355) NOT NULL,
           created_on TIMESTAMP NOT NULL
        );"""
        time_db_cursor.execute(sql);
    except Exception as e:
        print("Table can't be created: " + e)
        return False
    return True


def connect(
    time_db_host, time_db_port, time_db_dbname, time_db_username, time_db_password, ssl
):
    conn = psycopg2.connect(
        dbname=time_db_dbname,
        user=time_db_username,
        password=time_db_password,
        host=time_db_host,
    )
    cursor = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    check_db_status(cursor, time_db_dbname)
    cursor.close()
    conn.commit()
    return conn


def report(conn, asset_id, risk, timeframe):
    sql = """INSERT INTO mltd(asset_id,probability,timeframe,created_on)
                 VALUES(%s,%s,%s,%s);"""
    cursor = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    cursor.execute(sql, (asset_id, risk, timeframe, datetime.now()))
    conn.commit()