[GitHub] [airflow] josh-fell commented on a change in pull request #19431: Update Azure modules to comply with AIP-21

[GitBox <gi...@apache.org>]

josh-fell commented on a change in pull request #19431:
URL: https://github.com/apache/airflow/pull/19431#discussion_r743918847



##########
File path: airflow/providers/microsoft/azure/operators/azure_cosmos.py
##########
@@ -15,56 +14,14 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+"""This module is deprecated. Please use :mod:`airflow.providers.microsoft.azure.operators.cosmos`."""
 
-from airflow.models import BaseOperator
-from airflow.providers.microsoft.azure.hooks.azure_cosmos import AzureCosmosDBHook
-
-
-class AzureCosmosInsertDocumentOperator(BaseOperator):
-    """
-    Inserts a new document into the specified Cosmos database and collection
-    It will create both the database and collection if they do not already exist
-
-    :param database_name: The name of the database. (templated)
-    :type database_name: str
-    :param collection_name: The name of the collection. (templated)
-    :type collection_name: str
-    :param document: The document to insert
-    :type document: dict
-    :param azure_cosmos_conn_id: Reference to the
-        :ref:`Azure CosmosDB connection<howto/connection:azure_cosmos>`.
-    :type azure_cosmos_conn_id: str
-    """
-
-    template_fields = ('database_name', 'collection_name')
-    ui_color = '#e4f0e8'
-
-    def __init__(
-        self,
-        *,
-        database_name: str,
-        collection_name: str,
-        document: dict,
-        azure_cosmos_conn_id: str = 'azure_cosmos_default',
-        **kwargs,
-    ) -> None:
-        super().__init__(**kwargs)
-        self.database_name = database_name
-        self.collection_name = collection_name
-        self.document = document
-        self.azure_cosmos_conn_id = azure_cosmos_conn_id
-
-    def execute(self, context: dict) -> None:
-        # Create the hook
-        hook = AzureCosmosDBHook(azure_cosmos_conn_id=self.azure_cosmos_conn_id)
-
-        # Create the DB if it doesn't already exist
-        if not hook.does_database_exist(self.database_name):
-            hook.create_database(self.database_name)
+import warnings
 
-        # Create the collection as well
-        if not hook.does_collection_exist(self.collection_name, self.database_name):
-            hook.create_collection(self.collection_name, self.database_name)
+from airflow.providers.microsoft.azure.operators.cosmo import AzureCosmosInsertDocumentOperator  # noqa

Review comment:
       Yeah just saw that test fail too. 




-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@airflow.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org