artemis-model 0.1.147__tar.gz → 0.1.149__tar.gz

{artemis_model-0.1.147 → artemis_model-0.1.149}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: artemis-model
-Version: 0.1.147
+Version: 0.1.149
 Summary: 
 Author: Jukeboxy
 Requires-Python: >=3.10.6,<4.0.0

{artemis_model-0.1.147 → artemis_model-0.1.149}/artemis_model/__init__.py

@@ -5,6 +5,7 @@ from artemis_model.category import *  # noqa
 from artemis_model.dj_set import *  # noqa
 from artemis_model.genre import *  # noqa
 from artemis_model.location import *  # noqa
+from artemis_model.location_genre_exclusion import *  # noqa
 from artemis_model.message import *  # noqa
 from artemis_model.organization import *  # noqa
 from artemis_model.playlist import *  # noqa

{artemis_model-0.1.147 → artemis_model-0.1.149}/artemis_model/location.py

@@ -86,6 +86,14 @@ class LocationMixin(TimeStampMixin):
             cascade="all, delete-orphan",
         )
 
+    @declared_attr
+    def genre_exclusions(cls) -> Mapped[list["LocationGenreExclusion"]]:
+        return relationship(
+            "LocationGenreExclusion",
+            back_populates="location",
+            cascade="all, delete-orphan",
+        )
+
     timezone: Mapped[str] = mapped_column(nullable=False, default="America/New_York")
 
 

artemis_model-0.1.149/artemis_model/location_genre_exclusion.py

@@ -0,0 +1,38 @@
+import uuid
+from sqlalchemy import ForeignKey, UniqueConstraint
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+from sqlalchemy.ext.declarative import declared_attr
+
+from artemis_model.base import TimeStampMixin, CustomSyncBase, CustomBase
+
+
+class LocationGenreExclusionMixin(TimeStampMixin):
+    """Association table for tracking which genres are excluded at the location level."""
+
+    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
+    location_id: Mapped[uuid.UUID] = mapped_column(
+        ForeignKey("location.id", ondelete="CASCADE"), nullable=False, index=True
+    )
+    genre_id: Mapped[int] = mapped_column(
+        ForeignKey("genre.id", ondelete="CASCADE"), nullable=False, index=True
+    )
+
+    @declared_attr
+    def location(cls) -> Mapped["Location"]:
+        return relationship(back_populates="genre_exclusions")
+
+    @declared_attr
+    def genre(cls) -> Mapped["Genre"]:
+        return relationship()
+
+    __table_args__ = (
+        UniqueConstraint("location_id", "genre_id", name="unique_location_genre_exclusion"),
+    )
+
+
+class LocationGenreExclusionSync(CustomSyncBase, LocationGenreExclusionMixin):
+    pass
+
+
+class LocationGenreExclusion(CustomBase, LocationGenreExclusionMixin):
+    pass

{artemis_model-0.1.147 → artemis_model-0.1.149}/artemis_model/sqs/messages.py

@@ -103,6 +103,7 @@ class BanTrackIn(BaseMessage):
     zone_id: int
     track_id: UUID
     playlist_id: int
+    bucket_key: str
 
 
 class ZoneActivityIn(BaseMessage):

{artemis_model-0.1.147 → artemis_model-0.1.149}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "artemis-model"
-version = "0.1.147"
+version = "0.1.149"
 description = ""
 authors = ["Jukeboxy"]
 readme = "README.md"