本文整理汇总了Python中airflow.contrib.hooks.azure_cosmos_hook.AzureCosmosDBHook.does_database_exist方法的典型用法代码示例。如果您正苦于以下问题:Python AzureCosmosDBHook.does_database_exist方法的具体用法?Python AzureCosmosDBHook.does_database_exist怎么用?Python AzureCosmosDBHook.does_database_exist使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类airflow.contrib.hooks.azure_cosmos_hook.AzureCosmosDBHook
的用法示例。
在下文中一共展示了AzureCosmosDBHook.does_database_exist方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: execute
# 需要导入模块: from airflow.contrib.hooks.azure_cosmos_hook import AzureCosmosDBHook [as 别名]
# 或者: from airflow.contrib.hooks.azure_cosmos_hook.AzureCosmosDBHook import does_database_exist [as 别名]
def execute(self, context):
# 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)
# 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)
# finally insert the document
hook.upsert_document(self.document, self.database_name, self.collection_name)