from flask import g
import psycopg2
try: from marrow_config import config
except ImportError:
class config:
class db:
db = "marrow"
user = "marrow"
password = "marrowpass"
host = "pgsqlserver.elangley.org"
def get_db(close=True):
db = getattr(g, '_database', None)
_config = config.db
if db is None:
db = g._database = [psycopg2.connect(
database=_config.db,
user=_config.user,
password=_config.password,
host=_config.host
),close];
return db[0]
def close_connection(exception):
db = getattr(g, '_database', None)
if db is not None and db[1]:
db[0].close()
def check_ak(db, username, ak):
with db.cursor() as cur:
cur.callproc('check_ak', (username, ak))
return cur.rowcount > 0