azalea/MTfin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[O] File name whitelist

Browse Source
This commit is contained in:
Azalea Gui
2026-03-10 19:17:49 -04:00
parent c39089fce4
commit 5755bfb650
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
workflow.py
+5 -1
View File
@@ -27,11 +27,15 @@ from utils_ai import (
) )
VALID_MEDIA_EXTS = {".mkv", ".mp4", ".avi", ".ts", ".m2ts", ".webm", ".srt", ".ass", ".ssa", ".sub", ".idx"}
def format_file_tree(file_tree: list) -> str: def format_file_tree(file_tree: list) -> str:
"""Helper to convert the qB file tree output into simple relative paths for the LLM prompt""" """Helper to convert the qB file tree output into simple relative paths for the LLM prompt"""
lines = [] lines = []
for f in file_tree: for f in file_tree:
lines.append(f.get("name", "")) name = f.get("name", "")
if Path(name).suffix.lower() in VALID_MEDIA_EXTS:
lines.append(name)
return "\n".join(lines) return "\n".join(lines)
def sanitize_filename(name: str) -> str: def sanitize_filename(name: str) -> str: