From 0a53e68b54be04eeef9d314b6c0baa70833c7657 Mon Sep 17 00:00:00 2001 From: Thomas Peetz Date: Tue, 9 Dec 2025 13:47:48 +0100 Subject: [PATCH] add limit for downloads --- kontor-scripts/download.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/kontor-scripts/download.py b/kontor-scripts/download.py index 4c1a44c..0e101e2 100644 --- a/kontor-scripts/download.py +++ b/kontor-scripts/download.py @@ -17,6 +17,7 @@ parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter) parser.add_argument('--verbose', '-v', action='count', default=0) parser.add_argument('--config', '-c', default='kontor-docker') parser.add_argument('--dir', '-d', default='/data/media') +parser.add_argument('--limit', '-l', type=int, help='maximum number of links to check') parser.add_argument('--tool', '-t', default='yt-dlp') parser.add_argument('--dry-run', '-m', action='store_true') args = parser.parse_args() @@ -111,8 +112,13 @@ if __name__ == '__main__': response = requests.get("http://127.0.0.1:8800/api/media/files?download=true") log.info(f"Status: {response.status_code}") data = response.json() - log.info(f"data: {len(data)}") - for item in data: + entries_count = len(data) + log.info(f"data: {entries_count}") + mediafile_index = 1 + log.debug(f"data: {len(data)}") + missing_actors = {} + if args.limit: + 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}") @@ -129,5 +135,9 @@ if __name__ == '__main__': rename_file(item) log.info(f'{item}') update_status(file_id, item) + log.warning(f"processed {mediafile_index}/{entries_count}") + if args.limit and args.limit <= mediafile_index: + break + mediafile_index += 1 log.info('kontor.download finished')