From e4da9692cd9ee325bd336768fbbf32c83664fd92 Mon Sep 17 00:00:00 2001
From: Dean Abargil <deanabargil@gmail.com>
Date: Sun, 7 Jan 2024 17:35:00 +0200
Subject: [PATCH] Updated audit logs API (#151)

* COR-29935: Updated audit logs API

* COR-29935: Updated paginated audit logs API
---
 fireblocks_sdk/sdk.py | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/fireblocks_sdk/sdk.py b/fireblocks_sdk/sdk.py
index 99bc084..aa85bb4 100644
--- a/fireblocks_sdk/sdk.py
+++ b/fireblocks_sdk/sdk.py
@@ -2289,7 +2289,7 @@ def get_off_exchanges(self):
 
         return self._get_request(url)
 
-    def get_audit_logs(self, time_period: TimePeriod):
+    def get_audit_logs(self, time_period: TimePeriod = TimePeriod.DAY):
         """
         Get audit logs
         :param time_period: The last time period to fetch audit logs
@@ -2298,6 +2298,23 @@ def get_audit_logs(self, time_period: TimePeriod):
         url = "/v1/audits"
 
         return self._get_request(url, query_params={"timePeriod": time_period.value})
+        
+    def get_paginated_audit_logs(self, time_period: TimePeriod = TimePeriod.DAY, cursor = None):
+        """
+        Get paginated audit logs
+        :param time_period: The last time period to fetch audit logs
+        :param cursor: The next id to fetch audit logs from
+        """
+        url = "/v1/management/audit_logs"
+        params = {}
+
+        if cursor:
+            params["cursor"] = cursor
+
+        if time_period:
+            params["timePeriod"] = time_period.value
+
+        return self._get_request(url, query_params=params)
 
     def get_off_exchange_by_id(self, off_exchange_id):
         """