From 967dbd8d4fbcf2217fd008dba60468f53ac62877 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alo=C3=AFs=20Micard?= Date: Fri, 15 Nov 2024 00:15:25 +0100 Subject: [PATCH] Allow to add process limit --- exif_database/__main__.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/exif_database/__main__.py b/exif_database/__main__.py index 42c89ac..53fb14f 100644 --- a/exif_database/__main__.py +++ b/exif_database/__main__.py @@ -45,6 +45,10 @@ if __name__ == '__main__': # Load saved pictures cache saved_pictures = _load_pictures_cache() + # Keep track of processed pictures + processed_pictures = 0 + max_processed_pictures = None if len(sys.argv) < 3 else int(sys.argv[2]) + for file in Path(sys.argv[1]).rglob("*.ARW"): filename = os.fsdecode(file) @@ -63,6 +67,11 @@ if __name__ == '__main__': saved_pictures[picture_metadata['path']] = True + processed_pictures = processed_pictures + 1 + + if max_processed_pictures is not None and processed_pictures >= max_processed_pictures: + break + # Insert into MongoDB if len(metadata_pictures) > 0: collection.insert_many(metadata_pictures)