Update authentication system, database models, and UI components
This commit is contained in:
23
database.py
23
database.py
@ -1,14 +1,33 @@
|
||||
# database.py — SQLModel engine/session for PostgreSQL
|
||||
|
||||
from sqlmodel import create_engine, Session, SQLModel
|
||||
from contextlib import contextmanager
|
||||
import os
|
||||
|
||||
# Debug messages disabled
|
||||
|
||||
POSTGRES_URL = os.getenv("DATABASE_URL", "postgresql://d2s:kuTy4ZKs2VcjgDh6@localhost:5432/dictastream")
|
||||
engine = create_engine(POSTGRES_URL, echo=False)
|
||||
engine = create_engine(POSTGRES_URL, echo=False) # Disable echo for production
|
||||
|
||||
# SQLAlchemy Base class for models
|
||||
Base = SQLModel
|
||||
|
||||
@contextmanager
|
||||
def get_db():
|
||||
with Session(engine) as session:
|
||||
"""Session management context manager that ensures proper commit/rollback."""
|
||||
session = Session(engine)
|
||||
try:
|
||||
# Debug messages disabled
|
||||
yield session
|
||||
session.commit()
|
||||
# Debug messages disabled
|
||||
except Exception as e:
|
||||
# Debug messages disabled
|
||||
session.rollback()
|
||||
raise
|
||||
finally:
|
||||
# Debug messages disabled
|
||||
session.close()
|
||||
|
||||
# For backward compatibility
|
||||
get_db_deprecated = get_db
|
||||
|
Reference in New Issue
Block a user