artemis-model 0.1.95__tar.gz → 0.1.97__tar.gz

{artemis_model-0.1.95 → artemis_model-0.1.97}/PKG-INFO

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

{artemis_model-0.1.95 → artemis_model-0.1.97}/artemis_model/redis/__init__.py

@@ -2,6 +2,7 @@
 
 from .zone_state import ZoneState, NowPlaying, SessionId, BucketId
 from .device import ActiveDevice
+from .bucket import RedisTrackBucketItem
 from .keys import (
     KEY_ZONE_PLAY_HISTORY_TEMPLATE,
     KEY_ZONE_PUSH_PLAYLIST_BUCKET_TEMPLATE,
@@ -21,4 +22,5 @@ __all__ = [
     "KEY_ZONE_SCHEDULE_BUCKET_TEMPLATE",
     "KEY_ZONE_STATE_TEMPLATE",
     "KEY_ZONE_ACTIVE_DEVICE_TEMPLATE",
+    "RedisTrackBucketItem",
 ]

artemis_model-0.1.97/artemis_model/redis/bucket.py

@@ -0,0 +1,54 @@
+"""Redis track buckets"""
+from pydantic import BaseModel, Field, ConfigDict
+from typing import Annotated
+from datetime import datetime, timezone
+import uuid
+
+
+TrackId = Annotated[
+    uuid.UUID,
+    Field(description="Track ID (UUID)")
+]
+
+PlaylistId = Annotated[
+    int,
+    Field(description="Playlist ID (integer)")
+]
+
+Timestamp = Annotated[
+    int,
+    Field(description="Unix timestamp")
+]
+
+
+class RedisTrackBucketItem(BaseModel):
+    """
+    Represents a composite Redis key along with a timestamp.
+    The Redis entry will be:
+    { '{"track_id": "...", "playlist_id": ...}': ts }
+    ts is the timestamp of utc now.
+    """
+
+    track_id: TrackId
+    playlist_id: PlaylistId
+    ts: Timestamp = Field(default_factory=lambda: int(datetime.now(timezone.utc).timestamp()))
+
+    model_config = ConfigDict(
+        json_encoders={uuid.UUID: str},
+        populate_by_name=True
+    )
+
+    def as_redis_entry(self) -> dict[str, int]:
+        """
+        Generate a Redis-ready dictionary:
+        { '{"track_id": "...", "playlist_id": ...}': ts }
+        """
+        key = self.model_dump_json(include={"track_id", "playlist_id"})
+        return {key: self.ts}
+    
+    @classmethod
+    def from_redis_entry(cls, entry: dict[str, int]) -> "RedisTrackBucketItem":
+        """
+        Parse a Redis entry into a RedisTrackBucketItem.
+        """
+        return cls.model_validate_json(list(entry.keys())[0])

{artemis_model-0.1.95 → artemis_model-0.1.97}/artemis_model/redis/keys.py

@@ -6,6 +6,7 @@ KEY_ZONE_SCHEDULE_BUCKET_TEMPLATE = "zone_{zone_id}_sc_bucket_{timeslot_ts}"
 KEY_ZONE_STATE_TEMPLATE = "zone_{zone_id}_state"
 KEY_ZONE_ACTIVE_DEVICE_TEMPLATE = "zone_{zone_id}_active_devices"
 
+
 __all__ = [
     "KEY_ZONE_PLAY_HISTORY_TEMPLATE",
     "KEY_ZONE_PUSH_PLAYLIST_BUCKET_TEMPLATE",

artemis_model-0.1.97/artemis_model/redis/play_history

@@ -0,0 +1,43 @@
+"""Redis play history"""
+
+from pydantic import BaseModel, Field
+from typing import Annotated
+from datetime import datetime, timezone
+
+
+TrackId = Annotated[
+    str,
+    Field(description="Track ID (UUID or string)")
+]
+
+Timestamp = Annotated[
+    int,
+    Field(description="Unix timestamp")
+]
+
+
+class TrackPlayHistoryValue(BaseModel):
+    """
+    Represents the value stored in Redis sorted set:
+    A string like "track_id#timestamp"
+    """
+    track_id: TrackId
+    ts: Timestamp = Field(default_factory=lambda: int(datetime.now(timezone.utc).timestamp()))
+
+    def as_redis_entry(self) -> str:
+        """
+        Generate a Redis-ready string:
+        "track_id#timestamp"
+        """
+        return f"{self.track_id}#{self.ts}"
+
+    @classmethod
+    def from_redis_entry(cls, value: str) -> "TrackPlayHistoryValue":
+        """
+        Parse a Redis entry into a TrackPlayHistoryValue.
+        """
+        try:
+            track_id, ts = value.rsplit("#", 1)
+            return cls(track_id=track_id, ts=int(ts))
+        except Exception as e:
+            raise ValueError(f"Invalid format for TrackPlayHistoryValue: {value}") from e

{artemis_model-0.1.95 → artemis_model-0.1.97}/pyproject.toml

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