Merge pull request #8 from Veridise/sas-1567-digest

SAS-1567: Add api_send_digest

Author: mkonstant

audithub_client/api/send_digest.py

@@ -0,0 +1,37 @@
+#!/usr/bin/env python3
+from dataclasses import dataclass
+from datetime import datetime, timedelta, timezone
+from typing import Optional
+
+from ..library.auth import authentication_retry
+from ..library.context import AuditHubContext
+from ..library.http import GET
+from ..library.net_utils import ensure_success, response_json
+
+
+@dataclass
+class SendDigestNotificationsArgs:
+    user_id: Optional[str]
+    days_to_include: Optional[int]
+
+
+def get_query_timestamp(days_to_include: int):
+    return (datetime.now(timezone.utc) - timedelta(days=days_to_include)).isoformat()
+
+
+def api_send_digest_notifications(
+    context: AuditHubContext, input: SendDigestNotificationsArgs
+):
+
+    query_params = {}
+    if input.user_id:
+        query_params["user_id"] = input.user_id
+    if input.days_to_include:
+        query_params["from_created_at"] = get_query_timestamp(input.days_to_include)
+
+    response = authentication_retry(
+        context, GET, url=f"{context.base_url}/admin/digest", params=query_params
+    )
+    ensure_success(response)
+    ret = response_json(response)
+    return ret

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "audithub-client"
-version = "1.1.3"
+version = "1.1.4"
 description = "A Python client that can access Veridise AuditHub via its REST API, providing CLI access"
 authors = ["Nikos Chondros <nikos@veridise.com>"]
 license = "AGPLv3"