ENH: default location to store STLs to is users home

fgebhart · Feb 6, 2022 · 0d91134 · 0d91134
1 parent f0d05d6
commit 0d91134
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/mapa/__init__.py b/mapa/__init__.py
@@ -1,4 +1,5 @@
 from pathlib import Path
+from typing import Union
 
 import click
 import numpy as np
@@ -92,15 +93,15 @@ def convert_tif_to_stl(
     input_file: str,
     as_ascii: bool,
     model_size: int,
-    output_file: str,
+    output_file: Union[str, None],
     max_res: bool,
     z_offset: float,
     z_scale: float,
     make_square: bool,
 ) -> Path:
     _verify_input_is_valid(input_file)
     if output_file is None:
-        output_file = Path.cwd() / str(Path(input_file).name).replace(".tiff", ".stl").replace(".tif", ".stl")
+        output_file = Path.home() / str(Path(input_file).name).replace(".tiff", ".stl").replace(".tif", ".stl")
     _verify_output_is_valid(output_file)
 
     tiff = rio.open(input_file)

diff --git a/mapa/cli.py b/mapa/cli.py
@@ -1,5 +1,6 @@
 import subprocess
 from pathlib import Path
+from typing import Union
 
 import click
 
@@ -51,7 +52,7 @@
 @click.version_option()
 def dem2stl(
     input: str,
-    output: str,
+    output: Union[str, None],
     z_offset: float,
     z_scale: float,
     max_res: bool = False,