Add output directory argument for write_results_to_file function

Author: f-PLT

geospatial_tools/planetary_computer/sentinel_2.py

@@ -193,12 +193,13 @@ def select_best_products_per_feature(self) -> GeoDataFrame:
         self.vector_features_with_products = spatial_join_results
         return self.vector_features_with_products
 
-    def to_file(self):
+    def to_file(self, output_dir: Union[str, pathlib.Path]) -> None:
         write_results_to_file(
             cloud_cover=self.max_cloud_cover,
             successful_results=self.successful_results,
             incomplete_results=self.incomplete_results,
             error_results=self.error_results,
+            output_dir=output_dir,
         )
 
 
@@ -322,25 +323,26 @@ def write_results_to_file(
     successful_results: dict,
     incomplete_results: Optional[list] = None,
     error_results: Optional[list] = None,
+    output_dir: Union[str, pathlib.Path] = DATA_DIR,
     logger: logging.Logger = LOGGER,
 ) -> dict:
-    tile_filename = DATA_DIR / f"data_lt{cloud_cover}cc.json"
+    tile_filename = output_dir / f"data_lt{cloud_cover}cc.json"
     with open(tile_filename, "w", encoding="utf-8") as json_file:
         json.dump(successful_results, json_file, indent=4)
     logger.info(f"Results have been written to {tile_filename}")
 
     incomplete_filename = "None"
     if incomplete_results:
         incomplete_dict = {"incomplete": incomplete_results}
-        incomplete_filename = DATA_DIR / f"incomplete_lt{cloud_cover}cc.json"
+        incomplete_filename = output_dir / f"incomplete_lt{cloud_cover}cc.json"
         with open(incomplete_filename, "w", encoding="utf-8") as json_file:
             json.dump(incomplete_dict, json_file, indent=4)
         logger.info(f"Incomplete results have been written to {incomplete_filename}")
 
     error_filename = "None"
     if error_results:
         error_dict = {"errors": error_results}
-        error_filename = DATA_DIR / f"errors_lt{cloud_cover}cc.json"
+        error_filename = output_dir / f"errors_lt{cloud_cover}cc.json"
         with open(error_filename, "w", encoding="utf-8") as json_file:
             json.dump(error_dict, json_file, indent=4)
         logger.info(f"Errors results have been written to {error_filename}")

scripts/sentinel_2_search_and_process/product_search.py

@@ -171,7 +171,7 @@ def product_search(
 
     best_products_client.create_date_ranges(START_YEAR, END_YEAR, START_MONTH, END_MONTH)
     best_products_client.find_best_complete_products()
-    best_products_client.to_file()
+    best_products_client.to_file(output_dir=S2_SCRIPT_DIR)
 
     # Selecting the best products for each vector tile
     #