Add total_size_mb property to IceflowSearchResult and log size when downloading

Author: trey-stafford

src/nsidc/iceflow/data/fetch.py

@@ -61,6 +61,7 @@ def _download_iceflow_search_result(
 
     logger.info(
         f"Downloading {len(iceflow_search_result.granules)} granules"
+        f" (approx. {iceflow_search_result.total_size_mb} MB)"
         f" to {output_subdir}."
     )
 

src/nsidc/iceflow/data/models.py

@@ -247,5 +247,10 @@ class IceflowSearchResult(pydantic.BaseModel):
     # this that type
     model_config = pydantic.ConfigDict(arbitrary_types_allowed=True)
 
+    @property
+    def total_size_mb(self):
+        granule_sizes_mb = [granule.size() for granule in self.granules]
+        return sum(granule_sizes_mb)
+
 
 IceflowSearchResults = list[IceflowSearchResult]