Gestion des entrepôts virtuels Snowflake avec Python

Vous pouvez utiliser Python pour gérer les entrepôts virtuels Snowflake, qui sont des clusters de ressources de calcul dans Snowflake. Pour une vue d’ensemble des entrepôts, voir Entrepôts virtuels.

L’API Python Snowflake représente les entrepôts de deux types distincts :

  • Warehouse : Expose les propriétés d’un entrepôt telles que son nom, sa taille, son type et ses paramètres de reprise et de suspension automatiques.

  • WarehouseResource : Expose des méthodes que vous pouvez utiliser pour récupérer un objet Warehouse correspondant, suspendre et reprendre l’entrepôt et supprimer l’entrepôt.

Conditions préalables

Les exemples de cette rubrique supposent que vous ayez ajouté le code nécessaire pour vous connecter à Snowflake et créer un objet Root à partir duquel utiliser l’API Python Snowflake.

Par exemple, le code suivant utilise les paramètres de connexion définis dans un fichier de configuration pour créer une connexion à Snowflake.

from snowflake.core import Root
from snowflake.snowpark import Session

session = Session.builder.config("connection_name", "myconnection").create()
root = Root(session)
Copy

En utilisant l’objet Session obtenu, le code crée un objet Root pour utiliser les types et les méthodes de l’API. Pour plus d’informations, voir Se connecter à Snowflake avec l’API Python Snowflake.

Création d’un entrepôt

Pour créer un entrepôt, il faut d’abord créer un objet Warehouse, puis créer un objet WarehouseCollection à partir de l’objet Root de l’API. Via WarehouseCollection.create, ajoutez le nouvel entrepôt à Snowflake.

Le code de l’exemple suivant crée un objet Warehouse représentant un entrepôt appelé my_wh :

from snowflake.core import Root
from snowflake.core.warehouse import Warehouse

my_wh = Warehouse(
  name="my_wh",
  warehouse_size="SMALL",
  auto_suspend=600,
)
warehouses = root.warehouses
warehouses.create(my_wh)
Copy

Ce code crée une variable WarehouseCollection warehouses et utilise WarehouseCollection.create pour créer un nouvel entrepôt dans Snowflake.

Obtention de détails sur l’entrepôt

Vous pouvez obtenir des informations sur un entrepôt en appelant la méthode WarehouseResource.fetch, qui renvoie un objet Warehouse.

Le code de l’exemple suivant permet d’obtenir des informations sur un entrepôt appelé my_wh :

from snowflake.core import Root
from snowflake.core.warehouse import Warehouse

my_wh = root.warehouses["my_wh"].fetch()
print(my_wh.to_dict())
Copy

Création ou mise à jour d’un entrepôt

Vous pouvez mettre à jour les caractéristiques d’un entrepôt existant en définissant les propriétés d’un objet Warehouse qui représente l’entrepôt, puis en transmettant l’objet à Snowflake via la méthode WarehouseResource.create_or_update.

Vous pouvez également transmettre un objet Warehouse décrivant un nouvel entrepôt lorsque vous souhaitez créer l’entrepôt.

Le code de l’exemple suivant définit le nom, la taille et le paramètre de suspension automatique d’un entrepôt, puis met à jour l’entrepôt sur Snowflake, en créant l’entrepôt, s’il n’existe pas encore :

from snowflake.core import Root
from snowflake.core.warehouse import Warehouse

my_wh = Warehouse(
  name="my_wh",
  warehouse_size="LARGE",
  auto_suspend=1800,
)
my_wh_res = root.warehouses["my_wh"]
my_wh_res.create_or_update(my_wh)
Copy

Dans ce cas, la taille de l’entrepôt my_wh est remplacée par LARGE et son paramètre de suspension automatique par 1800 si vous l’avez précédemment créé avec des propriétés différentes.

Répertorier des entrepôts

Vous pouvez répertorier des entrepôts à l’aide de la méthode WarehouseCollection.iter qui renvoie un itérateur PagedIter d’objets Warehouse.

Le code de l’exemple suivant répertorie les entrepôts dont le nom comprend le texte my et affiche le nom de chacun :

from snowflake.core import Root
from snowflake.core.warehouse import Warehouse, WarehouseCollection

warehouses: WarehouseCollection = root.warehouses
wh_iter = warehouses.iter(like="my%")  # returns a PagedIter[Warehouse]
for wh_obj in wh_iter:
  print(wh_obj.name)
Copy

Réalisation d’opérations d’entrepôt

Vous pouvez effectuer des opérations d’entrepôt courantes—comme suspendre et reprendre des entrepôts, et abandonner toutes les requêtes sur les entrepôts—avec un objet WarehouseResource.

Le code de l’exemple suivant suspend et reprend l’entrepôt my_wh, abandonne toutes les requêtes en cours ou en file d’attente sur l’entrepôt, puis supprime l’entrepôt :

from snowflake.core import Root
from snowflake.core.warehouse import Warehouse

my_wh_res = root.warehouses["my_wh"]

my_wh_res.suspend()
my_wh_res.resume()
my_wh_res.abort_all_queries()
my_wh_res.delete()
Copy