Mirrors the music flow: a finished download batch → refresh the media server → (after
it indexes) pull the new media into the DB — so a downloaded movie/episode shows as
owned without waiting for the 6h scheduled scan.
Two event-based system automations (owned_by='video'):
- 'Auto-Scan Video After Downloads' (video_batch_complete → video_scan_server)
- 'Auto-Update Video Database After Scan' (video_library_scan_completed → video_update_database)
Pieces:
- core/video/download_events.py: a callback registry (core/video can't import the
engine — isolation). The monitor publishes batch-complete; web_server bridges it to
automation_engine.emit('video_batch_complete', …), like music's web_scan_manager.
- download_monitor: fires the batch-complete event once, when the last in-flight
download finishes (none queued/downloading/searching left).
- video_scan_server handler (stage 1): refresh server, wait a debounce for indexing,
then emit 'video_library_scan_completed' (mirrors music's time-based completion).
- video_update_database handler (stage 2): incremental read (newest-first, stop after
25 consecutive known — same as music).
- blocks: 2 video triggers + 2 video actions (scope='video'); registration; seeds.
kettui: seam tests for both handlers (refresh/wait/emit, incremental read, error paths),
the event registry (idempotent register, isolated failures), and the monitor (fires once
on last completion, never while work remains). 298 automation tests + isolation green.
204 lines
8 KiB
Python
204 lines
8 KiB
Python
"""One-stop registration of every extracted automation handler.
|
|
|
|
``web_server`` builds the deps once at startup and calls
|
|
:func:`register_all` here. Each new handler module gets one line in
|
|
this file when it lands.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from core.automation.deps import AutomationDeps
|
|
from core.automation.handlers.process_wishlist import auto_process_wishlist
|
|
from core.automation.handlers.scan_watchlist import auto_scan_watchlist
|
|
from core.automation.handlers.scan_library import auto_scan_library
|
|
from core.automation.handlers.refresh_mirrored import auto_refresh_mirrored
|
|
from core.automation.handlers.sync_playlist import auto_sync_playlist
|
|
from core.automation.handlers.discover_playlist import auto_discover_playlist
|
|
from core.automation.handlers.playlist_pipeline import auto_playlist_pipeline
|
|
from core.automation.handlers.personalized_pipeline import auto_personalized_pipeline
|
|
from core.automation.handlers.database_update import (
|
|
auto_start_database_update, auto_deep_scan_library,
|
|
)
|
|
from core.automation.handlers.duplicate_cleaner import auto_run_duplicate_cleaner
|
|
from core.automation.handlers.quality_scanner import auto_start_quality_scan
|
|
from core.automation.handlers.maintenance import (
|
|
auto_clear_quarantine,
|
|
auto_cleanup_wishlist,
|
|
auto_update_discovery_pool,
|
|
auto_backup_database,
|
|
auto_refresh_beatport_cache,
|
|
)
|
|
from core.automation.handlers.download_cleanup import (
|
|
auto_clean_search_history,
|
|
auto_clean_completed_downloads,
|
|
auto_full_cleanup,
|
|
)
|
|
from core.automation.handlers.run_script import auto_run_script
|
|
from core.automation.handlers.search_and_download import auto_search_and_download
|
|
from core.automation.handlers.video_scan_library import (
|
|
auto_video_scan_library, auto_video_scan_server, auto_video_update_database,
|
|
)
|
|
from core.automation.handlers.progress_callbacks import (
|
|
progress_init,
|
|
progress_finish,
|
|
record_history,
|
|
register_library_scan_completed_emitter,
|
|
)
|
|
|
|
|
|
def register_all(deps: AutomationDeps) -> None:
|
|
"""Wire every extracted handler to the engine.
|
|
|
|
Each ``register_action_handler`` call binds the action name (the
|
|
string the trigger uses to look up its action) to a thin lambda
|
|
that injects ``deps`` and forwards the engine-supplied config.
|
|
Guards stay alongside their handler so duplicate-run prevention
|
|
behaves identically to the pre-extraction code.
|
|
"""
|
|
engine = deps.engine
|
|
|
|
# Self-guards prevent duplicate runs of the SAME operation, but
|
|
# different operations can run concurrently — wishlist downloads
|
|
# use bandwidth, watchlist scans use API calls, library scans use
|
|
# media-server CPU. Different resources, no contention.
|
|
engine.register_action_handler(
|
|
'process_wishlist',
|
|
lambda config: auto_process_wishlist(config, deps),
|
|
guard_fn=deps.is_wishlist_actually_processing,
|
|
)
|
|
engine.register_action_handler(
|
|
'scan_watchlist',
|
|
lambda config: auto_scan_watchlist(config, deps),
|
|
guard_fn=deps.is_watchlist_actually_scanning,
|
|
)
|
|
engine.register_action_handler(
|
|
'scan_library',
|
|
lambda config: auto_scan_library(config, deps),
|
|
deps.state.is_scan_library_active,
|
|
)
|
|
|
|
# Playlist lifecycle handlers. The pipeline composes refresh +
|
|
# sync + discover (it imports them directly), so all four ship
|
|
# together. The pipeline guard prevents an in-flight pipeline
|
|
# from being re-triggered mid-run.
|
|
engine.register_action_handler(
|
|
'refresh_mirrored',
|
|
lambda config: auto_refresh_mirrored(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'sync_playlist',
|
|
lambda config: auto_sync_playlist(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'discover_playlist',
|
|
lambda config: auto_discover_playlist(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'playlist_pipeline',
|
|
lambda config: auto_playlist_pipeline(config, deps),
|
|
deps.state.is_pipeline_running,
|
|
)
|
|
# Personalized pipeline shares the pipeline_running flag with the
|
|
# mirrored pipeline so the two can't overlap (single sync queue,
|
|
# single wishlist worker).
|
|
engine.register_action_handler(
|
|
'personalized_pipeline',
|
|
lambda config: auto_personalized_pipeline(config, deps),
|
|
deps.state.is_pipeline_running,
|
|
)
|
|
|
|
# Database update + deep scan share the db_update_state guard —
|
|
# only one operation can mutate that state at a time.
|
|
engine.register_action_handler(
|
|
'start_database_update',
|
|
lambda config: auto_start_database_update(config, deps),
|
|
lambda: deps.get_db_update_state().get('status') == 'running',
|
|
)
|
|
engine.register_action_handler(
|
|
'deep_scan_library',
|
|
lambda config: auto_deep_scan_library(config, deps),
|
|
lambda: deps.get_db_update_state().get('status') == 'running',
|
|
)
|
|
engine.register_action_handler(
|
|
'run_duplicate_cleaner',
|
|
lambda config: auto_run_duplicate_cleaner(config, deps),
|
|
lambda: deps.get_duplicate_cleaner_state().get('status') == 'running',
|
|
)
|
|
engine.register_action_handler(
|
|
'clear_quarantine',
|
|
lambda config: auto_clear_quarantine(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'cleanup_wishlist',
|
|
lambda config: auto_cleanup_wishlist(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'update_discovery_pool',
|
|
lambda config: auto_update_discovery_pool(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'start_quality_scan',
|
|
lambda config: auto_start_quality_scan(config, deps),
|
|
lambda: False, # repair worker dedupes Run-Now requests itself
|
|
)
|
|
engine.register_action_handler(
|
|
'backup_database',
|
|
lambda config: auto_backup_database(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'refresh_beatport_cache',
|
|
lambda config: auto_refresh_beatport_cache(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'clean_search_history',
|
|
lambda config: auto_clean_search_history(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'clean_completed_downloads',
|
|
lambda config: auto_clean_completed_downloads(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'full_cleanup',
|
|
lambda config: auto_full_cleanup(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'run_script',
|
|
lambda config: auto_run_script(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'search_and_download',
|
|
lambda config: auto_search_and_download(config, deps),
|
|
)
|
|
|
|
# Video side (isolated app, shared engine). The video twins are tagged
|
|
# owned_by='video' on their automation rows so they never surface on the
|
|
# music automations page; the handlers bridge into core.video.
|
|
engine.register_action_handler(
|
|
'video_scan_library',
|
|
lambda config: auto_video_scan_library(config, deps),
|
|
)
|
|
# Post-download chain: scan the server, then (on the scan-done event) update the DB.
|
|
engine.register_action_handler(
|
|
'video_scan_server',
|
|
lambda config: auto_video_scan_server(config, deps),
|
|
)
|
|
engine.register_action_handler(
|
|
'video_update_database',
|
|
lambda config: auto_video_update_database(config, deps),
|
|
)
|
|
|
|
# Progress + history callbacks: the engine invokes these around
|
|
# each handler run. Lift the closures from
|
|
# `web_server._register_automation_handlers` into thin lambdas
|
|
# that delegate into the extracted top-level functions.
|
|
engine.register_progress_callbacks(
|
|
lambda aid, name, action_type: progress_init(aid, name, action_type, deps),
|
|
lambda aid, result: progress_finish(aid, result, deps),
|
|
deps.update_progress,
|
|
lambda aid, result: record_history(aid, result, deps),
|
|
)
|
|
|
|
# `library_scan_completed` event: when the media-server scan
|
|
# manager finishes a scan, emit the event so any automation can
|
|
# trigger off it. No-op when no scan manager is configured.
|
|
register_library_scan_completed_emitter(deps)
|