Merge pull request #218 from atlanhq/DVX-68

DVX-68 Adds type checking to AtlanClient save and other methods
atlanhq · Jan 16, 2024 · e52dbab · e52dbab
2 parents 7ac08fc + ac24631
commit e52dbab
Show file tree

Hide file tree

Showing 17 changed files with 773 additions and 156 deletions.
diff --git a/pyatlan/client/admin.py b/pyatlan/client/admin.py
@@ -1,7 +1,7 @@
 # SPDX-License-Identifier: Apache-2.0
 # Copyright 2022 Atlan Pte. Ltd.
 
-from pydantic import ValidationError, parse_obj_as
+from pydantic import ValidationError, parse_obj_as, validate_arguments
 
 from pyatlan.client.common import ApiCaller
 from pyatlan.client.constants import ADMIN_EVENTS, KEYCLOAK_EVENTS
@@ -29,6 +29,7 @@ def __init__(self, client: ApiCaller):
             )
         self._client = client
 
+    @validate_arguments
     def get_keycloak_events(
         self, keycloak_request: KeycloakEventRequest
     ) -> KeycloakEventResponse:
@@ -60,6 +61,7 @@ def get_keycloak_events(
             events=events,
         )
 
+    @validate_arguments
     def get_admin_events(self, admin_request: AdminEventRequest) -> AdminEventResponse:
         """
         Retrieve admin events based on the supplied filters.