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.
##########
File path: airflow/contrib/hooks/azure_container_registry_hook.py
##########
@@ -15,20 +15,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 `airflow.providers.microsoft.azure.hooks.azure_container_registry`.
-"""
+"""This module is deprecated. Please use
`airflow.providers.microsoft.azure.hooks.container_registry`."""
import warnings
-from airflow.providers.microsoft.azure.hooks.azure_container_registry import (
# noqa
- AzureContainerRegistryHook,
-)
+from airflow.providers.microsoft.azure.hooks.container_registry import
AzureContainerRegistryHook # noqa
warnings.warn(
- "This module is deprecated. "
- "Please use
`airflow.providers.microsoft.azure.hooks.azure_container_registry`.",
+ "This module is deprecated. " "Please use
`airflow.providers.microsoft.azure.hooks.container_registry`.",
Review comment:
Nice catch, thanks!
--
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: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
