feat(01-01): create package structure, Pydantic models, and device abstractions
- pyproject.toml with FastAPI, uvicorn, Pydantic v2, aiofiles, structlog deps - DeviceConfig Pydantic model with StripType, UUID, validation constraints - ShowModel with AudioRef, TrackModel, CueModel, AnalysisBlock, schema_version=1 - BaseDevice ABC with encode_frame, encode_animation_cmd, bytes_per_pixel - SK6812Device (4 bytes/pixel) and WS2801Device (3 bytes/pixel) implementations
This commit is contained in:
1
lightsync/__init__.py
Normal file
1
lightsync/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
__version__ = "0.1.0"
|
||||
0
lightsync/devices/__init__.py
Normal file
0
lightsync/devices/__init__.py
Normal file
33
lightsync/devices/base.py
Normal file
33
lightsync/devices/base.py
Normal file
@@ -0,0 +1,33 @@
|
||||
from abc import ABC, abstractmethod
|
||||
from lightsync.models.device import DeviceConfig
|
||||
|
||||
|
||||
class BaseDevice(ABC):
|
||||
def __init__(self, config: DeviceConfig):
|
||||
self.config = config
|
||||
|
||||
@property
|
||||
def id(self):
|
||||
return self.config.id
|
||||
|
||||
@property
|
||||
def led_count(self) -> int:
|
||||
return self.config.led_count
|
||||
|
||||
@abstractmethod
|
||||
def encode_frame(self, pixels: list[tuple]) -> bytes:
|
||||
"""Encode a full pixel frame to UDP payload bytes.
|
||||
pixels: list of (R, G, B) or (R, G, B, W) tuples.
|
||||
"""
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def encode_animation_cmd(self, animation: str, params: dict) -> bytes:
|
||||
"""Encode an animation+params command packet."""
|
||||
...
|
||||
|
||||
@property
|
||||
@abstractmethod
|
||||
def bytes_per_pixel(self) -> int:
|
||||
"""Number of bytes per LED in frame mode (3 for RGB, 4 for RGBW)."""
|
||||
...
|
||||
28
lightsync/devices/sk6812.py
Normal file
28
lightsync/devices/sk6812.py
Normal file
@@ -0,0 +1,28 @@
|
||||
from lightsync.devices.base import BaseDevice
|
||||
from lightsync.models.device import DeviceConfig
|
||||
|
||||
|
||||
class SK6812Device(BaseDevice):
|
||||
"""SK6812 RGBW LED strip — 4 bytes per pixel (R, G, B, W)."""
|
||||
|
||||
def __init__(self, config: DeviceConfig):
|
||||
super().__init__(config)
|
||||
|
||||
@property
|
||||
def bytes_per_pixel(self) -> int:
|
||||
return 4
|
||||
|
||||
def encode_frame(self, pixels: list[tuple]) -> bytes:
|
||||
"""Encode pixel list to SK6812 frame bytes.
|
||||
Each pixel: (r, g, b, w) tuple — 4 bytes per LED.
|
||||
"""
|
||||
buf = bytearray()
|
||||
for pixel in pixels:
|
||||
r, g, b = pixel[0], pixel[1], pixel[2]
|
||||
w = pixel[3] if len(pixel) > 3 else 0
|
||||
buf.extend([r & 0xFF, g & 0xFF, b & 0xFF, w & 0xFF])
|
||||
return bytes(buf)
|
||||
|
||||
def encode_animation_cmd(self, animation: str, params: dict) -> bytes:
|
||||
"""Stub — Phase 3 implements real animation command encoding."""
|
||||
return b""
|
||||
27
lightsync/devices/ws2801.py
Normal file
27
lightsync/devices/ws2801.py
Normal file
@@ -0,0 +1,27 @@
|
||||
from lightsync.devices.base import BaseDevice
|
||||
from lightsync.models.device import DeviceConfig
|
||||
|
||||
|
||||
class WS2801Device(BaseDevice):
|
||||
"""WS2801 RGB LED strip — 3 bytes per pixel (R, G, B)."""
|
||||
|
||||
def __init__(self, config: DeviceConfig):
|
||||
super().__init__(config)
|
||||
|
||||
@property
|
||||
def bytes_per_pixel(self) -> int:
|
||||
return 3
|
||||
|
||||
def encode_frame(self, pixels: list[tuple]) -> bytes:
|
||||
"""Encode pixel list to WS2801 frame bytes.
|
||||
Each pixel: (r, g, b) tuple — 3 bytes per LED.
|
||||
"""
|
||||
buf = bytearray()
|
||||
for pixel in pixels:
|
||||
r, g, b = pixel[0], pixel[1], pixel[2]
|
||||
buf.extend([r & 0xFF, g & 0xFF, b & 0xFF])
|
||||
return bytes(buf)
|
||||
|
||||
def encode_animation_cmd(self, animation: str, params: dict) -> bytes:
|
||||
"""Stub — Phase 3 implements real animation command encoding."""
|
||||
return b""
|
||||
4
lightsync/models/__init__.py
Normal file
4
lightsync/models/__init__.py
Normal file
@@ -0,0 +1,4 @@
|
||||
from lightsync.models.device import DeviceConfig
|
||||
from lightsync.models.show import ShowModel
|
||||
|
||||
__all__ = ["DeviceConfig", "ShowModel"]
|
||||
17
lightsync/models/device.py
Normal file
17
lightsync/models/device.py
Normal file
@@ -0,0 +1,17 @@
|
||||
from uuid import UUID, uuid4
|
||||
from typing import Literal
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
StripType = Literal["sk6812", "ws2801", "generic"]
|
||||
|
||||
|
||||
class DeviceConfig(BaseModel):
|
||||
id: UUID = Field(default_factory=uuid4)
|
||||
name: str
|
||||
strip_type: StripType
|
||||
led_count: int = Field(gt=0, le=1000)
|
||||
ip: str
|
||||
port: int = Field(ge=1, le=65535)
|
||||
enabled: bool = True
|
||||
|
||||
model_config = {"populate_by_name": True}
|
||||
46
lightsync/models/show.py
Normal file
46
lightsync/models/show.py
Normal file
@@ -0,0 +1,46 @@
|
||||
from uuid import UUID, uuid4
|
||||
from datetime import datetime, timezone
|
||||
from typing import Any, Literal
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
from lightsync.models.device import DeviceConfig
|
||||
|
||||
|
||||
class AudioRef(BaseModel):
|
||||
source_type: Literal["file", "youtube"] = "file"
|
||||
path: str | None = None
|
||||
yt_url: str | None = None
|
||||
duration_seconds: float | None = None
|
||||
|
||||
|
||||
class CueModel(BaseModel):
|
||||
id: UUID = Field(default_factory=uuid4)
|
||||
timestamp: float
|
||||
mode: Literal["animation", "frame_sequence"] = "animation"
|
||||
animation: str | None = None
|
||||
params: dict[str, Any] = Field(default_factory=dict)
|
||||
|
||||
|
||||
class TrackModel(BaseModel):
|
||||
device_id: UUID
|
||||
cues: list[CueModel] = Field(default_factory=list)
|
||||
|
||||
|
||||
class AnalysisBlock(BaseModel):
|
||||
analysed_at: datetime | None = None
|
||||
tempo_bpm: float | None = None
|
||||
beat_times: list[float] = Field(default_factory=list)
|
||||
onset_times: list[float] = Field(default_factory=list)
|
||||
|
||||
|
||||
class ShowModel(BaseModel):
|
||||
schema_version: int = 1
|
||||
id: UUID = Field(default_factory=uuid4)
|
||||
name: str
|
||||
created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
|
||||
updated_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
|
||||
audio: AudioRef = Field(default_factory=AudioRef)
|
||||
devices: list[DeviceConfig] = Field(default_factory=list)
|
||||
analysis: AnalysisBlock = Field(default_factory=AnalysisBlock)
|
||||
tracks: list[TrackModel] = Field(default_factory=list)
|
||||
ai_sequences: list[Any] = Field(default_factory=list)
|
||||
21
pyproject.toml
Normal file
21
pyproject.toml
Normal file
@@ -0,0 +1,21 @@
|
||||
[build-system]
|
||||
requires = ["hatchling"]
|
||||
build-backend = "hatchling.build"
|
||||
|
||||
[project]
|
||||
name = "lightsync"
|
||||
version = "0.1.0"
|
||||
requires-python = ">=3.11"
|
||||
dependencies = [
|
||||
"fastapi>=0.115.0",
|
||||
"uvicorn[standard]>=0.34.0",
|
||||
"pydantic>=2.0.0",
|
||||
"aiofiles>=24.0.0",
|
||||
"python-dotenv>=1.0.0",
|
||||
"structlog>=24.0.0",
|
||||
]
|
||||
|
||||
[project.optional-dependencies]
|
||||
dev = [
|
||||
"httpx>=0.27.0",
|
||||
]
|
||||
Reference in New Issue
Block a user