Initial commit: Portal Auth Admin Dashboard
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
25
db.py
Normal file
25
db.py
Normal file
@@ -0,0 +1,25 @@
|
||||
import psycopg2
|
||||
from psycopg2.extras import RealDictCursor
|
||||
from contextlib import contextmanager
|
||||
import config
|
||||
|
||||
@contextmanager
|
||||
def get_cursor():
|
||||
conn = psycopg2.connect(
|
||||
host=config.DB_AUTH_HOST,
|
||||
port=config.DB_AUTH_PORT,
|
||||
dbname=config.DB_AUTH_NAME,
|
||||
user=config.DB_AUTH_USER,
|
||||
password=config.DB_AUTH_PASSWORD,
|
||||
options="-c timezone=UTC",
|
||||
)
|
||||
try:
|
||||
cur = conn.cursor(cursor_factory=RealDictCursor)
|
||||
yield cur
|
||||
conn.commit()
|
||||
except Exception:
|
||||
conn.rollback()
|
||||
raise
|
||||
finally:
|
||||
cur.close()
|
||||
conn.close()
|
||||
Reference in New Issue
Block a user