Simple web backend added with upload/status/process/download API

This commit is contained in:
2026-07-30 12:24:37 +00:00
parent 277c538f21
commit cd56637d2a
20 changed files with 594 additions and 0 deletions
+1
View File
@@ -0,0 +1 @@
"""Business logic services."""
+47
View File
@@ -0,0 +1,47 @@
"""Temporary file management for web operations."""
import shutil
from pathlib import Path
from backend.config import settings
class FileManager:
@staticmethod
def get_task_dir(task_id: str) -> Path:
return settings.temp_dir / task_id
@staticmethod
def get_input_path(task_id: str) -> Path:
task_dir = FileManager.get_task_dir(task_id)
for f in task_dir.iterdir():
if f.name.startswith("input."):
return f
return None
@staticmethod
def get_output_dir(task_id: str) -> Path:
return FileManager.get_task_dir(task_id) / "output"
@staticmethod
def ensure_output_dir(task_id: str) -> Path:
output_dir = FileManager.get_output_dir(task_id)
output_dir.mkdir(parents=True, exist_ok=True)
return output_dir
@staticmethod
def cleanup_task(task_id: str) -> None:
task_dir = FileManager.get_task_dir(task_id)
if task_dir.exists():
shutil.rmtree(task_dir, ignore_errors=True)
@staticmethod
def get_output_files(task_id: str) -> list:
output_dir = FileManager.get_output_dir(task_id)
if not output_dir.exists():
return []
files = []
for f in output_dir.iterdir():
if f.is_file():
files.append({"filename": f.name, "size": f.stat().st_size})
return files
+86
View File
@@ -0,0 +1,86 @@
"""Splitter service that calls the core audio_splitter logic."""
import os
import sys
import shutil
from pathlib import Path
from typing import List, Dict, Any
from backend.config import settings
from backend.services.task_manager import task_manager
from backend.services.file_manager import FileManager
from backend.models.request import TracklistEntry
from backend.models.response import TaskStatus
def run_split_task(task_id: str, tracklist: List[TracklistEntry], options: Dict[str, Any]) -> None:
try:
task_manager.update_task(task_id, progress=5, message="Initializing...")
input_path = FileManager.get_input_path(task_id)
if not input_path or not input_path.exists():
raise RuntimeError(f"Input file not found for task {task_id}")
output_dir = FileManager.ensure_output_dir(task_id)
tracks = []
for entry in tracklist:
track_dict = {
"ts": entry.ts,
"tn": entry.tn or "",
"an": entry.an or "",
"al": entry.al or "",
"date": entry.date or "",
"ext": entry.ext or ""
}
tracks.append(track_dict)
from types import SimpleNamespace
args = SimpleNamespace(
format=options.get("format", "mp3"),
transcode_to=options.get("transcode_to", None),
drop_video=options.get("drop_video", False),
drop_subs=options.get("drop_subs", False),
number_tracks=options.get("number_tracks", False),
replace_bad_chars=options.get("replace_bad_chars", False),
replacement_char=options.get("replacement_char", "_"),
bad_chars=options.get("bad_chars", r'!@#№$;:%^&?*(){}[]\/<>+=~`\' '),
skip_existing=options.get("skip_existing", False),
output_template=options.get("output_template", "%an-%tn.%ext"),
album=options.get("album", None),
comment=options.get("comment", None),
no_comment=options.get("no_comment", False),
comment_stream=options.get("comment_stream", None),
merge_comments=options.get("merge_comments", False),
comment_separator=options.get("comment_separator", "; "),
delete_original=False,
)
project_root = Path(__file__).parent.parent.parent.parent
if str(project_root) not in sys.path:
sys.path.insert(0, str(project_root))
from audio_splitter.core import split_audio
task_manager.update_task(task_id, progress=10, message="Starting split...")
split_audio(str(input_path), str(output_dir), tracks, args)
output_files = FileManager.get_output_files(task_id)
task_manager.update_task(
task_id,
status=TaskStatus.DONE,
progress=100,
message="Split complete",
tracks=output_files
)
except Exception as e:
task_manager.update_task(
task_id,
status=TaskStatus.ERROR,
error=str(e),
message="Split failed"
)
+64
View File
@@ -0,0 +1,64 @@
"""In-memory task state management."""
from typing import Dict, Optional, List
from datetime import datetime, timedelta
import threading
import time
from backend.config import settings
from backend.models.response import TaskStatus, TrackInfo
class TaskManager:
def __init__(self):
self._tasks: Dict[str, dict] = {}
self._lock = threading.Lock()
def create_task(self, task_id: str, filename: str, file_size: int) -> None:
with self._lock:
self._tasks[task_id] = {
"task_id": task_id,
"filename": filename,
"file_size": file_size,
"status": TaskStatus.PENDING,
"progress": 0,
"message": "Upload complete",
"error": None,
"tracks": [],
"created_at": datetime.now(),
"updated_at": datetime.now()
}
def has_task(self, task_id: str) -> bool:
return task_id in self._tasks
def get_status(self, task_id: str) -> dict:
with self._lock:
return self._tasks.get(task_id, {}).copy()
def update_task(self, task_id: str, **kwargs) -> None:
with self._lock:
if task_id in self._tasks:
self._tasks[task_id].update(kwargs)
self._tasks[task_id]["updated_at"] = datetime.now()
def cleanup_old_tasks(self) -> None:
now = datetime.now()
timeout = timedelta(seconds=settings.cleanup_after_seconds)
with self._lock:
to_delete = []
for task_id, data in self._tasks.items():
if now - data["created_at"] > timeout:
to_delete.append(task_id)
for task_id in to_delete:
self._delete_task_files(task_id)
del self._tasks[task_id]
def _delete_task_files(self, task_id: str) -> None:
import shutil
task_dir = settings.temp_dir / task_id
if task_dir.exists():
shutil.rmtree(task_dir, ignore_errors=True)
task_manager = TaskManager()