Limit number of imported images by default

google_photos/constants.py

@@ -17,4 +17,5 @@ ACTION_COMPLETED = "action_completed"
 SUCCESS = "success"
 FAIL = "fail"
 
-AUTH_TIMEOUT = 120
\ No newline at end of file
+AUTH_TIMEOUT = 120
+IMPORT_LIMIT = os.environ.get("IMPORT_LIMIT", 100)
\ No newline at end of file

google_photos/plugin.py

@@ -16,7 +16,7 @@ from google_photos.data.schema import (
     AuthenticationFlow,
 )
 from pymemri.plugin.pluginbase import PluginBase
-from .constants import SERVICE_NAME, CLIENT_ID, CLIENT_SECRET, PROJECT_ID
+from .constants import SERVICE_NAME, CLIENT_ID, CLIENT_SECRET, PROJECT_ID, IMPORT_LIMIT
 from .constants import IMAGE_FORMATS, VIDEO_FORMATS, THUMBNAIL_DIM
 from .constants import AWAIT_USER_ACTION, ACTION_COMPLETED, SUCCESS, FAIL, AUTH_TIMEOUT
 
@@ -323,6 +323,8 @@ class GooglePhotos(PluginBase):
         if "mediaItems" in items:
             self.number_of_images = len(items['mediaItems'])
             logger.info(f"Service has {self.number_of_images} images")
+            if IMPORT_LIMIT:
+                items["mediaItems"] = items["mediaItems"][:int(IMPORT_LIMIT)]
             for item in items["mediaItems"]:
                 executor.submit(self.import_media_file, item)
             executor.shutdown()