udl-sdk 0.1.0a10__py3-none-any.whl → 0.1.0a12__py3-none-any.whl

unifieddatalibrary/resources/scs/v2.py

@@ -56,7 +56,9 @@ class V2Resource(SyncAPIResource):
         self,
         *,
         path: str,
+        send_notification: bool | NotGiven = NOT_GIVEN,
         classification_marking: str | NotGiven = NOT_GIVEN,
+        delete_on: int | NotGiven = NOT_GIVEN,
         description: str | NotGiven = NOT_GIVEN,
         read_acl: str | NotGiven = NOT_GIVEN,
         tags: List[str] | NotGiven = NOT_GIVEN,
@@ -78,6 +80,8 @@ class V2Resource(SyncAPIResource):
         Args:
           path: The complete path for the object to be updated.
 
+          send_notification: Whether or not to send a notification that the target file/folder was updated.
+
           classification_marking: Classification marking of the folder or file in IC/CAPCO portion-marked format.
 
           description: Optional description for the file or folder.
@@ -105,6 +109,7 @@ class V2Resource(SyncAPIResource):
             body=maybe_transform(
                 {
                     "classification_marking": classification_marking,
+                    "delete_on": delete_on,
                     "description": description,
                     "read_acl": read_acl,
                     "tags": tags,
@@ -117,7 +122,13 @@ class V2Resource(SyncAPIResource):
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=maybe_transform({"path": path}, v2_update_params.V2UpdateParams),
+                query=maybe_transform(
+                    {
+                        "path": path,
+                        "send_notification": send_notification,
+                    },
+                    v2_update_params.V2UpdateParams,
+                ),
             ),
             cast_to=NoneType,
         )
@@ -265,8 +276,10 @@ class V2Resource(SyncAPIResource):
         classification_marking: str,
         path: str,
         body: FileTypes,
+        delete_after: str | NotGiven = NOT_GIVEN,
         description: str | NotGiven = NOT_GIVEN,
         overwrite: bool | NotGiven = NOT_GIVEN,
+        send_notification: bool | NotGiven = NOT_GIVEN,
         tags: str | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -287,10 +300,14 @@ class V2Resource(SyncAPIResource):
           path: The complete path for the upload including filename. Will attempt to create
               folders in path if necessary. Must start with '/'.
 
+          delete_after: Length of time after which to automatically delete the file.
+
           description: Optional description of uploaded document.
 
           overwrite: Whether or not to overwrite a file with the same name and path, if one exists.
 
+          send_notification: Whether or not to send a notification that this file was uploaded.
+
           tags: Optional array of provider/source specific tags for this data, used for
               implementing data owner conditional access controls to restrict access to the
               data.
@@ -316,8 +333,10 @@ class V2Resource(SyncAPIResource):
                     {
                         "classification_marking": classification_marking,
                         "path": path,
+                        "delete_after": delete_after,
                         "description": description,
                         "overwrite": overwrite,
+                        "send_notification": send_notification,
                         "tags": tags,
                     },
                     v2_file_upload_params.V2FileUploadParams,
@@ -330,7 +349,9 @@ class V2Resource(SyncAPIResource):
         self,
         *,
         path: str,
+        send_notification: bool | NotGiven = NOT_GIVEN,
         classification_marking: str | NotGiven = NOT_GIVEN,
+        delete_on: int | NotGiven = NOT_GIVEN,
         description: str | NotGiven = NOT_GIVEN,
         read_acl: str | NotGiven = NOT_GIVEN,
         tags: List[str] | NotGiven = NOT_GIVEN,
@@ -356,6 +377,8 @@ class V2Resource(SyncAPIResource):
           path: Path to create. Will attempt to create all folders in the path that do not
               exist. Must start and end with '/'.
 
+          send_notification: Whether or not to send a notification that this folder was created.
+
           classification_marking: Classification marking of the folder or file in IC/CAPCO portion-marked format.
 
           description: Optional description for the file or folder.
@@ -383,6 +406,7 @@ class V2Resource(SyncAPIResource):
             body=maybe_transform(
                 {
                     "classification_marking": classification_marking,
+                    "delete_on": delete_on,
                     "description": description,
                     "read_acl": read_acl,
                     "tags": tags,
@@ -395,7 +419,13 @@ class V2Resource(SyncAPIResource):
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=maybe_transform({"path": path}, v2_folder_create_params.V2FolderCreateParams),
+                query=maybe_transform(
+                    {
+                        "path": path,
+                        "send_notification": send_notification,
+                    },
+                    v2_folder_create_params.V2FolderCreateParams,
+                ),
             ),
             cast_to=NoneType,
         )
@@ -475,7 +505,9 @@ class AsyncV2Resource(AsyncAPIResource):
         self,
         *,
         path: str,
+        send_notification: bool | NotGiven = NOT_GIVEN,
         classification_marking: str | NotGiven = NOT_GIVEN,
+        delete_on: int | NotGiven = NOT_GIVEN,
         description: str | NotGiven = NOT_GIVEN,
         read_acl: str | NotGiven = NOT_GIVEN,
         tags: List[str] | NotGiven = NOT_GIVEN,
@@ -497,6 +529,8 @@ class AsyncV2Resource(AsyncAPIResource):
         Args:
           path: The complete path for the object to be updated.
 
+          send_notification: Whether or not to send a notification that the target file/folder was updated.
+
           classification_marking: Classification marking of the folder or file in IC/CAPCO portion-marked format.
 
           description: Optional description for the file or folder.
@@ -524,6 +558,7 @@ class AsyncV2Resource(AsyncAPIResource):
             body=await async_maybe_transform(
                 {
                     "classification_marking": classification_marking,
+                    "delete_on": delete_on,
                     "description": description,
                     "read_acl": read_acl,
                     "tags": tags,
@@ -536,7 +571,13 @@ class AsyncV2Resource(AsyncAPIResource):
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=await async_maybe_transform({"path": path}, v2_update_params.V2UpdateParams),
+                query=await async_maybe_transform(
+                    {
+                        "path": path,
+                        "send_notification": send_notification,
+                    },
+                    v2_update_params.V2UpdateParams,
+                ),
             ),
             cast_to=NoneType,
         )
@@ -684,8 +725,10 @@ class AsyncV2Resource(AsyncAPIResource):
         classification_marking: str,
         path: str,
         body: FileTypes,
+        delete_after: str | NotGiven = NOT_GIVEN,
         description: str | NotGiven = NOT_GIVEN,
         overwrite: bool | NotGiven = NOT_GIVEN,
+        send_notification: bool | NotGiven = NOT_GIVEN,
         tags: str | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -706,10 +749,14 @@ class AsyncV2Resource(AsyncAPIResource):
           path: The complete path for the upload including filename. Will attempt to create
               folders in path if necessary. Must start with '/'.
 
+          delete_after: Length of time after which to automatically delete the file.
+
           description: Optional description of uploaded document.
 
           overwrite: Whether or not to overwrite a file with the same name and path, if one exists.
 
+          send_notification: Whether or not to send a notification that this file was uploaded.
+
           tags: Optional array of provider/source specific tags for this data, used for
               implementing data owner conditional access controls to restrict access to the
               data.
@@ -735,8 +782,10 @@ class AsyncV2Resource(AsyncAPIResource):
                     {
                         "classification_marking": classification_marking,
                         "path": path,
+                        "delete_after": delete_after,
                         "description": description,
                         "overwrite": overwrite,
+                        "send_notification": send_notification,
                         "tags": tags,
                     },
                     v2_file_upload_params.V2FileUploadParams,
@@ -749,7 +798,9 @@ class AsyncV2Resource(AsyncAPIResource):
         self,
         *,
         path: str,
+        send_notification: bool | NotGiven = NOT_GIVEN,
         classification_marking: str | NotGiven = NOT_GIVEN,
+        delete_on: int | NotGiven = NOT_GIVEN,
         description: str | NotGiven = NOT_GIVEN,
         read_acl: str | NotGiven = NOT_GIVEN,
         tags: List[str] | NotGiven = NOT_GIVEN,
@@ -775,6 +826,8 @@ class AsyncV2Resource(AsyncAPIResource):
           path: Path to create. Will attempt to create all folders in the path that do not
               exist. Must start and end with '/'.
 
+          send_notification: Whether or not to send a notification that this folder was created.
+
           classification_marking: Classification marking of the folder or file in IC/CAPCO portion-marked format.
 
           description: Optional description for the file or folder.
@@ -802,6 +855,7 @@ class AsyncV2Resource(AsyncAPIResource):
             body=await async_maybe_transform(
                 {
                     "classification_marking": classification_marking,
+                    "delete_on": delete_on,
                     "description": description,
                     "read_acl": read_acl,
                     "tags": tags,
@@ -814,7 +868,13 @@ class AsyncV2Resource(AsyncAPIResource):
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=await async_maybe_transform({"path": path}, v2_folder_create_params.V2FolderCreateParams),
+                query=await async_maybe_transform(
+                    {
+                        "path": path,
+                        "send_notification": send_notification,
+                    },
+                    v2_folder_create_params.V2FolderCreateParams,
+                ),
             ),
             cast_to=NoneType,
         )

unifieddatalibrary/resources/secure_messaging.py

@@ -4,6 +4,8 @@ from __future__ import annotations
 
 import httpx
 
+from unifieddatalibrary.pagination import SyncKafkaOffsetPage, AsyncKafkaOffsetPage
+
 from ..types import (
     secure_messaging_get_messages_params,
     secure_messaging_describe_topic_params,
@@ -150,7 +152,7 @@ class SecureMessagingResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> SyncKafkaOffsetPage[object]:
         """Retrieve a set of messages from the given topic at the given offset.
 
         See Help >
@@ -168,8 +170,9 @@ class SecureMessagingResource(SyncAPIResource):
         if not topic:
             raise ValueError(f"Expected a non-empty value for `topic` but received {topic!r}")
         extra_headers = {"Accept": "*/*", **(extra_headers or {})}
-        return self._get(
+        return self._get_api_list(
             f"/sm/getMessages/{topic}/{offset}",
+            page=SyncKafkaOffsetPage.with_url_builder(lambda next_offset: f"/sm/getMessages/{topic}/{next_offset}"),
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -183,7 +186,7 @@ class SecureMessagingResource(SyncAPIResource):
                     secure_messaging_get_messages_params.SecureMessagingGetMessagesParams,
                 ),
             ),
-            cast_to=NoneType,
+            model=object,
         )
 
     def list_topics(
@@ -330,7 +333,7 @@ class AsyncSecureMessagingResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
+    ) -> AsyncKafkaOffsetPage[object]:
         """Retrieve a set of messages from the given topic at the given offset.
 
         See Help >
@@ -348,8 +351,9 @@ class AsyncSecureMessagingResource(AsyncAPIResource):
         if not topic:
             raise ValueError(f"Expected a non-empty value for `topic` but received {topic!r}")
         extra_headers = {"Accept": "*/*", **(extra_headers or {})}
-        return await self._get(
+        return await self._get_api_list(
             f"/sm/getMessages/{topic}/{offset}",
+            page=AsyncKafkaOffsetPage.with_url_builder(lambda next_offset: f"/sm/getMessages/{topic}/{next_offset}"),
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -363,7 +367,7 @@ class AsyncSecureMessagingResource(AsyncAPIResource):
                     secure_messaging_get_messages_params.SecureMessagingGetMessagesParams,
                 ),
             ),
-            cast_to=NoneType,
+            model=object,
         )
 
     async def list_topics(