kontor-scripts/download.py

@@ -35,6 +35,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)
+        log.info(f"found file: {file_name}")
         if file_name is None or not file_name.strip():
             file_info['review'] = True
             file_info['should_download'] = True
@@ -42,6 +43,7 @@ def download_file(url: str, file_info: dict, download_dir: str = "/data/media",
         else:
             download_file_name = Path(download_dir, file_name)
             file_info['should_download'] = False
+            file_info['review'] = False
             file_info['file_name'] = download_file_name.name
             file_info['cloud_link'] = str(download_file_name.absolute())
         file_info['last_modified_date'] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -51,14 +53,18 @@ def download_file(url: str, file_info: dict, download_dir: str = "/data/media",
 def __parse_output__(lines_list: list[str]) -> str | None:
     file_name = None
     for line in lines_list:
-        if 'has already been downloaded' in line:
-            end_len = len(' has already been downloaded')
+        log.debug(f"parse line: {line}")
+        if "has already been downloaded" in line:
+            end_len = len(" has already been downloaded")
             file_name = line[11:-end_len]
+            log.info(f"file_name: {file_name}")
+            break
         if 'Destination' in line:
             line_len = len(line)
             start_len = len('[download] Destination: ')
             file_len = line_len - start_len
             file_name = line[-file_len:]
+            break
         else:
             file_name = None
     return file_name
@@ -118,7 +124,8 @@ if __name__ == '__main__':
     log.debug(f"data: {len(data)}")
     missing_actors = {}
     if args.limit:
-        log.warning(f"check the first {args.limit} links")    for item in data:
+        log.warning(f"check the first {args.limit} links")
+    for item in data:
         link = item['url']
         file_id = item['id']
         log.info(f"{file_id} - {link}")