update get_[operation]_status to private

add dependency on azure-ai-language-questionanswering

Author: Qi-2077

sdk/cognitivelanguage/azure-ai-language-questionanswering-authoring/azure/ai/language/questionanswering/authoring/_operations/_patch.py

@@ -18,13 +18,15 @@
 from ._operations import (
     _QuestionAnsweringAuthoringClientOperationsMixin as _QuestionAnsweringAuthoringClientOperationsMixinGenerated,
 )
+
 JSON = MutableMapping[str, Any]
 
+
 class _QuestionAnsweringAuthoringClientOperationsMixin(_QuestionAnsweringAuthoringClientOperationsMixinGenerated):
     """Mixin class for patching methods with backward compatible parameter names."""
 
     # create_project overloads with 'options' parameter
-    @overload # type: ignore
+    @overload  # type: ignore
     def create_project(
         self,
         project_name: str,
@@ -48,7 +50,7 @@ def create_project(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    @overload # type: ignore
+    @overload  # type: ignore
     def create_project(
         self, project_name: str, options: JSON, *, content_type: str = "application/json", **kwargs: Any
     ) -> _models.QuestionAnsweringProject:
@@ -67,7 +69,7 @@ def create_project(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    @overload # type: ignore
+    @overload  # type: ignore
     def create_project(
         self, project_name: str, options: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> _models.QuestionAnsweringProject:
@@ -87,7 +89,7 @@ def create_project(
         """
 
     @distributed_trace
-    def create_project( # pyright: ignore[reportIncompatibleMethodOverride]
+    def create_project(  # pyright: ignore[reportIncompatibleMethodOverride]
         self, project_name: str, options: Union[_models.QuestionAnsweringProject, JSON, IO[bytes]], **kwargs: Any
     ) -> _models.QuestionAnsweringProject:
         """Create or update a project.
@@ -107,9 +109,14 @@ def create_project( # pyright: ignore[reportIncompatibleMethodOverride]
         return super().create_project(project_name=project_name, body=options, **kwargs)
 
     # update_synonyms overloads with 'synonyms' parameter
-    @overload # type: ignore
+    @overload  # type: ignore
     def update_synonyms(
-        self, project_name: str, synonyms: _models.SynonymAssets, *, content_type: str = "application/json", **kwargs: Any
+        self,
+        project_name: str,
+        synonyms: _models.SynonymAssets,
+        *,
+        content_type: str = "application/json",
+        **kwargs: Any
     ) -> None:
         """Updates all the synonyms of a project.
 
@@ -125,7 +132,7 @@ def update_synonyms(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    @overload # type: ignore
+    @overload  # type: ignore
     def update_synonyms(
         self, project_name: str, synonyms: JSON, *, content_type: str = "application/json", **kwargs: Any
     ) -> None:
@@ -143,7 +150,7 @@ def update_synonyms(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    @overload # type: ignore
+    @overload  # type: ignore
     def update_synonyms(
         self, project_name: str, synonyms: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> None:
@@ -181,7 +188,7 @@ def update_synonyms(  # pyright: ignore[reportIncompatibleMethodOverride]
         return super().update_synonyms(project_name=project_name, body=synonyms, **kwargs)
 
     # begin_update_qnas overloads with 'qnas' parameter
-    @overload # type: ignore
+    @overload  # type: ignore
     def begin_update_qnas(
         self,
         project_name: str,
@@ -204,7 +211,7 @@ def begin_update_qnas(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    @overload # type: ignore
+    @overload  # type: ignore
     def begin_update_qnas(
         self, project_name: str, qnas: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> LROPoller[None]:
@@ -223,7 +230,7 @@ def begin_update_qnas(
         """
 
     @distributed_trace
-    def begin_update_qnas( # pyright: ignore[reportIncompatibleMethodOverride]
+    def begin_update_qnas(  # pyright: ignore[reportIncompatibleMethodOverride]
         self, project_name: str, qnas: Union[list[_models.UpdateQnaRecord], IO[bytes]], **kwargs: Any
     ) -> LROPoller[None]:
         """Updates the QnAs of a project.
@@ -242,7 +249,7 @@ def begin_update_qnas( # pyright: ignore[reportIncompatibleMethodOverride]
         return super().begin_update_qnas(project_name=project_name, body=qnas, **kwargs)
 
     # begin_update_sources overloads with 'sources' parameter
-    @overload # type: ignore
+    @overload  # type: ignore
     def begin_update_sources(
         self,
         project_name: str,
@@ -265,7 +272,7 @@ def begin_update_sources(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    @overload # type: ignore
+    @overload  # type: ignore
     def begin_update_sources(
         self, project_name: str, sources: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> LROPoller[None]:
@@ -284,7 +291,7 @@ def begin_update_sources(
         """
 
     @distributed_trace
-    def begin_update_sources( # pyright: ignore[reportIncompatibleMethodOverride]
+    def begin_update_sources(  # pyright: ignore[reportIncompatibleMethodOverride]
         self, project_name: str, sources: Union[list[_models.UpdateSourceRecord], IO[bytes]], **kwargs: Any
     ) -> LROPoller[None]:
         """Updates the sources of a project.
@@ -300,9 +307,7 @@ def begin_update_sources( # pyright: ignore[reportIncompatibleMethodOverride]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
         # Call the parent implementation with 'body' parameter for backward compatibility
-        return super().begin_update_sources(
-            project_name=project_name, body=sources, **kwargs
-        )
+        return super().begin_update_sources(project_name=project_name, body=sources, **kwargs)
 
 
 __all__: list[str] = [
@@ -317,3 +322,41 @@ def patch_sdk():
     you can't accomplish using the techniques described in
     https://aka.ms/azsdk/python/dpcodegen/python/customize
     """
+    # Hide the generated get_*_status LRO status peek methods from the public surface
+    # without editing the generated _operations.py file. We do this by:
+    # 1. Capturing references to the original methods.
+    # 2. Deleting those attributes from the generated mixin class so normal attribute
+    #    resolution on the public client no longer finds them.
+    # 3. Exposing private underscore-prefixed versions on our patch subclass for
+    #    internal/testing use if ever needed.
+    #
+    # NOTE: This is a breaking change for any user code calling these methods directly.
+    #       Provide release notes / changelog entry accordingly.
+    original_methods: dict[str, Any] = {}
+    to_hide = [
+        "get_delete_status",
+        "get_export_status",
+        "get_import_status",
+        "get_deploy_status",
+    ]
+    generated_cls = _QuestionAnsweringAuthoringClientOperationsMixinGenerated
+    for name in to_hide:
+        if hasattr(generated_cls, name):
+            original_methods[name] = getattr(generated_cls, name)
+            try:
+                delattr(generated_cls, name)
+            except AttributeError:
+                pass  # Already removed
+
+    # Attach private replicas to the patch subclass so internal code could still call them
+    patch_cls = _QuestionAnsweringAuthoringClientOperationsMixin
+    for public_name, func in original_methods.items():
+        private_name = f"_{public_name}"  # e.g. _get_delete_status
+        # Avoid overwriting if already defined.
+        if not hasattr(patch_cls, private_name):
+            # Bind function so it behaves like an instance method on patch_cls.
+            setattr(patch_cls, private_name, func)
+
+    # Ensure they are not exported via __all__ even if someone adds them later.
+    # (__all__ currently only lists the mixin class.)
+    # No action needed unless we want to explicitly assert.

sdk/cognitivelanguage/azure-ai-language-questionanswering-authoring/azure/ai/language/questionanswering/authoring/aio/_operations/_patch.py

@@ -16,13 +16,15 @@
 from ._operations import (
     _QuestionAnsweringAuthoringClientOperationsMixin as _QuestionAnsweringAuthoringClientOperationsMixinGenerated,
 )
+
 JSON = MutableMapping[str, Any]
 
+
 class _QuestionAnsweringAuthoringClientOperationsMixin(_QuestionAnsweringAuthoringClientOperationsMixinGenerated):
     """Mixin class for patching methods with backward compatible parameter names."""
 
     # create_project overloads with 'options' parameter
-    @overload # type: ignore
+    @overload  # type: ignore
     async def create_project(
         self,
         project_name: str,
@@ -45,7 +47,7 @@ async def create_project(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    @overload # type: ignore
+    @overload  # type: ignore
     async def create_project(
         self, project_name: str, options: JSON, *, content_type: str = "application/json", **kwargs: Any
     ) -> _models.QuestionAnsweringProject:
@@ -64,7 +66,7 @@ async def create_project(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    @overload # type: ignore
+    @overload  # type: ignore
     async def create_project(
         self, project_name: str, options: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> _models.QuestionAnsweringProject:
@@ -84,10 +86,8 @@ async def create_project(
         """
 
     @distributed_trace_async
-    async def create_project( # pyright: ignore[reportIncompatibleMethodOverride]
-        self,
-        project_name: str,
-        options: Union[_models.QuestionAnsweringProject, JSON, IO[bytes]], **kwargs: Any
+    async def create_project(  # pyright: ignore[reportIncompatibleMethodOverride]
+        self, project_name: str, options: Union[_models.QuestionAnsweringProject, JSON, IO[bytes]], **kwargs: Any
     ) -> _models.QuestionAnsweringProject:
         """Create or update a project.
 
@@ -106,9 +106,14 @@ async def create_project( # pyright: ignore[reportIncompatibleMethodOverride]
         return await super().create_project(project_name=project_name, body=options, **kwargs)
 
     # update_synonyms overloads with 'synonyms' parameter
-    @overload # type: ignore
+    @overload  # type: ignore
     async def update_synonyms(
-        self, project_name: str, synonyms: _models.SynonymAssets, *, content_type: str = "application/json", **kwargs: Any
+        self,
+        project_name: str,
+        synonyms: _models.SynonymAssets,
+        *,
+        content_type: str = "application/json",
+        **kwargs: Any
     ) -> None:
         """Updates all the synonyms of a project.
 
@@ -124,7 +129,7 @@ async def update_synonyms(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    @overload # type: ignore
+    @overload  # type: ignore
     async def update_synonyms(
         self, project_name: str, synonyms: JSON, *, content_type: str = "application/json", **kwargs: Any
     ) -> None:
@@ -142,7 +147,7 @@ async def update_synonyms(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    @overload # type: ignore
+    @overload  # type: ignore
     async def update_synonyms(
         self, project_name: str, synonyms: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> None:
@@ -161,7 +166,7 @@ async def update_synonyms(
         """
 
     @distributed_trace_async
-    async def update_synonyms( # pyright: ignore[reportIncompatibleMethodOverride]
+    async def update_synonyms(  # pyright: ignore[reportIncompatibleMethodOverride]
         self, project_name: str, synonyms: Union[_models.SynonymAssets, JSON, IO[bytes]], **kwargs: Any
     ) -> None:
         """Updates all the synonyms of a project.
@@ -180,7 +185,7 @@ async def update_synonyms( # pyright: ignore[reportIncompatibleMethodOverride]
         return await super().update_synonyms(project_name=project_name, body=synonyms, **kwargs)
 
     # begin_update_qnas overloads with 'qnas' parameter
-    @overload # type: ignore
+    @overload  # type: ignore
     async def begin_update_qnas(
         self,
         project_name: str,
@@ -203,7 +208,7 @@ async def begin_update_qnas(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    @overload # type: ignore
+    @overload  # type: ignore
     async def begin_update_qnas(
         self, project_name: str, qnas: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> AsyncLROPoller[None]:
@@ -222,7 +227,7 @@ async def begin_update_qnas(
         """
 
     @distributed_trace_async
-    async def begin_update_qnas( # pyright: ignore[reportIncompatibleMethodOverride]
+    async def begin_update_qnas(  # pyright: ignore[reportIncompatibleMethodOverride]
         self, project_name: str, qnas: Union[list[_models.UpdateQnaRecord], IO[bytes]], **kwargs: Any
     ) -> AsyncLROPoller[None]:
         """Updates the QnAs of a project.
@@ -238,12 +243,10 @@ async def begin_update_qnas( # pyright: ignore[reportIncompatibleMethodOverride]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
         # Call the parent implementation with 'body' parameter for backward compatibility
-        return await super().begin_update_qnas(
-            project_name=project_name, body=qnas, **kwargs
-        )
+        return await super().begin_update_qnas(project_name=project_name, body=qnas, **kwargs)
 
     # begin_update_sources overloads with 'sources' parameter
-    @overload # type: ignore
+    @overload  # type: ignore
     async def begin_update_sources(
         self,
         project_name: str,
@@ -266,7 +269,7 @@ async def begin_update_sources(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    @overload # type: ignore
+    @overload  # type: ignore
     async def begin_update_sources(
         self, project_name: str, sources: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> AsyncLROPoller[None]:
@@ -285,7 +288,7 @@ async def begin_update_sources(
         """
 
     @distributed_trace_async
-    async def begin_update_sources( # pyright: ignore[reportIncompatibleMethodOverride]
+    async def begin_update_sources(  # pyright: ignore[reportIncompatibleMethodOverride]
         self, project_name: str, sources: Union[list[_models.UpdateSourceRecord], IO[bytes]], **kwargs: Any
     ) -> AsyncLROPoller[None]:
         """Updates the sources of a project.
@@ -301,9 +304,7 @@ async def begin_update_sources( # pyright: ignore[reportIncompatibleMethodOverri
         :raises ~azure.core.exceptions.HttpResponseError:
         """
         # Call the parent implementation with 'body' parameter for backward compatibility
-        return await super().begin_update_sources(
-            project_name=project_name, body=sources, **kwargs
-        )
+        return await super().begin_update_sources(project_name=project_name, body=sources, **kwargs)
 
 
 __all__: list[str] = [
@@ -318,3 +319,28 @@ def patch_sdk():
     you can't accomplish using the techniques described in
     https://aka.ms/azsdk/python/dpcodegen/python/customize
     """
+    # Hide the generated async get_*_status methods (status peek endpoints) without editing
+    # the generated _operations.py file. Same approach as sync variant.
+    original_methods: dict[str, Any] = {}
+    to_hide = [
+        "get_delete_status",
+        "get_export_status",
+        "get_import_status",
+        "get_deploy_status",
+    ]
+    generated_cls = _QuestionAnsweringAuthoringClientOperationsMixinGenerated
+    for name in to_hide:
+        if hasattr(generated_cls, name):
+            original_methods[name] = getattr(generated_cls, name)
+            try:
+                delattr(generated_cls, name)
+            except AttributeError:
+                pass
+
+    patch_cls = _QuestionAnsweringAuthoringClientOperationsMixin
+    for public_name, func in original_methods.items():
+        private_name = f"_{public_name}"  # e.g. _get_delete_status
+        if not hasattr(patch_cls, private_name):
+            setattr(patch_cls, private_name, func)
+
+    # Do not export these private replicas; __all__ only includes the mixin class currently.

sdk/cognitivelanguage/azure-ai-language-questionanswering-authoring/pyproject.toml

@@ -34,6 +34,7 @@ dependencies = [
     "isodate>=0.6.1",
     "azure-core>=1.35.0",
     "typing-extensions>=4.6.0",
+    "azure-ai-language-questionanswering>=2.0.0"
 ]
 dynamic = [
 "version", "readme"