"vscode:/vscode.git/clone" did not exist on "01c056f09441a8670d0a88f24e2d4fb4a2956ae8"
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 peewee import *
import logging
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 = ContextVar("db_state", default=db_state_default.copy())
......@@ -22,11 +26,13 @@ class PeeweeConnectionState(object):
def register_connection(db_url):
db = connect(db_url)
if isinstance(db, PostgresqlDatabase):
# Directly use PostgresqlDatabase without pooling
# Enable autoconnect for SQLite databases, managed by Peewee
db.autoconnect = True
log.info("Connected to PostgreSQL database")
elif isinstance(db, SqliteDatabase):
# Directly use SqliteDatabase without pooling
# Enable autoconnect for SQLite databases, managed by Peewee
db.autoconnect = True
log.info("Connected to SQLite database")
else:
raise ValueError('Unsupported database connection')
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