diracx-client 0.0.1a11__py3-none-any.whl → 0.0.1a12__py3-none-any.whl

diracx/client/aio/_client.py

@@ -1,6 +1,6 @@
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.1, generator: @autorest/python@6.4.11)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.13.2)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
@@ -8,6 +8,7 @@ from copy import deepcopy
 from typing import Any, Awaitable
 
 from azure.core import AsyncPipelineClient
+from azure.core.pipeline import policies
 from azure.core.rest import AsyncHttpResponse, HttpRequest
 
 from .. import models as _models
@@ -40,8 +41,29 @@ class Dirac:  # pylint: disable=client-accepts-api-version-keyword
         self, *, endpoint: str = "", **kwargs: Any
     ) -> None:
         self._config = DiracConfiguration(**kwargs)
+        _policies = kwargs.pop("policies", None)
+        if _policies is None:
+            _policies = [
+                policies.RequestIdPolicy(**kwargs),
+                self._config.headers_policy,
+                self._config.user_agent_policy,
+                self._config.proxy_policy,
+                policies.ContentDecodePolicy(**kwargs),
+                self._config.redirect_policy,
+                self._config.retry_policy,
+                self._config.authentication_policy,
+                self._config.custom_hook_policy,
+                self._config.logging_policy,
+                policies.DistributedTracingPolicy(**kwargs),
+                (
+                    policies.SensitiveHeaderCleanupPolicy(**kwargs)
+                    if self._config.redirect_policy
+                    else None
+                ),
+                self._config.http_logging_policy,
+            ]
         self._client: AsyncPipelineClient = AsyncPipelineClient(
-            base_url=endpoint, config=self._config, **kwargs
+            base_url=endpoint, policies=_policies, **kwargs
         )
 
         client_models = {
@@ -53,7 +75,7 @@ class Dirac:  # pylint: disable=client-accepts-api-version-keyword
         self.well_known = WellKnownOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
-        self.auth = AuthOperations(  # pylint: disable=abstract-class-instantiated
+        self.auth = AuthOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
         self.config = ConfigOperations(
@@ -64,7 +86,7 @@ class Dirac:  # pylint: disable=client-accepts-api-version-keyword
         )
 
     def send_request(
-        self, request: HttpRequest, **kwargs: Any
+        self, request: HttpRequest, *, stream: bool = False, **kwargs: Any
     ) -> Awaitable[AsyncHttpResponse]:
         """Runs the network request through the client's chained policies.
 
@@ -85,7 +107,7 @@ class Dirac:  # pylint: disable=client-accepts-api-version-keyword
 
         request_copy = deepcopy(request)
         request_copy.url = self._client.format_url(request_copy.url)
-        return self._client.send_request(request_copy, **kwargs)
+        return self._client.send_request(request_copy, stream=stream, **kwargs)  # type: ignore
 
     async def close(self) -> None:
         await self._client.close()

diracx/client/aio/_configuration.py

@@ -1,18 +1,17 @@
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.1, generator: @autorest/python@6.4.11)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.13.2)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
 from typing import Any
 
-from azure.core.configuration import Configuration
 from azure.core.pipeline import policies
 
 VERSION = "unknown"
 
 
-class DiracConfiguration(Configuration):  # pylint: disable=too-many-instance-attributes
+class DiracConfiguration:  # pylint: disable=too-many-instance-attributes
     """Configuration for Dirac.
 
     Note that all parameters used to create this instance are saved as instance
@@ -20,9 +19,9 @@ class DiracConfiguration(Configuration):  # pylint: disable=too-many-instance-at
     """
 
     def __init__(self, **kwargs: Any) -> None:
-        super(DiracConfiguration, self).__init__(**kwargs)
 
         kwargs.setdefault("sdk_moniker", "dirac/{}".format(VERSION))
+        self.polling_interval = kwargs.get("polling_interval", 30)
         self._configure(**kwargs)
 
     def _configure(self, **kwargs: Any) -> None:
@@ -39,13 +38,13 @@ class DiracConfiguration(Configuration):  # pylint: disable=too-many-instance-at
         self.http_logging_policy = kwargs.get(
             "http_logging_policy"
         ) or policies.HttpLoggingPolicy(**kwargs)
-        self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(
-            **kwargs
-        )
         self.custom_hook_policy = kwargs.get(
             "custom_hook_policy"
         ) or policies.CustomHookPolicy(**kwargs)
         self.redirect_policy = kwargs.get(
             "redirect_policy"
         ) or policies.AsyncRedirectPolicy(**kwargs)
+        self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(
+            **kwargs
+        )
         self.authentication_policy = kwargs.get("authentication_policy")

diracx/client/aio/_vendor.py

@@ -1,8 +1,12 @@
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.1, generator: @autorest/python@6.4.11)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.13.2)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
+from typing import Optional
+
+from azure.core import MatchConditions
+
 
 def raise_if_not_implemented(cls, abstract_methods):
     not_implemented = [
@@ -15,3 +19,37 @@ def raise_if_not_implemented(cls, abstract_methods):
                 cls.__name__, "', '".join(not_implemented)
             )
         )
+
+
+def quote_etag(etag: Optional[str]) -> Optional[str]:
+    if not etag or etag == "*":
+        return etag
+    if etag.startswith("W/"):
+        return etag
+    if etag.startswith('"') and etag.endswith('"'):
+        return etag
+    if etag.startswith("'") and etag.endswith("'"):
+        return etag
+    return '"' + etag + '"'
+
+
+def prep_if_match(
+    etag: Optional[str], match_condition: Optional[MatchConditions]
+) -> Optional[str]:
+    if match_condition == MatchConditions.IfNotModified:
+        if_match = quote_etag(etag) if etag else None
+        return if_match
+    if match_condition == MatchConditions.IfPresent:
+        return "*"
+    return None
+
+
+def prep_if_none_match(
+    etag: Optional[str], match_condition: Optional[MatchConditions]
+) -> Optional[str]:
+    if match_condition == MatchConditions.IfModified:
+        if_none_match = quote_etag(etag) if etag else None
+        return if_none_match
+    if match_condition == MatchConditions.IfMissing:
+        return "*"
+    return None

diracx/client/aio/operations/__init__.py

@@ -1,6 +1,6 @@
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.1, generator: @autorest/python@6.4.11)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.13.2)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------