ENH: add progress bar when creating local indexes

ci/envs/latest.yml

@@ -19,6 +19,7 @@ dependencies:
   - rasterstats
   - rioxarray
   - scikit-learn
+  - tqdm
   # Optional
   - qgis
   # testing

ci/envs/minimal.yml

@@ -19,6 +19,7 @@ dependencies:
   - rasterstats
   - rioxarray
   - scikit-learn
+  - tqdm
   # Optional
   # - qgis
   # testing

cropclassification/util/mosaic_util.py

@@ -9,6 +9,8 @@
 from typing import Any, Optional
 
 import pyproj
+from tqdm import tqdm
+from tqdm.contrib.logging import logging_redirect_tqdm
 
 from . import date_util, openeo_util, raster_index_util, raster_util
 
@@ -254,16 +256,17 @@ def calc_periodic_mosaic(
         )
 
         # Process the mosaic images to be generated locally.
-        for image_local in images_local:
-            # Prepare index output file path
-            raster_index_util.calc_index(
-                image_local["base_image_path"],
-                image_local["path"],
-                index=image_local["index_type"],
-                pixel_type=image_local["pixel_type"],
-                process_options=image_local["process_options"],
-                force=force,
-            )
+        with logging_redirect_tqdm():
+            for image_local in tqdm(images_local):
+                # Prepare index output file path
+                raster_index_util.calc_index(
+                    image_local["base_image_path"],
+                    image_local["path"],
+                    index=image_local["index_type"],
+                    pixel_type=image_local["pixel_type"],
+                    process_options=image_local["process_options"],
+                    force=force,
+                )
 
     return periodic_mosaic_params
 

environment-dev.yml

@@ -19,6 +19,7 @@ dependencies:
   - rasterstats
   - rioxarray
   - scikit-learn
+  - tqdm
   # Optional
   - qgis
   # testing

setup.py

@@ -31,6 +31,7 @@
         "tensorflow",
         "numpy<2",
         "exactextract",
+        "tqdm",
     ],
     entry_points="""
         [console_scripts]