Add Initial CLI Setup for xDEM

dev-environment.yml

@@ -14,6 +14,7 @@ dependencies:
   - scikit-image=0.*
   - scikit-gstat>=1.0.18,<1.1
   - geoutils=0.1.9
+  - argcomplete
 
   # Development-specific, to mirror manually in setup.cfg [options.extras_require].
   - pip

doc/source/quick_start.md

@@ -97,6 +97,21 @@ import os
 os.remove("dh_error.tif")
 ```
 
+## Command Line Interface (CLI)
+
+The xDEM package can be executed from the command line using the `xdem` command.
+
+### Usage
+
+```bash
+xdem path_ref path_sec [options]
+```
+
+### Options
+
+- `--loglevel`: Set the logging level (default: INFO).
+- `-v`, `--version`: Show the version of the xDEM package.
+
 (quick-gallery)=
 ## More examples
 

environment.yml

@@ -15,6 +15,7 @@ dependencies:
   - scikit-gstat>=1.0.18,<1.1
   - geoutils=0.1.9
   - pip
+  - argcomplete
 
    # To run CI against latest GeoUtils
 #  - pip:

requirements.txt

@@ -13,3 +13,4 @@ scikit-image==0.*
 scikit-gstat>=1.0.18,<1.1
 geoutils==0.1.9
 pip
+argcomplete

setup.cfg

@@ -77,3 +77,7 @@ dev =
     %(doc)s
 all =
     %(dev)s
+
+[options.entry_points]
+console_scripts =
+    xdem = xdem_cli:main

setup.py

@@ -1,4 +1,13 @@
 """This file now only serves for backward-compatibility for routines explicitly calling python setup.py"""
-from setuptools import setup
+from setuptools import find_packages, setup
 
-setup()
+setup(
+    name="xdem",
+    version="0.1",
+    packages=find_packages(),
+    entry_points={
+        "console_scripts": [
+            "xdem = xdem.xdem_cli:main",
+        ],
+    },
+)

tests/test_cli.py

@@ -0,0 +1,30 @@
+"""Function to test the CLI"""
+
+import subprocess
+
+import xdem
+
+
+class TestCLI:
+    # Define paths to the DEM files using xDEM examples
+    ref_dem_path = xdem.examples.get_path("longyearbyen_ref_dem")
+    tba_dem_path = xdem.examples.get_path("longyearbyen_tba_dem")
+
+    def test_xdem_cli(self) -> None:
+        try:
+            # Run the xDEM CLI command with the reference and secondary DEM files
+            result = subprocess.run(
+                ["xdem", self.ref_dem_path, self.tba_dem_path],
+                capture_output=True,
+                text=True,
+            )
+            assert "hello world" in result.stdout
+            assert result.returncode == 0
+
+        except FileNotFoundError as e:
+            # In case 'xdem' is not found
+            raise AssertionError(f"CLI command 'xdem' not found : {e}")
+
+        except Exception as e:
+            # Any other errors during subprocess run
+            raise AssertionError(f"An error occurred while running the CLI: {e}")

xdem/__init__.py

@@ -22,3 +22,10 @@
         "virtualenv) and then install it in-place by running: "
         "pip install -e ."
     )
+
+
+def run(reference_dem: str, dem_to_be_aligned: str, verbose: str) -> None:
+    """
+    Function to compare DEMs
+    """
+    print("hello world")

xdem/xdem_cli.py

@@ -0,0 +1,61 @@
+import argparse
+from argparse import ArgumentParser
+
+import argcomplete
+
+import xdem
+
+
+def get_parser() -> ArgumentParser:
+    """
+    ArgumentParser for xdem
+
+    :return: parser
+    """
+    parser = argparse.ArgumentParser(
+        description="Compare Digital Elevation Models",
+        fromfile_prefix_chars="@",
+    )
+
+    parser.add_argument(
+        "reference_dem",
+        help="path to a reference dem",
+    )
+
+    parser.add_argument(
+        "dem_to_be_aligned",
+        help="path to a second dem",
+    )
+
+    parser.add_argument(
+        "--loglevel",
+        default="INFO",
+        choices=("DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"),
+        help="Logger level (default: INFO. Should be one of (DEBUG, INFO, WARNING, ERROR, CRITICAL)",
+    )
+
+    parser.add_argument(
+        "--version",
+        "-v",
+        action="version",
+        version=f"%(prog)s {xdem.__version__}",
+    )
+
+    return parser
+
+
+def main() -> None:
+    """
+    Call xDEM's main
+    """
+    parser = get_parser()
+    argcomplete.autocomplete(parser)
+    args = parser.parse_args()
+    try:
+        xdem.run(args.reference_dem, args.dem_to_be_aligned, args.loglevel)
+    except Exception as e:
+        print(f"Error: {e}")
+
+
+if __name__ == "__main__":
+    main()