Adding alias functionality to the alerts command

Author: armiiller

commands/alerts.py

@@ -92,50 +92,130 @@ def delete_alert_cmd(ctx, alert_id, force):
         handle_api_error(e, action="deleting alert")
 
 @alerts.command(name="acknowledge")
-@click.argument("alert_id", required=True)
+@click.argument("alert_id", required=False)  # Make alert_id optional
+@click.option("--alias", help="Alias for the alert")
 @click.pass_context
-def acknowledge_alert_cmd(ctx, alert_id):
+def acknowledge_alert_cmd(ctx, alert_id, alias):
     """Acknowledge an alert in PagerTree."""
     try:
         client = ctx.obj  # Get PagerTreeClient from context
+
+        # Ensure at least one of alert_id or alias is provided
+        if not alert_id and not alias:
+            click.echo("Error: Either alert_id or alias must be provided.")
+            return
+
+        # If alias is provided, resolve it to alert_id
+        if alias:
+            alias_result = client.list_alerts(search=alias, limit=1, offset=0)
+            if alias_result["total"] == 0:
+                click.echo(f"No alert found with alias: {alias}")
+                return
+            alert_id = alias_result["data"][0]["id"]
+
+        # If alert_id is still not set, raise an error
+        if not alert_id:
+            click.echo("Error: Could not resolve alert_id.")
+            return
+
         result = client.acknowledge_alert(alert_id)
         click.echo(f"Alert acknowledged successfully: {result.get('id')}")
     except Exception as e:
         handle_api_error(e, action="acknowledging alert")
 
 @alerts.command(name="reject")
-@click.argument("alert_id", required=True)
+@click.argument("alert_id", required=False)  # Make alert_id optional
+@click.option("--alias", help="Alias for the alert")
 @click.pass_context
-def reject_alert_cmd(ctx, alert_id):
+def reject_alert_cmd(ctx, alert_id, alias):
     """Reject an alert in PagerTree."""
     try:
         client = ctx.obj  # Get PagerTreeClient from context
+
+        # Ensure at least one of alert_id or alias is provided
+        if not alert_id and not alias:
+            click.echo("Error: Either alert_id or alias must be provided.")
+            return
+
+        # If alias is provided, resolve it to alert_id
+        if alias:
+            alias_result = client.list_alerts(search=alias, limit=1, offset=0, status="open")
+            if alias_result["total"] == 0:
+                click.echo(f"No alert found with alias: {alias}")
+                return
+            alert_id = alias_result["data"][0]["id"]
+
+        # If alert_id is still not set, raise an error
+        if not alert_id:
+            click.echo("Error: Could not resolve alert_id.")
+            return
+
         result = client.reject_alert(alert_id)
         click.echo(f"Alert rejected successfully: {result.get('id')}")
     except Exception as e:
         handle_api_error(e, action="rejecting alert")
 
 @alerts.command(name="resolve")
-@click.argument("alert_id", required=True)
+@click.argument("alert_id", required=False)  # Make alert_id optional
+@click.option("--alias", help="Alias for the alert")
 @click.pass_context
-def resolve_alert_cmd(ctx, alert_id):
+def resolve_alert_cmd(ctx, alert_id, alias):
     """Resolve an alert in PagerTree."""
     try:
         client = ctx.obj  # Get PagerTreeClient from context
+
+        # Ensure at least one of alert_id or alias is provided
+        if not alert_id and not alias:
+            click.echo("Error: Either alert_id or alias must be provided.")
+            return
+
+        # If alias is provided, resolve it to alert_id
+        if alias:
+            alias_result = client.list_alerts(search=alias, limit=1, offset=0)
+            if alias_result["total"] == 0:
+                click.echo(f"No alert found with alias: {alias}")
+                return
+            alert_id = alias_result["data"][0]["id"]
+
+        # If alert_id is still not set, raise an error
+        if not alert_id:
+            click.echo("Error: Could not resolve alert_id.")
+            return
+
         result = client.resolve_alert(alert_id)
         click.echo(f"Alert resolved successfully: {result.get('id')}")
     except Exception as e:
         handle_api_error(e, action="resolving alert")
 
 @alerts.command(name="list-comments")
-@click.argument("alert_id", required=True)
+@click.argument("alert_id", required=False)  # Make alert_id optional
+@click.option("--alias", help="Alias for the alert")
 @click.option("--limit", default=10, type=click.IntRange(1, 100), help="Number of alerts per page")
 @click.option("--offset", default=0, type=click.IntRange(0), help="Starting point for pagination")
 @click.pass_context
-def list_alert_comment_cmd(ctx, alert_id, limit, offset):
-    """List an alerts comments in PagerTree."""
+def list_alert_comment_cmd(ctx, alert_id, alias, limit, offset):
+    """List an alert's comments in PagerTree."""
     try:
         client = ctx.obj  # Get PagerTreeClient from context
+
+        # Ensure at least one of alert_id or alias is provided
+        if not alert_id and not alias:
+            click.echo("Error: Either alert_id or alias must be provided.")
+            return
+
+        # If alias is provided, resolve it to alert_id
+        if alias:
+            alias_result = client.list_alerts(search=alias, limit=1, offset=0)
+            if alias_result["total"] == 0:
+                click.echo(f"No alert found with alias: {alias}")
+                return
+            alert_id = alias_result["data"][0]["id"]
+
+        # If alert_id is still not set, raise an error
+        if not alert_id:
+            click.echo("Error: Could not resolve alert_id.")
+            return
+
         result = client.list_alert_comments(alert_id, limit=limit, offset=offset)
         comments_list = result["data"]
         total = result["total"]
@@ -147,13 +227,33 @@ def list_alert_comment_cmd(ctx, alert_id, limit, offset):
         handle_api_error(e, action="listing alert comments")
 
 @alerts.command(name="comment")
-@click.argument("alert_id", required=True)
+@click.argument("alert_id", required=False)  # Make alert_id optional
+@click.option("--alias", help="Alias for the alert")
 @click.option("--comment", required=True, help="Comment to add to the alert")
 @click.pass_context
-def create_alert_comment_cmd(ctx, alert_id, comment):
+def create_alert_comment_cmd(ctx, alert_id, alias, comment):
     """Add a comment to an alert in PagerTree."""
     try:
         client = ctx.obj  # Get PagerTreeClient from context
+
+        # Ensure at least one of alert_id or alias is provided
+        if not alert_id and not alias:
+            click.echo("Error: Either alert_id or alias must be provided.")
+            return
+
+        # If alias is provided, resolve it to alert_id
+        if alias:
+            alias_result = client.list_alerts(search=alias, limit=1, offset=0)
+            if alias_result["total"] == 0:
+                click.echo(f"No alert found with alias: {alias}")
+                return
+            alert_id = alias_result["data"][0]["id"]
+
+        # If alert_id is still not set, raise an error
+        if not alert_id:
+            click.echo("Error: Could not resolve alert_id.")
+            return
+
         result = client.create_alert_comment(alert_id, comment)
         click.echo(f"Comment added successfully to alert {alert_id}")
     except Exception as e: