things kinda work

This commit is contained in:
Pablo Martin 2024-08-13 15:02:03 +02:00
parent c38ce5cfe6
commit 7eb697fecd
10 changed files with 342 additions and 29 deletions

View file

@ -1,4 +1,5 @@
import azure.cosmos.cosmos_client as cosmos_client
from azure.cosmos.container import ContainerProxy
from anaxi.config import CosmosDBDatabaseConfig
@ -10,3 +11,13 @@ def create_cosmos_client_from_config(
url=config.host,
credential={"masterKey": config.master_key},
)
def get_container_client(
config: CosmosDBDatabaseConfig, container_name: str
) -> ContainerProxy:
cosmos_client = create_cosmos_client_from_config(config)
database_client = cosmos_client.get_database_client(database=config.database_id)
container_client = database_client.get_container_client(container=container_name)
return container_client