diff --git a/wowstash/library/db.py b/wowstash/library/db.py new file mode 100644 index 0000000..e4d5d4d --- /dev/null +++ b/wowstash/library/db.py @@ -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