potiuk commented on a change in pull request #19431:
URL: https://github.com/apache/airflow/pull/19431#discussion_r743918092
##########
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:
```suggestion
from airflow.providers.microsoft.azure.operators.cosmos import
AzureCosmosInsertDocumentOperator # noqa
```
--
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-unsubscr...@airflow.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
