database.py

import psycopg2


class Database:
    def __init__(self, dbname, user, password):
        self.conn = psycopg2.connect(database=dbname, user=user, password=password)
        self.cur = self.conn.cursor()

    def query(self, query):
        self.cur.execute(query)

    def close(self):
        self.cur.close()
        self.conn.close()