Add overwrite option

CHANGELOG.md

@@ -19,7 +19,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 
 ### Changed
 
-- Use `Config` instead of standalone arguments ([#45](https://github.com/stac-utils/stac-asset/pull/45))
+- Use `Config` instead of standalone arguments ([#45](https://github.com/stac-utils/stac-asset/pull/45), [#67](https://github.com/stac-utils/stac-asset/pull/67))
 - s/CantIncludeAndExclude/CannotIncludeAndExclude/ ([#45](https://github.com/stac-utils/stac-asset/pull/45))
 - Re-use the same client for an entire item collection ([#59](https://github.com/stac-utils/stac-asset/pull/59))
 

src/stac_asset/_cli.py

@@ -96,6 +96,13 @@ def cli() -> None:
     is_flag=True,
     show_default=True,
 )
+@click.option(
+    "--overwrite",
+    help="Overwrite existing files if they exist on the filesystem",
+    default=False,
+    is_flag=True,
+    show_default=True,
+)
 # TODO add option to disable content type checking
 def download(
     href: Optional[str],
@@ -109,6 +116,7 @@ def download(
     s3_retry_mode: str,
     s3_max_attempts: int,
     warn: bool,
+    overwrite: bool,
 ) -> None:
     """Download STAC assets from an item or item collection.
 
@@ -142,6 +150,7 @@ def download(
             s3_retry_mode,
             s3_max_attempts,
             warn,
+            overwrite,
         )
     )
 
@@ -158,16 +167,21 @@ async def download_async(
     s3_retry_mode: str,
     s3_max_attempts: int,
     warn: bool,
+    overwrite: bool,
 ) -> None:
+    if warn:
+        download_strategy = DownloadStrategy.DELETE
+    else:
+        download_strategy = DownloadStrategy.ERROR
     config = Config(
         alternate_assets=alternate_assets,
         include=include,
         exclude=exclude,
         s3_requester_pays=s3_requester_pays,
         s3_retry_mode=s3_retry_mode,
         s3_max_attempts=s3_max_attempts,
-        # TODO allow configuring of download strategy
-        download_strategy=DownloadStrategy.DELETE,
+        download_strategy=download_strategy,
+        overwrite=overwrite,
     )
 
     if href is None or href == "-":

src/stac_asset/_download.py

@@ -45,16 +45,18 @@ class Download:
     asset: Asset
     path: Path
     client: Client
+    overwrite: bool
 
     async def download(
         self, make_directory: bool, clean: bool, queue: Optional[AnyQueue]
     ) -> Union[Download, WrappedError]:
-        try:
-            self.asset = await self.client.download_asset(
-                self.key, self.asset, self.path, make_directory, clean, queue
-            )
-        except Exception as error:
-            return WrappedError(self, error)
+        if not self.overwrite and not os.path.exists(self.path):
+            try:
+                await self.client.download_asset(
+                    self.key, self.asset, self.path, make_directory, clean, queue
+                )
+            except Exception as error:
+                return WrappedError(self, error)
         self.asset.href = str(self.path)
         return self
 
@@ -122,6 +124,7 @@ async def add(
                     asset=asset,
                     path=root / asset_file_name,
                     client=client,
+                    overwrite=self.config.overwrite,
                 )
             )
 

src/stac_asset/config.py

@@ -46,6 +46,9 @@ class Config:
     clean: bool = True
     """If true, clean up the downloaded file if it errors."""
 
+    overwrite: bool = False
+    """Download files even if they already exist locally."""
+
     earthdata_token: Optional[str] = None
     """A token for logging in to Earthdata."""