adding simple db class for db operations

mm-logging
lza_menace 4 years ago
parent 9c0638da7a
commit 0d1d3c9506

@ -0,0 +1,22 @@
from psycopg2 import Error as PGError
from psycopg2 import connect as PGConnect
from wowstash import config
class Database(object):
def __init__(self):
self.conn = PGConnect(
user=config.DB_USER,
password=config.DB_PASS,
host=config.DB_HOST,
port=config.DB_PORT,
database=config.DB_NAME
)
cursor = self.conn.cursor()
cursor.execute("SELECT VERSION()")
results = cursor.fetchone()
if results:
self.connected = True
else:
self.connected = False