neroswap/app/library/db.py

23 lines
590 B
Python

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