[rewrite] Minor fixes to models, session generator and example .env

This commit is contained in:
Thastertyn committed 2025-03-10 22:02:30 +01:00
1 parent 71e916586e
commit 272e765ca8
4 files changed
+16 -10

No files matched your search

+1 -2
View File
@@ -9,7 +9,7 @@ from app.core.config import settings
from app.database.exceptions import DatabaseError
import app.database.models
import app.database.models # pylint: disable=unused-import
logger = logging.getLogger(__name__)
@@ -34,7 +34,6 @@ def cleanup() -> None:
engine.dispose()
@contextmanager
def get_session() -> Generator[Session, None, None]:
with Session(engine) as session:
yield session
+2 -2
View File
@@ -26,7 +26,7 @@ class User(SQLModel, table=True):
owned_shops: List["Shop"] = Relationship(back_populates="owner")
registered_shop: List["Shop"] = Relationship(back_populates="registered_users")
role: Optional["UserRole"] = Relationship(back_populates="users")
user_assigned_role: Optional["UserRole"] = Relationship(back_populates="role_users", sa_relationship_kwargs={"foreign_keys": "[User.user_role_id]"})
preferences: Optional["UserPreferences"] = Relationship(back_populates="user")
statistics: Optional["UserStatistics"] = Relationship(back_populates="user_statistics")
@@ -45,7 +45,7 @@ class UserRole(SQLModel, table=True):
id: int = Field(primary_key=True)
name: str = Field(nullable=False, unique=True, max_length=50)
users = Relationship(back_populates="role")
role_users: List["User"] = Relationship(back_populates="user_assigned_role")
class UserStatistics(SQLModel, table=True):