artemis-model 0.1.90__py3-none-any.whl → 0.1.92__py3-none-any.whl

artemis_model/redis/zone_state.py

@@ -1,9 +1,8 @@
 """Zone state schema."""
 
-from dataclasses import Field
 from typing import Annotated, Literal
 
-from pydantic import BaseModel, field_validator
+from pydantic import BaseModel, Field, field_validator
 
 
 class NowPlaying(BaseModel):

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/sqs/__init__.py

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

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.90.dist-info → artemis_model-0.1.92.dist-info}/METADATA

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

{artemis_model-0.1.90.dist-info → artemis_model-0.1.92.dist-info}/RECORD

@@ -14,12 +14,14 @@ artemis_model/permission.py,sha256=Bn1Bg1aCS4Z4_3tqEqvtrzqAYDCImsvmGyIEMoVycEk,1
 artemis_model/playlist.py,sha256=L22KSJgTZmNEc9CqM3TFD4Wnx3TW77GQJ5EFLJCxZ20,6393
 artemis_model/redis/__init__.py,sha256=tLPIfZSrHCYBuldoQbFVTb5PYffo0l76TBj-cemggzg,201
 artemis_model/redis/device.py,sha256=ZUiIyVmtWiEwayFIg9mktiFaVuHQ7Fg0rvSjjWyryCI,484
-artemis_model/redis/zone_state.py,sha256=rRoyK-VTj-1SxLlORopO5RiXuRxUgAeSFbVG3_t7bQM,1735
-artemis_model/schedule.py,sha256=pnoKDMlUh28o61I7kZddNYn6z-uqOt_ml1nbWxKVeRM,1722
+artemis_model/redis/zone_state.py,sha256=PgHiGUZ2WI3bqO-8CW2cwzLEQu3l4wJAzIABSWErfBM,1712
+artemis_model/schedule.py,sha256=BbiRqpsRReMQtx0KnCUWwSTAwdEPMqQggxpNR12mKmY,2323
 artemis_model/setting.py,sha256=EWzOmE31mLNWwmV_wmqttHPMycvQZW7SJUIozF2b8bo,1190
+artemis_model/sqs/__init__.py,sha256=nHpXQns64qQ5Cqjyo6w9fDGO_wWhprqn1bhKf3eWnio,17
+artemis_model/sqs/messages.py,sha256=98cWxrj48T1AcEnltqVGTts6rcmTYimeN8fC6wkdNQk,2452
 artemis_model/track.py,sha256=QwUF0QKVn1I64648B-NI75-IzGQvnLt9B0emD4GnS6E,3757
 artemis_model/user.py,sha256=eqIdCiBJRNLjCwPPCn-gQ6si0O5JUBGfp9oWJL5zVW4,2131
 artemis_model/zone.py,sha256=iGRUtzUwKh9LHT3MOfzzg1DnkPBts_ZBzZVTi2EmIgs,2282
-artemis_model-0.1.90.dist-info/METADATA,sha256=iw4LR2vVj3BmgFn84Wx2FehKux9C1IgZI0eW2KRPz_s,3444
-artemis_model-0.1.90.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-artemis_model-0.1.90.dist-info/RECORD,,
+artemis_model-0.1.92.dist-info/METADATA,sha256=2t9bSqu2hdefShf6kvgDTnNCk6D-f0lrxUs_6kzrHts,3444
+artemis_model-0.1.92.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+artemis_model-0.1.92.dist-info/RECORD,,