artemis-model 0.1.91__tar.gz → 0.1.92__tar.gz

{artemis_model-0.1.91 → artemis_model-0.1.92}/PKG-INFO

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

{artemis_model-0.1.91 → artemis_model-0.1.92}/artemis_model/schedule.py

@@ -1,8 +1,12 @@
+from datetime import datetime
+from typing import Annotated, NotRequired
+from typing_extensions import TypedDict
 import uuid
 
 from sqlalchemy import ForeignKey
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 from sqlalchemy.types import JSON, UUID
+from pydantic import Field, RootModel, BaseModel
 
 from artemis_model.base import CustomBase, TimeStampMixin, AuditMixin, CustomSyncBase
 
@@ -52,3 +56,27 @@ class SchedulePresetSync(CustomSyncBase, SchedulePresetMixin):
 
 class SchedulePreset(CustomBase, SchedulePresetMixin):
     pass
+
+
+class TimeslotData(TypedDict):
+    """Timeslot data schema."""
+
+    p: NotRequired[list[int]]
+    d: NotRequired[list[int]]
+    scheduled_at: NotRequired[datetime]
+
+
+class Timesheet(RootModel):
+    """Timesheet schema."""
+
+    root: dict[Annotated[str, Field(pattern=r"^\d{2}:\d{2}$")], TimeslotData]
+
+
+class ScheduleItem(BaseModel):
+    """Schedule item schema."""
+
+    zone_id: int
+    timesheet: Timesheet
+
+
+__all__ = ["ScheduleItem"]

artemis_model-0.1.92/artemis_model/sqs/__init__.py

@@ -0,0 +1 @@
+"""SQS model."""

artemis_model-0.1.92/artemis_model/sqs/messages.py

@@ -0,0 +1,91 @@
+"""SQS messages."""
+
+from datetime import datetime, timezone
+from enum import Enum
+from typing import Literal
+
+from artemis_model.redis.zone_state import SessionId
+from artemis_model.schedule import ScheduleItem
+from pydantic import BaseModel, Field
+
+
+class Action(str, Enum):
+    """Message type enum."""
+
+    CALCULATE_ALL_SCHEDULES = "calculate-all-schedules"
+    TRIAGE_SCHEDULE = "triage-schedule"
+    TRIGGER_DATABASE_MIGRATION = "trigger-database-migration"
+    MOVE_TIME_SLOT = "move-time-slot"
+    PUSH_PLAYLIST = "push-playlist"
+    EXPIRE_PUSHED_PLAYLIST = "expire-pushed-playlist"
+    REFRESH_TRACK_BUCKET = "refresh-track-bucket"
+    RECALCULATE_SCHEDULE = "recalculate-schedule"
+    PLAYER_MODE_CHANGE = "player-mode-change"
+
+
+class BaseMessage(BaseModel):
+    """Base message for incoming messages schema."""
+
+    ts: str = Field(
+        default_factory=lambda: datetime.now(timezone.utc).isoformat(),
+        description="Timestamp value of the message",
+    )
+    action: Action
+
+
+class MoveTimeSlotIn(BaseMessage):
+    """Move to time slot message schema."""
+
+    action: Action = Action.MOVE_TIME_SLOT
+    zone_id: int
+    timeslot: str
+
+
+class TriageScheduleIn(BaseMessage):
+    """Triage schedule message schema."""
+
+    action: Action = Action.TRIAGE_SCHEDULE
+    schedule: list[ScheduleItem]
+
+
+class CalculateAllSchedulesIn(BaseMessage):
+    """Calculate all schedules message schema."""
+
+    action: Action = Action.CALCULATE_ALL_SCHEDULES
+
+
+class PushPlaylistIn(BaseMessage):
+    """Push playlist message schema."""
+
+    action: Action = Action.PUSH_PLAYLIST
+    zone_id: int
+    playlist_ids: list[int]
+    expire_at: datetime | None = Field(
+        default=None, description="The datetime that the playlist will expire"
+    )
+    session_id: SessionId = Field(
+        default_factory=lambda: int(datetime.now(timezone.utc).timestamp()),
+    )
+
+
+class PushPlaylistExpireIn(BaseMessage):
+    """Push playlist expire message schema."""
+
+    action: Action = Action.EXPIRE_PUSHED_PLAYLIST
+    zone_id: int
+    new_mode: Literal["pushplaylist", "scheduled"] = "scheduled"
+    session_id: SessionId | None = None
+
+
+class RefreshTrackBucketIn(BaseMessage):
+    """Refresh track bucket message schema."""
+
+    action: Action = Action.REFRESH_TRACK_BUCKET
+    zone_id: int
+
+
+class RecalculateScheduleIn(BaseMessage):
+    """Recalculate schedule message schema."""
+
+    action: Action = Action.RECALCULATE_SCHEDULE
+    zone_id: int

{artemis_model-0.1.91 → artemis_model-0.1.92}/pyproject.toml

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