adding simple db class for db operations
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
|
Loading…
Reference in New Issue