9f687c061a
Phase 1 of the real grab→transfer pipeline:
- video.db video_downloads table (kind/title/release/source/username/filename/size/
target_dir/dest_path/status/progress/error/timestamps) + CRUD (add/list/get_active/
update/clear_finished). Schema v14.
- core/video/slskd_download.py: start_download (POST /transfers/downloads/{user}),
list_downloads (GET → flattened), + pure classify_state (completed/failed/active),
progress_pct, find_transfer.
- core/video/download_pipeline.py: pure find_completed_file (locate by basename in the
download dir), dest_path_for, target_dir_for (kind → movies/tv/youtube library).
All filesystem/HTTP is injected or thin glue; 7 tests, isolation guard + ruff clean.
Monitor + grab endpoint + Downloads page next.
85 lines
3.7 KiB
Python
85 lines
3.7 KiB
Python
"""Video download pipeline — the pure seams: slskd state classification + flatten,
|
|
file location + destination resolution, and the video.db downloads CRUD. Isolated."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from core.video.download_pipeline import (
|
|
basename_of,
|
|
dest_path_for,
|
|
find_completed_file,
|
|
target_dir_for,
|
|
)
|
|
from core.video.slskd_download import (
|
|
classify_state,
|
|
find_transfer,
|
|
flatten_downloads,
|
|
progress_pct,
|
|
)
|
|
|
|
|
|
def test_classify_state():
|
|
assert classify_state("Completed, Succeeded") == "completed"
|
|
assert classify_state("InProgress") == "active"
|
|
assert classify_state("Queued, Remotely") == "active"
|
|
assert classify_state("Completed, Errored") == "failed"
|
|
assert classify_state("Completed, Cancelled") == "failed"
|
|
assert classify_state("Completed, TimedOut") == "failed"
|
|
assert classify_state("") == "active"
|
|
|
|
|
|
def test_flatten_and_find_transfer():
|
|
data = [{"username": "neo", "directories": [
|
|
{"files": [{"filename": r"@@a\Movie\movie.mkv", "id": "t1", "state": "InProgress",
|
|
"size": 100, "bytesTransferred": 25}]}]}]
|
|
flat = flatten_downloads(data)
|
|
assert len(flat) == 1 and flat[0]["username"] == "neo" and flat[0]["id"] == "t1"
|
|
assert progress_pct(flat[0]) == 25.0
|
|
assert find_transfer(flat, "neo", r"@@a\Movie\movie.mkv")["id"] == "t1"
|
|
assert find_transfer(flat, "neo", "other") == {}
|
|
assert flatten_downloads(None) == []
|
|
|
|
|
|
def test_progress_is_100_when_completed():
|
|
assert progress_pct({"state": "Completed, Succeeded", "size": 100, "transferred": 0}) == 100.0
|
|
|
|
|
|
def test_basename_handles_both_separators():
|
|
assert basename_of(r"@@x\The.Wire.S02\ep.mkv") == "ep.mkv"
|
|
assert basename_of("a/b/c/movie.mp4") == "movie.mp4"
|
|
assert basename_of("") == ""
|
|
|
|
|
|
def test_find_completed_file_by_basename():
|
|
files = ["/dl/SomeFolder/movie.mkv", "/dl/Other/nope.mkv"]
|
|
assert find_completed_file("/dl", r"@@u\Remote\movie.mkv", lambda d: files) == "/dl/SomeFolder/movie.mkv"
|
|
assert find_completed_file("/dl", "missing.mkv", lambda d: files) is None
|
|
|
|
|
|
def test_dest_and_target_resolution():
|
|
assert dest_path_for("/media/movies", "/dl/x/movie.mkv") == "/media/movies/movie.mkv"
|
|
paths = {"movies_path": "/m", "tv_path": "/t", "youtube_path": "/y"}
|
|
assert target_dir_for("movie", paths) == "/m"
|
|
assert target_dir_for("show", paths) == "/t"
|
|
assert target_dir_for("season", paths) == "/t"
|
|
assert target_dir_for("youtube", paths) == "/y"
|
|
assert target_dir_for("weird", paths) == ""
|
|
|
|
|
|
# ── DB CRUD ───────────────────────────────────────────────────────────────────
|
|
def test_video_downloads_crud(tmp_path):
|
|
from database.video_database import VideoDatabase
|
|
db = VideoDatabase(database_path=str(tmp_path / "video_library.db"))
|
|
dl_id = db.add_video_download({
|
|
"kind": "movie", "title": "The Matrix", "release_title": "The.Matrix.1999.1080p",
|
|
"source": "soulseek", "username": "neo", "filename": r"@@a\x\m.mkv",
|
|
"size_bytes": 8_000_000_000, "target_dir": "/media/movies", "status": "downloading",
|
|
})
|
|
assert dl_id > 0
|
|
active = db.get_active_video_downloads()
|
|
assert len(active) == 1 and active[0]["title"] == "The Matrix"
|
|
db.update_video_download(dl_id, status="completed", progress=100, dest_path="/media/movies/m.mkv")
|
|
assert db.get_active_video_downloads() == []
|
|
listed = db.list_video_downloads()
|
|
assert listed[0]["status"] == "completed" and listed[0]["dest_path"] == "/media/movies/m.mkv"
|
|
assert db.clear_finished_video_downloads() == 1
|
|
assert db.list_video_downloads() == []
|