f3-data-models 0.1.7__tar.gz → 0.1.10__tar.gz

{f3_data_models-0.1.7 → f3_data_models-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: f3-data-models
-Version: 0.1.7
+Version: 0.1.10
 Summary: The data schema and models for F3 Nation applications.
 Home-page: https://github.com/F3-Nation/f3-data-models
 License: MIT
@@ -56,11 +56,26 @@ If you would like to make a change, you will need to:
 source .env && alembic revision --autogenerate -m "Your Message Here"
 ```
 3. Make any edits to the migration script in `alembic/versions`
-4. The github pages documentation will be updated when you push to `main`, but if you would like to preview locally, run:
+4. Run the upgrade on your local db:
+```sh
+source .env && alembic upgrade head
+```
+5. Bump the version on `pyproject.toml`:
+```sh
+poetry version patch[minor][major]
+```
+6. Tag your final commit and make sure to push those tags to trigger the pypi package build:
+```sh
+git tag <new_version> -a -m "Your message here"
+git push origin --tags
+```
+> [!NOTE] The github pages documentation will be updated when you push to `main`, but if you would like to preview locally, run:
+
 ```sh
 poetry run sphinx-build -b html docs docs/_build/html
 cd docs
 poetry run python -m http.server --directory _build/html
 ```
+
 > [!TIP]  
 > Adding new fields as nullable (ie `Optional[]`) has the best chance of reducing breaking changes to the apps.

{f3_data_models-0.1.7 → f3_data_models-0.1.10}/README.md

@@ -27,11 +27,26 @@ If you would like to make a change, you will need to:
 source .env && alembic revision --autogenerate -m "Your Message Here"
 ```
 3. Make any edits to the migration script in `alembic/versions`
-4. The github pages documentation will be updated when you push to `main`, but if you would like to preview locally, run:
+4. Run the upgrade on your local db:
+```sh
+source .env && alembic upgrade head
+```
+5. Bump the version on `pyproject.toml`:
+```sh
+poetry version patch[minor][major]
+```
+6. Tag your final commit and make sure to push those tags to trigger the pypi package build:
+```sh
+git tag <new_version> -a -m "Your message here"
+git push origin --tags
+```
+> [!NOTE] The github pages documentation will be updated when you push to `main`, but if you would like to preview locally, run:
+
 ```sh
 poetry run sphinx-build -b html docs docs/_build/html
 cd docs
 poetry run python -m http.server --directory _build/html
 ```
+
 > [!TIP]  
 > Adding new fields as nullable (ie `Optional[]`) has the best chance of reducing breaking changes to the apps.

{f3_data_models-0.1.7 → f3_data_models-0.1.10}/f3_data_models/models.py

@@ -1,5 +1,5 @@
 from datetime import datetime, date, time
-from typing import Any, Dict, Optional
+from typing import Any, Dict, List, Optional
 from sqlalchemy import (
     JSON,
     TEXT,
@@ -17,6 +17,7 @@ from sqlalchemy.orm import (
     DeclarativeBase,
     mapped_column,
     Mapped,
+    relationship,
 )
 
 # Custom Annotations
@@ -282,7 +283,7 @@ class Org(Base):
 
     __tablename__ = "orgs"
 
-    id: Mapped[intpk]
+    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
     parent_id: Mapped[Optional[int]] = mapped_column(ForeignKey("orgs.id"))
     org_type_id: Mapped[int] = mapped_column(ForeignKey("org_types.id"))
     default_location_id: Mapped[Optional[int]]
@@ -300,6 +301,19 @@ class Org(Base):
     created: Mapped[dt_create]
     updated: Mapped[dt_update]
 
+    event_types: Mapped[Optional[List["EventType"]]] = relationship(
+        "EventType", secondary="event_types_x_org", cascade="expunge"
+    )
+    event_tags: Mapped[Optional[List["EventTag"]]] = relationship(
+        "EventTag", secondary="event_tags_x_org", cascade="expunge"
+    )
+    achievements: Mapped[Optional[List["Achievement"]]] = relationship(
+        "Achievement", secondary="achievements_x_org", cascade="expunge"
+    )
+    parent_org: Mapped[Optional["Org"]] = relationship(
+        "Org", remote_side=[id], cascade="expunge"
+    )
+
 
 class EventType(Base):
     """
@@ -545,6 +559,15 @@ class Event(Base):
     created: Mapped[dt_create]
     updated: Mapped[dt_update]
 
+    org: Mapped[Org] = relationship(innerjoin=True, cascade="expunge")
+    location: Mapped[Location] = relationship(innerjoin=True, cascade="expunge")
+    event_types: Mapped[List[EventType]] = relationship(
+        secondary="events_x_event_types", innerjoin=True, cascade="expunge"
+    )
+    event_tags: Mapped[Optional[List[EventTag]]] = relationship(
+        secondary="event_tags_x_events", cascade="expunge"
+    )
+
 
 class AttendanceType(Base):
     """
@@ -564,32 +587,6 @@ class AttendanceType(Base):
     updated: Mapped[dt_update]
 
 
-class Attendance(Base):
-    """
-    Model representing an attendance record.
-
-    Attributes:
-        id (int): Primary Key of the model.
-        event_id (int): The ID of the associated event.
-        user_id (Optional[int]): The ID of the associated user.
-        is_planned (bool): Whether this is planned attendance (True) vs actual attendance (False).
-        meta (Optional[Dict[str, Any]]): Additional metadata for the attendance.
-        created (datetime): The timestamp when the record was created.
-        updated (datetime): The timestamp when the record was last updated.
-    """
-
-    __tablename__ = "attendance"
-    __table_args__ = (UniqueConstraint("event_id", "user_id", "is_planned"),)
-
-    id: Mapped[intpk]
-    event_id: Mapped[int] = mapped_column(ForeignKey("events.id"))
-    user_id: Mapped[int] = mapped_column(ForeignKey("users.id"))
-    is_planned: Mapped[bool]
-    meta: Mapped[Optional[Dict[str, Any]]]
-    created: Mapped[dt_create]
-    updated: Mapped[dt_update]
-
-
 class Attendance_x_AttenanceType(Base):
     """
     Model representing the association between attendance and attendance types.
@@ -692,6 +689,38 @@ class SlackUser(Base):
     updated: Mapped[dt_update]
 
 
+class Attendance(Base):
+    """
+    Model representing an attendance record.
+
+    Attributes:
+        id (int): Primary Key of the model.
+        event_id (int): The ID of the associated event.
+        user_id (Optional[int]): The ID of the associated user.
+        is_planned (bool): Whether this is planned attendance (True) vs actual attendance (False).
+        meta (Optional[Dict[str, Any]]): Additional metadata for the attendance.
+        created (datetime): The timestamp when the record was created.
+        updated (datetime): The timestamp when the record was last updated.
+    """
+
+    __tablename__ = "attendance"
+    __table_args__ = (UniqueConstraint("event_id", "user_id", "is_planned"),)
+
+    id: Mapped[intpk]
+    event_id: Mapped[int] = mapped_column(ForeignKey("events.id"))
+    user_id: Mapped[int] = mapped_column(ForeignKey("users.id"))
+    is_planned: Mapped[bool]
+    meta: Mapped[Optional[Dict[str, Any]]]
+    created: Mapped[dt_create]
+    updated: Mapped[dt_update]
+
+    event: Mapped[Event] = relationship(innerjoin=True, cascade="expunge")
+    user: Mapped[User] = relationship(innerjoin=True, cascade="expunge", viewonly=True)
+    slack_user: Mapped[Optional[SlackUser]] = relationship(
+        innerjoin=False, cascade="expunge", secondary="users"
+    )
+
+
 class Achievement(Base):
     """
     Model representing an achievement.
@@ -894,3 +923,20 @@ class MagicLinkAuthSession(Base):
     session_token: Mapped[str]
     created: Mapped[dt_create]
     expiration: Mapped[dt_create]
+
+
+# class Org_x_SlackChannel(Base):
+#     """
+#     Model representing the association between organizations (specifically AOs) and Slack channels.
+
+#     Attributes:
+#         org_id (int): The ID of the associated organization.
+#         slack_channel_id (str): The Slack-internal ID of the associated Slack channel.
+#     """
+
+#     __tablename__ = "orgs_x_slack_channels"
+
+#     org_id: Mapped[int] = mapped_column(ForeignKey("orgs.id"), primary_key=True)
+#     slack_channel_id: Mapped[str] = mapped_column(
+#         primary_key=True
+#     )  # Do we need a slack channel table?

{f3_data_models-0.1.7 → f3_data_models-0.1.10}/f3_data_models/utils.py

@@ -3,11 +3,11 @@ from dataclasses import dataclass
 from typing import List, Optional, Tuple, TypeVar
 
 import sqlalchemy
-from sqlalchemy import and_, select
+from sqlalchemy import Select, and_, select
 
 from sqlalchemy.dialects.postgresql import insert
 from sqlalchemy.engine import Engine
-from sqlalchemy.orm import sessionmaker
+from sqlalchemy.orm import sessionmaker, joinedload
 
 from f3_data_models.models import Base
 
@@ -76,30 +76,35 @@ def close_session(session):
 T = TypeVar("T")
 
 
-class DbManager:
-    def get_record(cls: T, id) -> T:
-        session = get_session()
-        try:
-            x = session.query(cls).filter(cls.get_id() == id).first()
-            if x:
-                session.expunge(x)
-            return x
-        finally:
-            session.rollback()
-            close_session(session)
+def _joinedloads(cls: T, query: Select, joinedloads: list | str) -> Select:
+    if joinedloads == "all":
+        joinedloads = [
+            getattr(cls, relationship.key)
+            for relationship in cls.__mapper__.relationships
+        ]
+    return query.options(*[joinedload(load) for load in joinedloads])
 
-    def get(cls: T, id: int) -> T:
+
+class DbManager:
+    def get(cls: T, id: int, joinedloads: list | str = []) -> T:
         session = get_session()
         try:
-            return session.scalars(select(cls).filter(cls.id == id)).one()
+            query = select(cls).filter(cls.id == id)
+            query = _joinedloads(cls, query, joinedloads)
+            return session.scalars(query).unique().one()
         finally:
             session.rollback()
             close_session(session)
 
-    def find_records(cls: T, filters: Optional[List]) -> List[T]:
+    def find_records(
+        cls: T, filters: Optional[List], joinedloads: List | str = []
+    ) -> List[T]:
         session = get_session()
         try:
-            records = session.scalars(select(cls).filter(*filters)).all()
+            query = select(cls)
+            query = _joinedloads(cls, query, joinedloads)
+            query = query.filter(*filters)
+            records = session.scalars(query).unique().all()
             for r in records:
                 session.expunge(r)
             return records
@@ -107,10 +112,15 @@ class DbManager:
             session.rollback()
             close_session(session)
 
-    def find_first_record(cls: T, filters: Optional[List]) -> T:
+    def find_first_record(
+        cls: T, filters: Optional[List], joinedloads: List | str = []
+    ) -> T:
         session = get_session()
         try:
-            record = session.scalars(select(cls).filter(*filters)).first()
+            query = select(cls)
+            query = _joinedloads(cls, query, joinedloads)
+            query = query.filter(*filters)
+            record = session.scalars(query).unique().first()
             if record:
                 session.expunge(record)
             return record

{f3_data_models-0.1.7 → f3_data_models-0.1.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "f3-data-models"
-version = "0.1.7"
+version = "0.1.10"
 description = "The data schema and models for F3 Nation applications."
 authors = ["Evan Petzoldt <evan.petzoldt@protonmail.com>"]
 readme = "README.md"