homesec 1.2.1__py3-none-any.whl → 1.2.2__py3-none-any.whl

homesec/models/__init__.py

@@ -25,7 +25,9 @@ from homesec.models.config import (
 )
 from homesec.models.enums import RiskLevel, RiskLevelField
 from homesec.models.filter import FilterConfig, FilterOverrides, FilterResult, YoloFilterSettings
-from homesec.models.source import FtpSourceConfig, LocalFolderSourceConfig, RTSPSourceConfig
+from homesec.models.source.ftp import FtpSourceConfig
+from homesec.models.source.local_folder import LocalFolderSourceConfig
+from homesec.models.source.rtsp import RTSPSourceConfig
 from homesec.models.vlm import (
     AnalysisResult,
     EntityTimeline,

homesec/models/config.py

@@ -8,7 +8,9 @@ from pydantic import BaseModel, Field, field_validator, model_validator
 
 from homesec.models.enums import RiskLevel, RiskLevelField
 from homesec.models.filter import FilterConfig
-from homesec.models.source import FtpSourceConfig, LocalFolderSourceConfig, RTSPSourceConfig
+from homesec.models.source.ftp import FtpSourceConfig
+from homesec.models.source.local_folder import LocalFolderSourceConfig
+from homesec.models.source.rtsp import RTSPSourceConfig
 from homesec.models.vlm import VLMConfig
 
 

homesec/models/source/__init__.py

@@ -0,0 +1,3 @@
+"""Source configuration models."""
+
+__all__: list[str] = []

homesec/models/source/ftp.py

@@ -0,0 +1,97 @@
+"""FTP source configuration model."""
+
+from __future__ import annotations
+
+from pydantic import BaseModel, Field, field_validator
+
+
+class FtpSourceConfig(BaseModel):
+    """FTP source configuration."""
+
+    model_config = {"extra": "forbid"}
+
+    camera_name: str | None = Field(
+        default=None,
+        description="Optional human-friendly camera name.",
+    )
+    host: str = Field(
+        default="0.0.0.0",
+        description="FTP bind address.",
+    )
+    port: int = Field(
+        default=2121,
+        ge=0,
+        le=65535,
+        description="FTP listen port (0 lets the OS choose an ephemeral port).",
+    )
+    root_dir: str = Field(
+        default="./ftp_incoming",
+        description="FTP root directory for uploads.",
+    )
+    ftp_subdir: str | None = Field(
+        default=None,
+        description="Optional subdirectory under root_dir.",
+    )
+    anonymous: bool = Field(
+        default=True,
+        description="Allow anonymous FTP uploads.",
+    )
+    username_env: str | None = Field(
+        default=None,
+        description="Environment variable containing FTP username.",
+    )
+    password_env: str | None = Field(
+        default=None,
+        description="Environment variable containing FTP password.",
+    )
+    perms: str = Field(
+        default="elw",
+        description="pyftpdlib permissions string.",
+    )
+    passive_ports: str | None = Field(
+        default=None,
+        description="Passive ports range (e.g., '60000-60100' or '60000,60010').",
+    )
+    masquerade_address: str | None = Field(
+        default=None,
+        description="Optional masquerade address for passive mode.",
+    )
+    heartbeat_s: float = Field(
+        default=30.0,
+        ge=0.0,
+        description="Seconds between FTP health checks.",
+    )
+    allowed_extensions: list[str] = Field(
+        default_factory=lambda: [".mp4"],
+        description="Allowed file extensions for uploaded clips.",
+    )
+    delete_non_matching: bool = Field(
+        default=True,
+        description="Delete files with disallowed extensions.",
+    )
+    delete_incomplete: bool = Field(
+        default=True,
+        description="Delete incomplete uploads when enabled.",
+    )
+    default_duration_s: float = Field(
+        default=10.0,
+        ge=0.0,
+        description="Fallback clip duration when timestamps are missing.",
+    )
+    log_level: str = Field(
+        default="INFO",
+        description="FTP server log level.",
+    )
+
+    @field_validator("allowed_extensions")
+    @classmethod
+    def _normalize_extensions(cls, value: list[str]) -> list[str]:
+        cleaned: list[str] = []
+        for item in value:
+            ext = str(item).strip().lower()
+            if not ext:
+                continue
+            if not ext.startswith("."):
+                ext = f".{ext}"
+            cleaned.append(ext)
+        return cleaned

homesec/models/source/local_folder.py

@@ -0,0 +1,30 @@
+"""Local folder source configuration model."""
+
+from __future__ import annotations
+
+from pydantic import BaseModel, Field
+
+
+class LocalFolderSourceConfig(BaseModel):
+    """Local folder source configuration."""
+
+    model_config = {"extra": "forbid"}
+
+    camera_name: str | None = Field(
+        default=None,
+        description="Optional human-friendly camera name.",
+    )
+    watch_dir: str = Field(
+        default="recordings",
+        description="Directory to watch for new clips.",
+    )
+    poll_interval: float = Field(
+        default=1.0,
+        ge=0.0,
+        description="Polling interval in seconds.",
+    )
+    stability_threshold_s: float = Field(
+        default=3.0,
+        ge=0.0,
+        description="Seconds to wait for file size to stabilize before accepting a clip.",
+    )

homesec/models/source/rtsp.py

@@ -0,0 +1,165 @@
+"""RTSP source configuration models."""
+
+from __future__ import annotations
+
+from pydantic import BaseModel, Field, model_validator
+
+
+class RTSPMotionConfig(BaseModel):
+    """Motion detection configuration."""
+
+    model_config = {"extra": "forbid"}
+
+    pixel_threshold: int = Field(
+        default=45,
+        ge=0,
+        description="Pixel intensity delta required to count a pixel as changed.",
+    )
+    min_changed_pct: float = Field(
+        default=1.0,
+        ge=0.0,
+        description="Percent of pixels that must change to trigger motion (idle state).",
+    )
+    recording_sensitivity_factor: float = Field(
+        default=2.0,
+        ge=1.0,
+        description="Factor to reduce the threshold while recording (>=1.0).",
+    )
+    blur_kernel: int = Field(
+        default=5,
+        ge=0,
+        description="Gaussian blur kernel size (odd or zero; even values are normalized).",
+    )
+
+
+class RTSPRecordingConfig(BaseModel):
+    """Recording lifecycle configuration."""
+
+    model_config = {"extra": "forbid"}
+
+    stop_delay: float = Field(
+        default=10.0,
+        ge=0.0,
+        description="Seconds to keep recording after motion stops.",
+    )
+    max_recording_s: float = Field(
+        default=60.0,
+        gt=0.0,
+        description="Maximum seconds per recording before rotating.",
+    )
+
+
+class RTSPStreamConfig(BaseModel):
+    """RTSP/ffmpeg transport configuration."""
+
+    model_config = {"extra": "forbid"}
+
+    connect_timeout_s: float = Field(
+        default=2.0,
+        ge=0.0,
+        description="RTSP connect timeout (seconds) passed to ffmpeg/ffprobe when supported.",
+    )
+    io_timeout_s: float = Field(
+        default=2.0,
+        ge=0.0,
+        description="RTSP I/O timeout (seconds) passed to ffmpeg/ffprobe when supported.",
+    )
+    ffmpeg_flags: list[str] = Field(
+        default_factory=list,
+        description="Additional ffmpeg flags appended to the command.",
+    )
+    disable_hwaccel: bool = Field(
+        default=False,
+        description="Disable hardware-accelerated decoding.",
+    )
+
+
+class RTSPReconnectConfig(BaseModel):
+    """Reconnect and fallback policy."""
+
+    model_config = {"extra": "forbid"}
+
+    max_attempts: int = Field(
+        default=0,
+        ge=0,
+        description="Max reconnect attempts (0 = retry forever).",
+    )
+    backoff_s: float = Field(
+        default=1.0,
+        ge=0.0,
+        description="Base backoff (seconds) between reconnect attempts.",
+    )
+    detect_fallback_attempts: int = Field(
+        default=3,
+        ge=0,
+        description="Failures before falling back from detect stream to main stream.",
+    )
+
+
+class RTSPRuntimeConfig(BaseModel):
+    """Runtime loop configuration."""
+
+    model_config = {"extra": "forbid"}
+
+    frame_timeout_s: float = Field(
+        default=2.0,
+        ge=0.0,
+        description="Seconds without frames before considering the pipeline stalled.",
+    )
+    frame_queue_size: int = Field(
+        default=20,
+        ge=1,
+        description="Frame queue size used by the frame reader thread.",
+    )
+    heartbeat_s: float = Field(
+        default=30.0,
+        ge=0.0,
+        description="Seconds between heartbeat logs.",
+    )
+    debug_motion: bool = Field(
+        default=False,
+        description="Enable verbose motion detection logging.",
+    )
+
+
+class RTSPSourceConfig(BaseModel):
+    """RTSP source configuration."""
+
+    model_config = {"extra": "forbid"}
+
+    camera_name: str | None = Field(
+        default=None,
+        description="Optional human-friendly camera name.",
+    )
+    rtsp_url_env: str | None = Field(
+        default=None,
+        description="Environment variable containing the RTSP URL.",
+    )
+    rtsp_url: str | None = Field(
+        default=None,
+        description="RTSP URL for the main stream.",
+    )
+    detect_rtsp_url_env: str | None = Field(
+        default=None,
+        description="Environment variable containing the detect stream RTSP URL.",
+    )
+    detect_rtsp_url: str | None = Field(
+        default=None,
+        description="RTSP URL for the detect stream.",
+    )
+    output_dir: str = Field(
+        default="./recordings",
+        description="Directory to store recordings and logs.",
+    )
+
+    motion: RTSPMotionConfig = Field(default_factory=RTSPMotionConfig)
+    recording: RTSPRecordingConfig = Field(default_factory=RTSPRecordingConfig)
+    stream: RTSPStreamConfig = Field(default_factory=RTSPStreamConfig)
+    reconnect: RTSPReconnectConfig = Field(default_factory=RTSPReconnectConfig)
+    runtime: RTSPRuntimeConfig = Field(default_factory=RTSPRuntimeConfig)
+
+    @model_validator(mode="after")
+    def _require_rtsp_url(self) -> RTSPSourceConfig:
+        if not (self.rtsp_url or self.rtsp_url_env):
+            raise ValueError("rtsp_url_env or rtsp_url required for RTSP source")
+        return self

homesec/plugins/sources/ftp.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 from homesec.interfaces import ClipSource
-from homesec.models.source import FtpSourceConfig
+from homesec.models.source.ftp import FtpSourceConfig
 from homesec.plugins.registry import PluginType, plugin
 
 # Import the actual implementation from sources module

homesec/plugins/sources/local_folder.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 from homesec.interfaces import ClipSource
-from homesec.models.source import LocalFolderSourceConfig
+from homesec.models.source.local_folder import LocalFolderSourceConfig
 from homesec.plugins.registry import PluginType, plugin
 from homesec.sources.local_folder import LocalFolderSource as LocalFolderSourceImpl
 

homesec/plugins/sources/rtsp.py

@@ -5,9 +5,9 @@ from __future__ import annotations
 from typing import TYPE_CHECKING
 
 from homesec.interfaces import ClipSource
-from homesec.models.source import RTSPSourceConfig
+from homesec.models.source.rtsp import RTSPSourceConfig
 from homesec.plugins.registry import PluginType, plugin
-from homesec.sources.rtsp import RTSPSource as RTSPSourceImpl
+from homesec.sources.rtsp.core import RTSPSource as RTSPSourceImpl
 
 if TYPE_CHECKING:
     pass

homesec/sources/__init__.py

@@ -1,10 +1,12 @@
 """Clip source implementations."""
 
-from homesec.models.source import FtpSourceConfig, LocalFolderSourceConfig, RTSPSourceConfig
+from homesec.models.source.ftp import FtpSourceConfig
+from homesec.models.source.local_folder import LocalFolderSourceConfig
+from homesec.models.source.rtsp import RTSPSourceConfig
 from homesec.sources.base import ThreadedClipSource
 from homesec.sources.ftp import FtpSource
 from homesec.sources.local_folder import LocalFolderSource
-from homesec.sources.rtsp import RTSPSource
+from homesec.sources.rtsp.core import RTSPSource
 
 __all__ = [
     "FtpSource",

homesec/sources/ftp.py

@@ -10,7 +10,7 @@ from threading import Thread
 from typing import Any
 
 from homesec.models.clip import Clip
-from homesec.models.source import FtpSourceConfig
+from homesec.models.source.ftp import FtpSourceConfig
 from homesec.sources.base import ThreadedClipSource
 
 logger = logging.getLogger(__name__)

homesec/sources/local_folder.py

@@ -14,7 +14,7 @@ from pathlib import Path
 from anyio import Path as AsyncPath
 
 from homesec.models.clip import Clip
-from homesec.models.source import LocalFolderSourceConfig
+from homesec.models.source.local_folder import LocalFolderSourceConfig
 from homesec.sources.base import AsyncClipSource
 
 logger = logging.getLogger(__name__)

homesec/sources/rtsp/__init__.py

@@ -0,0 +1,5 @@
+"""RTSP source package."""
+
+from homesec.sources.rtsp.core import RTSPSource
+
+__all__ = ["RTSPSource"]

homesec/sources/rtsp/clock.py

@@ -0,0 +1,18 @@
+from __future__ import annotations
+
+import time
+from typing import Protocol
+
+
+class Clock(Protocol):
+    def now(self) -> float: ...
+
+    def sleep(self, seconds: float) -> None: ...
+
+
+class SystemClock:
+    def now(self) -> float:
+        return time.monotonic()
+
+    def sleep(self, seconds: float) -> None:
+        time.sleep(seconds)