openmeter 1.0.0b147__tar.gz → 1.0.0b149__tar.gz

{openmeter-1.0.0b147 → openmeter-1.0.0b149}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: openmeter
-Version: 1.0.0b147
+Version: 1.0.0b149
 Summary: Client for OpenMeter: Real-Time and Scalable Usage Metering
 Home-page: https://openmeter.io
 License: Apache-2.0

{openmeter-1.0.0b147 → openmeter-1.0.0b149}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "openmeter"
-version = "v1.0.0-beta.147"
+version = "v1.0.0-beta.149"
 description = "Client for OpenMeter: Real-Time and Scalable Usage Metering"
 authors = ["Andras Toth <4157749+tothandras@users.noreply.github.com>"]
 license = "Apache-2.0"

{openmeter-1.0.0b147 → openmeter-1.0.0b149}/src/openmeter/_operations/_operations.py

@@ -1142,6 +1142,25 @@ def build_update_notification_rule_request(rule_id: str, **kwargs: Any) -> HttpR
     return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs)
 
 
+def build_test_notification_rule_request(rule_id: str, **kwargs: Any) -> HttpRequest:
+    _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
+
+    accept = _headers.pop("Accept", "application/json, application/problem+json")
+
+    # Construct URL
+    _url = "/api/v1/notification/rules/{ruleId}/test"
+    path_format_arguments = {
+        "ruleId": _SERIALIZER.url("rule_id", rule_id, "str"),
+    }
+
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
+
+    # Construct headers
+    _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
+
+    return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs)
+
+
 def build_list_notification_events_request(
     *,
     page: int = 1,
@@ -1186,23 +1205,6 @@ def build_list_notification_events_request(
     return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs)
 
 
-def build_create_notification_event_request(**kwargs: Any) -> HttpRequest:
-    _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
-
-    content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
-    accept = _headers.pop("Accept", "application/json, application/problem+json")
-
-    # Construct URL
-    _url = "/api/v1/notification/events"
-
-    # Construct headers
-    if content_type is not None:
-        _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str")
-    _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
-
-    return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs)
-
-
 def build_get_notification_event_request(event_id: str, **kwargs: Any) -> HttpRequest:
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
 
@@ -6694,6 +6696,91 @@ class ClientOperationsMixin(ClientMixinABC):  # pylint: disable=too-many-public-
 
         return cast(JSON, deserialized)  # type: ignore
 
+    @distributed_trace
+    def test_notification_rule(self, rule_id: str, **kwargs: Any) -> JSON:
+        """Test notification rule.
+
+        Test a notification rule by sending a test event with random data.
+
+        :param rule_id: A unique ULID identifier for a notification rule. Required.
+        :type rule_id: str
+        :return: JSON object
+        :rtype: JSON
+        :raises ~azure.core.exceptions.HttpResponseError:
+
+        Example:
+            .. code-block:: python
+
+                # response body for status code(s): 409
+                response == {
+                    "detail": "str",  # A human-readable explanation specific to this occurrence
+                      of the problem. Required.
+                    "status": 0,  # The HTTP status code generated by the origin server for this
+                      occurrence of the problem. Required.
+                    "title": "str",  # A a short, human-readable summary of the problem type.
+                      Required.
+                    "type": "str",  # Type contains a URI that identifies the problem type.
+                      Required.
+                    "extensions": {
+                        "conflictingEntityId": "str"  # The id of the conflicting entity.
+                          Required.
+                    },
+                    "instance": "str"  # Optional. A URI reference that identifies the specific
+                      occurrence of the problem.
+                }
+        """
+        error_map = {
+            404: ResourceNotFoundError,
+            409: ResourceExistsError,
+            304: ResourceNotModifiedError,
+            400: HttpResponseError,
+            401: lambda response: ClientAuthenticationError(response=response),
+            501: HttpResponseError,
+        }
+        error_map.update(kwargs.pop("error_map", {}) or {})
+
+        _headers = kwargs.pop("headers", {}) or {}
+        _params = kwargs.pop("params", {}) or {}
+
+        cls: ClsType[JSON] = kwargs.pop("cls", None)
+
+        _request = build_test_notification_rule_request(
+            rule_id=rule_id,
+            headers=_headers,
+            params=_params,
+        )
+        _request.url = self._client.format_url(_request.url)
+
+        _stream = False
+        pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
+            _request, stream=_stream, **kwargs
+        )
+
+        response = pipeline_response.http_response
+
+        if response.status_code not in [201, 409]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
+            map_error(status_code=response.status_code, response=response, error_map=error_map)  # type: ignore
+            raise HttpResponseError(response=response)
+
+        if response.status_code == 201:
+            if response.content:
+                deserialized = response.json()
+            else:
+                deserialized = None
+
+        if response.status_code == 409:
+            if response.content:
+                deserialized = response.json()
+            else:
+                deserialized = None
+
+        if cls:
+            return cls(pipeline_response, cast(JSON, deserialized), {})  # type: ignore
+
+        return cast(JSON, deserialized)  # type: ignore
+
     @distributed_trace
     def list_notification_events(
         self,
@@ -6770,8 +6857,12 @@ class ClientOperationsMixin(ClientMixinABC):  # pylint: disable=too-many-public-
                               event. Required.
                             "payload": {},
                             "rule": {},
-                            "type": "str"  # The type of the notification event.
+                            "type": "str",  # The type of the notification event.
                               Required. "entitlements.balance.threshold"
+                            "annotations": {
+                                "str": "str"  # Optional. List of annotations managed
+                                  by the system.
+                            }
                         }
                     ],
                     "page": 0,  # Current page number. Required.
@@ -6830,195 +6921,6 @@ class ClientOperationsMixin(ClientMixinABC):  # pylint: disable=too-many-public-
 
         return cast(JSON, deserialized)  # type: ignore
 
-    @overload
-    def create_notification_event(self, body: JSON, *, content_type: str = "application/json", **kwargs: Any) -> JSON:
-        """Create a notification event.
-
-        Create a new notification event.
-
-        :param body: The notification event to create. Required.
-        :type body: JSON
-        :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
-         Default value is "application/json".
-        :paramtype content_type: str
-        :return: JSON object
-        :rtype: JSON
-        :raises ~azure.core.exceptions.HttpResponseError:
-
-        Example:
-            .. code-block:: python
-
-                # JSON input template you can fill out and use as your body input.
-                body = {
-                    "payload": {},
-                    "ruleId": "str",  # Required.
-                    "type": "str"  # The type of the notification event. Required.
-                      "entitlements.balance.threshold"
-                }
-
-                # response body for status code(s): 409
-                response == {
-                    "detail": "str",  # A human-readable explanation specific to this occurrence
-                      of the problem. Required.
-                    "status": 0,  # The HTTP status code generated by the origin server for this
-                      occurrence of the problem. Required.
-                    "title": "str",  # A a short, human-readable summary of the problem type.
-                      Required.
-                    "type": "str",  # Type contains a URI that identifies the problem type.
-                      Required.
-                    "extensions": {
-                        "conflictingEntityId": "str"  # The id of the conflicting entity.
-                          Required.
-                    },
-                    "instance": "str"  # Optional. A URI reference that identifies the specific
-                      occurrence of the problem.
-                }
-        """
-
-    @overload
-    def create_notification_event(
-        self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
-    ) -> JSON:
-        """Create a notification event.
-
-        Create a new notification event.
-
-        :param body: The notification event to create. Required.
-        :type body: IO[bytes]
-        :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
-         Default value is "application/json".
-        :paramtype content_type: str
-        :return: JSON object
-        :rtype: JSON
-        :raises ~azure.core.exceptions.HttpResponseError:
-
-        Example:
-            .. code-block:: python
-
-                # response body for status code(s): 409
-                response == {
-                    "detail": "str",  # A human-readable explanation specific to this occurrence
-                      of the problem. Required.
-                    "status": 0,  # The HTTP status code generated by the origin server for this
-                      occurrence of the problem. Required.
-                    "title": "str",  # A a short, human-readable summary of the problem type.
-                      Required.
-                    "type": "str",  # Type contains a URI that identifies the problem type.
-                      Required.
-                    "extensions": {
-                        "conflictingEntityId": "str"  # The id of the conflicting entity.
-                          Required.
-                    },
-                    "instance": "str"  # Optional. A URI reference that identifies the specific
-                      occurrence of the problem.
-                }
-        """
-
-    @distributed_trace
-    def create_notification_event(self, body: Union[JSON, IO[bytes]], **kwargs: Any) -> JSON:
-        """Create a notification event.
-
-        Create a new notification event.
-
-        :param body: The notification event to create. Is either a JSON type or a IO[bytes] type.
-         Required.
-        :type body: JSON or IO[bytes]
-        :return: JSON object
-        :rtype: JSON
-        :raises ~azure.core.exceptions.HttpResponseError:
-
-        Example:
-            .. code-block:: python
-
-                # JSON input template you can fill out and use as your body input.
-                body = {
-                    "payload": {},
-                    "ruleId": "str",  # Required.
-                    "type": "str"  # The type of the notification event. Required.
-                      "entitlements.balance.threshold"
-                }
-
-                # response body for status code(s): 409
-                response == {
-                    "detail": "str",  # A human-readable explanation specific to this occurrence
-                      of the problem. Required.
-                    "status": 0,  # The HTTP status code generated by the origin server for this
-                      occurrence of the problem. Required.
-                    "title": "str",  # A a short, human-readable summary of the problem type.
-                      Required.
-                    "type": "str",  # Type contains a URI that identifies the problem type.
-                      Required.
-                    "extensions": {
-                        "conflictingEntityId": "str"  # The id of the conflicting entity.
-                          Required.
-                    },
-                    "instance": "str"  # Optional. A URI reference that identifies the specific
-                      occurrence of the problem.
-                }
-        """
-        error_map = {
-            404: ResourceNotFoundError,
-            409: ResourceExistsError,
-            304: ResourceNotModifiedError,
-            400: HttpResponseError,
-            401: lambda response: ClientAuthenticationError(response=response),
-            501: HttpResponseError,
-        }
-        error_map.update(kwargs.pop("error_map", {}) or {})
-
-        _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
-        _params = kwargs.pop("params", {}) or {}
-
-        content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
-        cls: ClsType[JSON] = kwargs.pop("cls", None)
-
-        content_type = content_type or "application/json"
-        _json = None
-        _content = None
-        if isinstance(body, (IOBase, bytes)):
-            _content = body
-        else:
-            _json = body
-
-        _request = build_create_notification_event_request(
-            content_type=content_type,
-            json=_json,
-            content=_content,
-            headers=_headers,
-            params=_params,
-        )
-        _request.url = self._client.format_url(_request.url)
-
-        _stream = False
-        pipeline_response: PipelineResponse = self._client._pipeline.run(  # pylint: disable=protected-access
-            _request, stream=_stream, **kwargs
-        )
-
-        response = pipeline_response.http_response
-
-        if response.status_code not in [201, 409]:
-            if _stream:
-                response.read()  # Load the body in memory and close the socket
-            map_error(status_code=response.status_code, response=response, error_map=error_map)  # type: ignore
-            raise HttpResponseError(response=response)
-
-        if response.status_code == 201:
-            if response.content:
-                deserialized = response.json()
-            else:
-                deserialized = None
-
-        if response.status_code == 409:
-            if response.content:
-                deserialized = response.json()
-            else:
-                deserialized = None
-
-        if cls:
-            return cls(pipeline_response, cast(JSON, deserialized), {})  # type: ignore
-
-        return cast(JSON, deserialized)  # type: ignore
-
     @distributed_trace
     def get_notification_event(self, event_id: str, **kwargs: Any) -> JSON:
         """Get notification event.

{openmeter-1.0.0b147 → openmeter-1.0.0b149}/src/openmeter/aio/_operations/_operations.py

@@ -28,7 +28,6 @@ from ..._operations._operations import (
     build_create_grant_request,
     build_create_meter_request,
     build_create_notification_channel_request,
-    build_create_notification_event_request,
     build_create_notification_rule_request,
     build_create_portal_token_request,
     build_delete_entitlement_request,
@@ -68,6 +67,7 @@ from ..._operations._operations import (
     build_query_portal_meter_request,
     build_receive_svix_operational_event_request,
     build_reset_entitlement_usage_request,
+    build_test_notification_rule_request,
     build_update_notification_channel_request,
     build_update_notification_rule_request,
     build_upsert_subject_request,
@@ -5551,6 +5551,91 @@ class ClientOperationsMixin(ClientMixinABC):  # pylint: disable=too-many-public-
 
         return cast(JSON, deserialized)  # type: ignore
 
+    @distributed_trace_async
+    async def test_notification_rule(self, rule_id: str, **kwargs: Any) -> JSON:
+        """Test notification rule.
+
+        Test a notification rule by sending a test event with random data.
+
+        :param rule_id: A unique ULID identifier for a notification rule. Required.
+        :type rule_id: str
+        :return: JSON object
+        :rtype: JSON
+        :raises ~azure.core.exceptions.HttpResponseError:
+
+        Example:
+            .. code-block:: python
+
+                # response body for status code(s): 409
+                response == {
+                    "detail": "str",  # A human-readable explanation specific to this occurrence
+                      of the problem. Required.
+                    "status": 0,  # The HTTP status code generated by the origin server for this
+                      occurrence of the problem. Required.
+                    "title": "str",  # A a short, human-readable summary of the problem type.
+                      Required.
+                    "type": "str",  # Type contains a URI that identifies the problem type.
+                      Required.
+                    "extensions": {
+                        "conflictingEntityId": "str"  # The id of the conflicting entity.
+                          Required.
+                    },
+                    "instance": "str"  # Optional. A URI reference that identifies the specific
+                      occurrence of the problem.
+                }
+        """
+        error_map = {
+            404: ResourceNotFoundError,
+            409: ResourceExistsError,
+            304: ResourceNotModifiedError,
+            400: HttpResponseError,
+            401: lambda response: ClientAuthenticationError(response=response),
+            501: HttpResponseError,
+        }
+        error_map.update(kwargs.pop("error_map", {}) or {})
+
+        _headers = kwargs.pop("headers", {}) or {}
+        _params = kwargs.pop("params", {}) or {}
+
+        cls: ClsType[JSON] = kwargs.pop("cls", None)
+
+        _request = build_test_notification_rule_request(
+            rule_id=rule_id,
+            headers=_headers,
+            params=_params,
+        )
+        _request.url = self._client.format_url(_request.url)
+
+        _stream = False
+        pipeline_response: PipelineResponse = await self._client._pipeline.run(  # type: ignore # pylint: disable=protected-access
+            _request, stream=_stream, **kwargs
+        )
+
+        response = pipeline_response.http_response
+
+        if response.status_code not in [201, 409]:
+            if _stream:
+                await response.read()  # Load the body in memory and close the socket
+            map_error(status_code=response.status_code, response=response, error_map=error_map)  # type: ignore
+            raise HttpResponseError(response=response)
+
+        if response.status_code == 201:
+            if response.content:
+                deserialized = response.json()
+            else:
+                deserialized = None
+
+        if response.status_code == 409:
+            if response.content:
+                deserialized = response.json()
+            else:
+                deserialized = None
+
+        if cls:
+            return cls(pipeline_response, cast(JSON, deserialized), {})  # type: ignore
+
+        return cast(JSON, deserialized)  # type: ignore
+
     @distributed_trace_async
     async def list_notification_events(
         self,
@@ -5627,8 +5712,12 @@ class ClientOperationsMixin(ClientMixinABC):  # pylint: disable=too-many-public-
                               event. Required.
                             "payload": {},
                             "rule": {},
-                            "type": "str"  # The type of the notification event.
+                            "type": "str",  # The type of the notification event.
                               Required. "entitlements.balance.threshold"
+                            "annotations": {
+                                "str": "str"  # Optional. List of annotations managed
+                                  by the system.
+                            }
                         }
                     ],
                     "page": 0,  # Current page number. Required.
@@ -5687,197 +5776,6 @@ class ClientOperationsMixin(ClientMixinABC):  # pylint: disable=too-many-public-
 
         return cast(JSON, deserialized)  # type: ignore
 
-    @overload
-    async def create_notification_event(
-        self, body: JSON, *, content_type: str = "application/json", **kwargs: Any
-    ) -> JSON:
-        """Create a notification event.
-
-        Create a new notification event.
-
-        :param body: The notification event to create. Required.
-        :type body: JSON
-        :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
-         Default value is "application/json".
-        :paramtype content_type: str
-        :return: JSON object
-        :rtype: JSON
-        :raises ~azure.core.exceptions.HttpResponseError:
-
-        Example:
-            .. code-block:: python
-
-                # JSON input template you can fill out and use as your body input.
-                body = {
-                    "payload": {},
-                    "ruleId": "str",  # Required.
-                    "type": "str"  # The type of the notification event. Required.
-                      "entitlements.balance.threshold"
-                }
-
-                # response body for status code(s): 409
-                response == {
-                    "detail": "str",  # A human-readable explanation specific to this occurrence
-                      of the problem. Required.
-                    "status": 0,  # The HTTP status code generated by the origin server for this
-                      occurrence of the problem. Required.
-                    "title": "str",  # A a short, human-readable summary of the problem type.
-                      Required.
-                    "type": "str",  # Type contains a URI that identifies the problem type.
-                      Required.
-                    "extensions": {
-                        "conflictingEntityId": "str"  # The id of the conflicting entity.
-                          Required.
-                    },
-                    "instance": "str"  # Optional. A URI reference that identifies the specific
-                      occurrence of the problem.
-                }
-        """
-
-    @overload
-    async def create_notification_event(
-        self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
-    ) -> JSON:
-        """Create a notification event.
-
-        Create a new notification event.
-
-        :param body: The notification event to create. Required.
-        :type body: IO[bytes]
-        :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
-         Default value is "application/json".
-        :paramtype content_type: str
-        :return: JSON object
-        :rtype: JSON
-        :raises ~azure.core.exceptions.HttpResponseError:
-
-        Example:
-            .. code-block:: python
-
-                # response body for status code(s): 409
-                response == {
-                    "detail": "str",  # A human-readable explanation specific to this occurrence
-                      of the problem. Required.
-                    "status": 0,  # The HTTP status code generated by the origin server for this
-                      occurrence of the problem. Required.
-                    "title": "str",  # A a short, human-readable summary of the problem type.
-                      Required.
-                    "type": "str",  # Type contains a URI that identifies the problem type.
-                      Required.
-                    "extensions": {
-                        "conflictingEntityId": "str"  # The id of the conflicting entity.
-                          Required.
-                    },
-                    "instance": "str"  # Optional. A URI reference that identifies the specific
-                      occurrence of the problem.
-                }
-        """
-
-    @distributed_trace_async
-    async def create_notification_event(self, body: Union[JSON, IO[bytes]], **kwargs: Any) -> JSON:
-        """Create a notification event.
-
-        Create a new notification event.
-
-        :param body: The notification event to create. Is either a JSON type or a IO[bytes] type.
-         Required.
-        :type body: JSON or IO[bytes]
-        :return: JSON object
-        :rtype: JSON
-        :raises ~azure.core.exceptions.HttpResponseError:
-
-        Example:
-            .. code-block:: python
-
-                # JSON input template you can fill out and use as your body input.
-                body = {
-                    "payload": {},
-                    "ruleId": "str",  # Required.
-                    "type": "str"  # The type of the notification event. Required.
-                      "entitlements.balance.threshold"
-                }
-
-                # response body for status code(s): 409
-                response == {
-                    "detail": "str",  # A human-readable explanation specific to this occurrence
-                      of the problem. Required.
-                    "status": 0,  # The HTTP status code generated by the origin server for this
-                      occurrence of the problem. Required.
-                    "title": "str",  # A a short, human-readable summary of the problem type.
-                      Required.
-                    "type": "str",  # Type contains a URI that identifies the problem type.
-                      Required.
-                    "extensions": {
-                        "conflictingEntityId": "str"  # The id of the conflicting entity.
-                          Required.
-                    },
-                    "instance": "str"  # Optional. A URI reference that identifies the specific
-                      occurrence of the problem.
-                }
-        """
-        error_map = {
-            404: ResourceNotFoundError,
-            409: ResourceExistsError,
-            304: ResourceNotModifiedError,
-            400: HttpResponseError,
-            401: lambda response: ClientAuthenticationError(response=response),
-            501: HttpResponseError,
-        }
-        error_map.update(kwargs.pop("error_map", {}) or {})
-
-        _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
-        _params = kwargs.pop("params", {}) or {}
-
-        content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
-        cls: ClsType[JSON] = kwargs.pop("cls", None)
-
-        content_type = content_type or "application/json"
-        _json = None
-        _content = None
-        if isinstance(body, (IOBase, bytes)):
-            _content = body
-        else:
-            _json = body
-
-        _request = build_create_notification_event_request(
-            content_type=content_type,
-            json=_json,
-            content=_content,
-            headers=_headers,
-            params=_params,
-        )
-        _request.url = self._client.format_url(_request.url)
-
-        _stream = False
-        pipeline_response: PipelineResponse = await self._client._pipeline.run(  # type: ignore # pylint: disable=protected-access
-            _request, stream=_stream, **kwargs
-        )
-
-        response = pipeline_response.http_response
-
-        if response.status_code not in [201, 409]:
-            if _stream:
-                await response.read()  # Load the body in memory and close the socket
-            map_error(status_code=response.status_code, response=response, error_map=error_map)  # type: ignore
-            raise HttpResponseError(response=response)
-
-        if response.status_code == 201:
-            if response.content:
-                deserialized = response.json()
-            else:
-                deserialized = None
-
-        if response.status_code == 409:
-            if response.content:
-                deserialized = response.json()
-            else:
-                deserialized = None
-
-        if cls:
-            return cls(pipeline_response, cast(JSON, deserialized), {})  # type: ignore
-
-        return cast(JSON, deserialized)  # type: ignore
-
     @distributed_trace_async
     async def get_notification_event(self, event_id: str, **kwargs: Any) -> JSON:
         """Get notification event.