improve download.py
This commit is contained in:
@@ -34,7 +34,7 @@ def download_file(url: str, file_info: dict, download_dir: str = "/data/media",
|
||||
output = re.sub(' +', ' ', output)
|
||||
lines_list = output.splitlines()
|
||||
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['should_download'] = True
|
||||
file_info['file_name'] = None
|
||||
@@ -58,6 +58,8 @@ def __parse_output__(lines_list: list[str]) -> str | None:
|
||||
start_len = len('[download] Destination: ')
|
||||
file_len = line_len - start_len
|
||||
file_name = line[-file_len:]
|
||||
else:
|
||||
file_name = None
|
||||
return file_name
|
||||
|
||||
|
||||
@@ -91,8 +93,10 @@ def update_status(item_id: UUID, file_info: dict):
|
||||
def rename_file(file_info: dict):
|
||||
item_id = file_info['id']
|
||||
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")
|
||||
file_info['review'] = True
|
||||
file_info['should_download'] = True
|
||||
return
|
||||
file = Path(args.dir, file_name)
|
||||
new_file_path = file.with_name(f"{item_id}{file.suffix}")
|
||||
|
||||
Reference in New Issue
Block a user