From 73a1645bbebebf4e9dc66f394834ac8c90ffa13d Mon Sep 17 00:00:00 2001
From: up-n-atom <adam.jaremko@gmail.com>
Date: Tue, 2 Jan 2024 20:19:46 -0500
Subject: [PATCH] Swap to context fail method on error

---
 xmo/xmo.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/xmo/xmo.py b/xmo/xmo.py
index 9f7e4be..cb31754 100644
--- a/xmo/xmo.py
+++ b/xmo/xmo.py
@@ -44,8 +44,7 @@ async def cli(ctx: click.Context, host: IPv4Address, username: str, password: st
     try:
         await client.login()
     except Exception as e:
-        click.echo(e, err=True)
-        raise click.Abort()
+        ctx.fail(e)
 
 
 @cli.command()
@@ -74,8 +73,7 @@ async def set_value(ctx: click.Context, path: str, value: str) -> None:
     try:
         value = await client.set_value_by_xpath(path, value)
     except Exception as e:
-        click.echo(e, err=True)
-        raise click.Abort()
+        ctx.fail(e)
 
 
 @asynccontextmanager