PowerShell からSnowflake のデータに接続してデータの取得・更新・挿入・削除・CSV エクスポートを実行する方法

こんにちは！ウェブ担当の加藤です。マーケ関連のデータ分析や整備もやっています。

CData Cmdlets for Snowflake を使えば、PowerShell からSnowflake に手軽に連携して、データのCRUD やエクスポートを実行できます。

本記事では、Snowflake への接続方法からCSV エクスポート、データの操作までサンプルコード付きで解説していきます。

Snowflake への接続を設定

それでは、まずはSnowflake への接続設定からはじめていきましょう。接続設定にはCData Snowflake Cmdlets が必要となります。右側のサイドバーから製品の全機能が使える30日間の無償トライアルがダウンロードできるので、ぜひご利用ください。

インストールが完了したら、プロファイルに以下の行を追加してください。次のPowerShell セッションでモジュールがロードされます。

    Import-Module SnowflakeCmdlets;

Connect-Snowflake コマンドを使ってSnowflake との接続を設定します。各接続プロパティの取得方法は次に説明します。

    $conn = Connect-Snowflake  -User "$User" -Password "$Password" -Server "$Server" -Database "$Database" -Warehouse "$Warehouse" -Account "$Account"

Snowflake への接続には以下の情報が必要となります。

1. User およびPassword をSnowflake ユーザーに設定し、AuthScheme をPASSWORD もしくはOKTA に設定します。
2. URL をSnowflake インスタンスのURL に設定します(i.e.: https://myaccount.snowflakecomputing.com)。
3. Warehouse をSnowflake warehouse に設定します。
4. (Optional) Account URL が上記の形式に当てはまらない場合、Snowflake のアカウントに設定します。
5. (Optional) Database およびSchema は、参照できるテーブルやビューを制限する場合に指定します。

詳細はヘルプドキュメントを参照してください。

これで接続設定は完了です。

Snowflake のデータを取得してCSV にパイプライン

接続が完了したので、Products テーブルデータを取得して結果をCSV ファイルにエクスポートします。

Select-Snowflake -Connection $conn -Table Products | Select -Property * -ExcludeProperty Connection,Table,Columns | Export-Csv -Path c:\myProductsData.csv -NoTypeInformation

このコードでは、Select-Snowflake から取得した結果をSelect-Object に流して、Export-Csv に渡す前にいくつかのプロパティを除外しています。これは、CData Cmdlets が接続情報、テーブル、およびカラム情報を結果セットのそれぞれの行に挿入するためです。それらの情報を表示したくない場合に、Export-Csv コマンドに渡す前に除外を先に行い、そのあとでCSV ファイルにエクスポートします。

それでは、続いてデータの削除や挿入・更新を実行してみます。

データの削除

以下のように、フィルタリングで合致するレコードを削除することができます。

Select-Snowflake -Connection $conn -Table Products -Where "Id = 1" | Remove-Snowflake

データの挿入と更新

データの挿入や更新を使って、加工やクレンジングも行えます。以下の手順では、CSV ファイルのデータを読み込んで挿入対象のオブジェクトに同じレコードが存在するかを確認した上で、存在する場合にはデータを更新、存在しない場合にはデータの挿入を行います。

Import-Csv -Path C:\MyProductsUpdates.csv | %{
  $record = Select-Snowflake -Connection $Snowflake -Table Products -Where ("Id = `'"+$_.Id+"`'")
  if($record){
    Update-Snowflake -Connection $snowflake -Table Products -Columns ("Id","ProductName") -Values ($_.Id, $_.ProductName) -Where ("Id = `'"+$_.Id+"`'")
  }else{
    Add-Snowflake -Connection $snowflake -Table Products -Columns ("Id","ProductName") -Values ($_.Id, $_.ProductName)
  }
}

おわりに

このように、CData Cmdlets を使えばPowerShell でのSnowflake のデータへの連携をシンプルに実現できます。ぜひCData PowerShell Cmdlets の30日間無償トライアルをダウンロードして、シンプルかつパワフルなデータ連携をお試しください。