Pythonを使用したSnowpark Container Servicesの管理¶

コンピューティングプールの作成¶

作成したいコンピュートプールを表す ComputePool オブジェクトを渡して ComputePoolCollection.create メソッドを呼び出すことで、コンピュートプールを作成できます。

ステージングされた仕様からコンピュートプールを作成するには、まず、以下のようなプールプロパティを指定する ComputePool オブジェクトを作成します。

• コンピュートプール名

• プールを構成するノードの最大数と最小数

• プール内のノードにプロビジョニングするマシンのタイプを識別するインスタンスファミリーの名前

• サービスまたはジョブがプールに送信されたときに、プールを自動的に再開するかどうか。

次の例のコードでは、 my_compute_pool というプールを表す ComputePool オブジェクトを作成します。次に、コードは ComputePool オブジェクトを ComputePoolCollection.create メソッドに渡すことで、コンピュートプールを作成します。

from snowflake.core import Root
from snowflake.core.compute_pool import ComputePool

compute_pool = ComputePool("my_compute_pool", min_node=1, max_node=2, instance_family="STANDARD_1", auto_resume=False)
root.compute_pools.create(compute_pool)

コンピュートプール詳細の取得¶

ComputePool オブジェクトを返す ComputePoolResource.fetch メソッドを呼び出すことで、コンピュートプールに関する情報を取得できます。

次の例のコードは、 my_compute_pool というプールに関する情報を取得します。

from snowflake.core import Root
from snowflake.core.compute_pool import ComputePool

compute_pool = root.compute_pools["my_compute_pool"].fetch()

コンピュートプールの作成または更新¶

既存のコンピュートプールを表す ComputePool オブジェクトのプロパティを設定し、更新されたオブジェクトを ComputePoolResource オブジェクトを返す create_or_update メソッドで Snowflake に渡すことで、既存のコンピュートプールの特性を更新できます。

プールを作成するときに、新しいプールを記述する ComputePool オブジェクトを渡すこともできます。

次の例のコードは、 my_compute_pool コンピュートプールの最大許容ノードを設定し、Snowflake 上のプールを更新します。

from snowflake.core import Root
from snowflake.core.compute_pool import ComputePool

compute_pool = root.compute_pools["my_compute_pool"].fetch()
compute_pool.max_node = 3
compute_pool_res = root.compute_pools.create_or_update(compute_pool)

コンピュートプールの一覧表示¶

iter メソッドを使用してコンピュートプールを一覧表示することができます。メソッドは PagedIter 反復子を返します。

次の例のコードは、名前が「abc」で始まるコンピュートプールを一覧表示し、それぞれの名前を印刷します。

from snowflake.core import Root

compute_pools = root.compute_pools.iter(like="abc%")
for compute_pool in compute_pools:
  print(compute_pool.name)

コンピュートプール操作の実行¶

ComputePoolResource オブジェクトを使用して、プールの中断、再開、停止など一般的なコンピュートプールの操作を実行できます。 ComputePool.fetch メソッドを使用して ComputePoolResource オブジェクトを取得できます。

次の例のコードは、 my_compute_pool コンピュートプールを中断、再開、停止します。これは、 Root.compute_pools メソッドを使用して、コンピュートプールを表す ComputePool オブジェクトを作成します。 ComputePool オブジェクトから、 ComputePoolResource オブジェクトをフェッチし、コンピュートプール操作を実行します。

from snowflake.core import Root
from snowflake.core.compute_pool import ComputePool

compute_pool = root.compute_pools["my_compute_pool"]
compute_pool_res = compute_pool.fetch()
compute_pool_res.suspend()
compute_pool_res.resume()
compute_pool_res.stop_all_services()

イメージリポジトリの作成¶

ステージングされた仕様からイメージリポジトリを作成するには、まずリポジトリ名を指定する ImageRepository オブジェクトを作成します。

次の例のコードでは、 my_repo というリポジトリを表す ImageRepository オブジェクトを作成します。次に、コードは ImageRepository オブジェクトを ImageRepositoryCollection.create メソッドに渡してイメージリポジトリを作成することで、 my_db データベースと my_schema スキーマにイメージリポジトリを作成します。

from snowflake.core import Root
from snowflake.core.image_repository import ImageRepository

my_repo = ImageRepository("my_repo")
root.databases["my_db"].schemas["my_schema"].image_repositories.create(my_repo)

イメージリポジトリ詳細の取得¶

ImageRepository オブジェクトを返す ImageRepositoryResource.fetch メソッドを呼び出すことで、イメージリポジトリに関する情報を取得できます。

次の例のコードは、 my_repo イメージリポジトリを表す ImageRepository オブジェクトを取得し、リポジトリの所有者の名前を印刷します。

from snowflake.core import Root
from snowflake.core.image_repository import ImageRepository

my_repo_res = root.databases["my_db"].schemas["my_schema"].image_repositories["my_repo"]
my_repo = my_repo_res.fetch()
print(my_repo.owner)

イメージリポジトリの一覧表示¶

iter メソッドを使用して指定したスキーマのイメージリポジトリを一覧表示することができます。このメソッドは ImageRepository オブジェクトの PagedIter 反復子を返します。

次の例のコードは、 my_db データベースと my_schema スキーマのリポジトリを一覧表示し、各リポジトリの名前を印刷します。

from snowflake.core import Root

repo_list = root.databases["my_db"].schemas["my_schema"].image_repositories.iter()
for repo_obj in repo_list:
  print(repo_obj.name)

イメージリポジトリの削除¶

ImageRepositoryResource.delete メソッドを使用してイメージリポジトリを削除できます。

次の例のコードは、 my_repo リポジトリを削除します。

from snowflake.core import Root
from snowflake.core.image_repository import ImageRepository

my_repo_res = root.databases["my_db"].schemas["my_schema"].image_repositories["my_repo"]
my_repo_res.delete()

サービスの作成¶

サービスを作成するには、 services.create メソッドを実行し、作成したいサービスを表す Service オブジェクトを渡します。

ステージにアップロードされたサービス仕様 .yaml ファイルからサービスを作成します。サービス仕様の作成の詳細については、 サービス仕様リファレンス をご参照ください。

session.file.put("/local_location/my_service_spec.yaml", "@my_stage")

service_spec_string = """
// Specification as a string.
"""
session.file.put_stream(StringIO(sepc_in_string), "@my_stage/my_service_spec.yaml")

from snowflake.core import Root
from snowflake.core.service import Service

my_service = Service("my_service", min_instances=1, max_instances=2,compute_pool="my_compute_pool", from_stage="@my_stage", spec="my_service_spec.yaml")
root.databases["my_db"].schemas["my_schema"].services.create(my_service)

サービス詳細の取得¶

Service オブジェクトを返す ServiceResource.fetch メソッドを呼び出すことで、Snowflakeサービスに関する情報を取得できます。

次の例のコードは、 my_service というサービスに関する情報を取得します。

from snowflake.core import Root
from snowflake.core.service import Service

my_service = root.databases["my_db"].schemas["my_schema"].services["my_service"].fetch()

サービスの一覧表示¶

iter メソッドを使用して指定したスキーマのサービスを一覧表示することができます。このメソッドは Service オブジェクトの PagedIter 反復子を返します。

次の例のコードは、名前が「abc」で始まるサービスを一覧表示し、それぞれの名前を印刷します。

from snowflake.core import Root

services = root.databases["my_db"].schemas["my_schema"].services.iter(like="abc%")
for service_obj in services:
  print(service_obj.name)

サービス操作の実行¶

ServiceResource オブジェクトを使用して、サービスの中断、再開、そのステータスの取得など一般的なサービスの操作を実行できます。

次の例のコードは、 my_service サービスを中断および再開します。また、サービスのステータスも取得します。

from snowflake.core import Root
from snowflake.core.service import Service

my_service_res = root.databases["my_db"].schemas["my_schema"].services["my_service"]

my_service_res.suspend()
my_service_res.resume()
status = my_service_res.get_service_status(10)