tpeetz/kontor
1
0
Fork 0
Code Issues 6 Pull Requests 2 Actions Packages Projects Releases 2 Wiki Activity

improve download.py

Browse Source
This commit is contained in:
Thomas Peetz
2025-09-08 13:38:47 +02:00
parent 8b1b84b195
commit 55a09d32b3
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
kontor-scripts/download.py
+6 -2
View File
@@ -34,7 +34,7 @@ def download_file(url: str, file_info: dict, download_dir: str = "/data/media",
output = re.sub(' +', ' ', output) output = re.sub(' +', ' ', output)
lines_list = output.splitlines() lines_list = output.splitlines()
file_name = __parse_output__(lines_list) file_name = __parse_output__(lines_list)
if file_name is None: if file_name is None or not file_name.strip():
file_info['review'] = True file_info['review'] = True
file_info['should_download'] = True file_info['should_download'] = True
file_info['file_name'] = None file_info['file_name'] = None
@@ -58,6 +58,8 @@ def __parse_output__(lines_list: list[str]) -> str | None:
start_len = len('[download] Destination: ') start_len = len('[download] Destination: ')
file_len = line_len - start_len file_len = line_len - start_len
file_name = line[-file_len:] file_name = line[-file_len:]
else:
file_name = None
return file_name return file_name
@@ -91,8 +93,10 @@ def update_status(item_id: UUID, file_info: dict):
def rename_file(file_info: dict): def rename_file(file_info: dict):
item_id = file_info['id'] item_id = file_info['id']
file_name = file_info['file_name'] file_name = file_info['file_name']
if file_name is None: if file_name is None or not file_name.strip():
log.info("file_name is not set, rename is not executed") log.info("file_name is not set, rename is not executed")
file_info['review'] = True
file_info['should_download'] = True
return return
file = Path(args.dir, file_name) file = Path(args.dir, file_name)
new_file_path = file.with_name(f"{item_id}{file.suffix}") new_file_path = file.with_name(f"{item_id}{file.suffix}")