'''
Storage module
'''
from models.house import House
import sqlite3



class Storage():
    def __init__(self, path="storage.db"):
        self.db = sqlite3.connect(path, check_same_thread=False)
        self.cursor = self.db.cursor()
        housestable = """CREATE TABLE IF NOT EXISTS houses (
        	id integer PRIMARY KEY,
	        name text NOT NULL,
	        image text NOT NULL,
	        url text,
	        checked integer
            );"""
        self.db.execute(housestable)
        self.db.commit()

    def close(self):
        self.db.close()

    def AddHouse(self, house : House):
        self.db.execute(f"""
            INSERT INTO houses(name, image, url, checked)
            VALUES
            ('{house.name}', '{house.image}', '{str(house.url)}', {int(house.checked)})
        """)
        self.db.commit()

    def GetHouses(self):
        hs = self.cursor.execute("select * from houses").fetchall()
        ret = []
        for h in hs:
            ret.append(House(id=h[0], url=h[3], name=h[1], image=h[2], checked=h[4]))
        return ret