fietsupdater: use Context.fail to abort

This makes sure that the exit code is set properly

1 files changed, 11 insertions, 8 deletions

diff --git a/fietsboek/updater/cli.py b/fietsboek/updater/cli.py
index 6a59100..45655d4 100644
--- a/fietsboek/updater/cli.py
+++ b/fietsboek/updater/cli.py
@@ -54,7 +54,8 @@ def cli():
     help="Skip the safety question and just run the update",
 )
 @click.argument("version", required=False)
-def update(config, version, force):
+@click.pass_context
+def update(ctx, config, version, force):
     """Run the update process.
 
     Make sure to consult the documentation and ensure that you have a backup
@@ -66,16 +67,17 @@ def update(config, version, force):
     updater = Updater(config)
     updater.load()
     if version and not updater.exists(version):
-        click.secho("Revision not found", fg="red")
-        return
+        ctx.fail(f"Version {version!r} not found")
+
     version_str = ", ".join(updater.current_versions())
     click.echo(f"Current version: {version_str}")
     if not version:
         heads = updater.heads()
         if len(heads) != 1:
-            click.secho("Ambiguous heads, please specify the version to update to", fg="red")
-            return
+            ctx.fail("Ambiguous heads, please specify the version to update to")
+
         version = heads[0]
+
     click.echo(f"Selected version: {version}")
     if not force:
         user_confirm()
@@ -92,7 +94,8 @@ def update(config, version, force):
     help="Skip the safety question and just run the downgrade",
 )
 @click.argument("version")
-def downgrade(config, version, force):
+@click.pass_context
+def downgrade(ctx, config, version, force):
     """Run the downgrade process.
 
     Make sure to consult the documentation and ensure that you have a backup
@@ -103,8 +106,8 @@ def downgrade(config, version, force):
     updater = Updater(config)
     updater.load()
     if version and not updater.exists(version):
-        click.secho("Revision not found", fg="red")
-        return
+        ctx.fail(f"Version {version!r} not found")
+
     version_str = ", ".join(updater.current_versions())
     click.echo(f"Current version: {version_str}")
     click.echo(f"Downgrade to version {version}")