Unverified Commit 48e1356e authored by perf3ct's avatar perf3ct
Browse files

add logging for user upon db connection

parent 5c655f29
from contextvars import ContextVar from contextvars import ContextVar
from peewee import * from peewee import *
import logging
from playhouse.db_url import connect from playhouse.db_url import connect
from playhouse.pool import PooledPostgresqlExtDatabase
from playhouse.pool import PooledSqliteDatabase from config import SRC_LOG_LEVELS
log = logging.getLogger(__name__)
log.setLevel(SRC_LOG_LEVELS["DB"])
db_state_default = {"closed": None, "conn": None, "ctx": None, "transactions": None} db_state_default = {"closed": None, "conn": None, "ctx": None, "transactions": None}
db_state = ContextVar("db_state", default=db_state_default.copy()) db_state = ContextVar("db_state", default=db_state_default.copy())
...@@ -22,11 +26,13 @@ class PeeweeConnectionState(object): ...@@ -22,11 +26,13 @@ class PeeweeConnectionState(object):
def register_connection(db_url): def register_connection(db_url):
db = connect(db_url) db = connect(db_url)
if isinstance(db, PostgresqlDatabase): if isinstance(db, PostgresqlDatabase):
# Directly use PostgresqlDatabase without pooling # Enable autoconnect for SQLite databases, managed by Peewee
db.autoconnect = True db.autoconnect = True
log.info("Connected to PostgreSQL database")
elif isinstance(db, SqliteDatabase): elif isinstance(db, SqliteDatabase):
# Directly use SqliteDatabase without pooling # Enable autoconnect for SQLite databases, managed by Peewee
db.autoconnect = True db.autoconnect = True
log.info("Connected to SQLite database")
else: else:
raise ValueError('Unsupported database connection') raise ValueError('Unsupported database connection')
return db return db
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment