Python でOkta のデータを変換・出力するETL 処理を作る方法

加藤龍彦
デジタルマーケティング

CData Python Connector とpetl モジュールを使って、Okta のデータを変換後にCSV ファイルに吐き出すETL 処理を実装します。

Pythonエコシステムには多くのモジュールがあり、システム構築を素早く効率的に行うことができます。本記事では、CData Python Connector for Okta とpetl フレームワークを使って、Okta のデータにPython から接続してデータを変換、CSV に出力するETL 変換を実装してみます。

CData Python Connector は効率的なデータ処理によりOkta のデータにPython から接続し、高いパフォーマンスを発揮します。Okta にデータをクエリする際、ドライバーはフィルタリング、集計などがサポートされている場合SQL 処理を直接Okta 側に行わせ、サポートされていないSQL 処理については、組み込みのSQL エンジンによりクライアント側で処理を行います（JOIN やSQL 関数など）。

必要なモジュールのインストール

pip で必要なモジュールおよびフレームワークをインストールします：

pip install petl
pip install pandas

Python でOkta のデータをETL 処理するアプリを構築

モジュールとフレームワークをインストールしたら、ETL アプリケーションを組んでいきます。コードのスニペットは以下の通りです。フルコードは記事の末尾に付いています。

CData Connector を含むモジュールをインポートします。

import petl as etl
import pandas as pd
import cdata.okta as mod

接続文字列で接続を確立します。connect 関数を使って、CData Okta Connector からOkta への接続を行います

cnxn = mod.connect("Domain=dev-44876464.okta.com;")

Okta に接続するには、Domain 接続文字列プロパティをお使いのOkta ドメインに設定します。 Okta はOAuth およびOAuthJWT 認証をサポートしています。

OAuth認証

OAuth で認証するには、AuthScheme をOAuth に設定する必要があります。また、カスタムOAuth アプリケーションを作成する必要があります。

カスタムOAuth アプリケーションの作成

Okta アカウントから：

管理者アカウントでOkta のデベロッパーエディションにサインインします。
Admin Consoleで、Applications > Applications に移動します。
Create App Integration をクリックします。
Sign-in method は、OIDC - OpenID Connect を選択します。
Application type は、Web Application を選択します。
カスタムアプリケーションの名前を入力します。
Grant TypeをAuthorization Code に設定します。トークンを自動的に更新したい場合は、Refresh Token もチェックしてください。
コールバックURL を設定します。
- デスクトップアプリケーションやヘッドレスマシンでは、http://localhost:33333 または任意の別のポート番号を使用します。ここで設定したURI が、CallbackURL プロパティになります。
- Web アプリケーションの場合、コールバックURL を信頼できるリダイレクトURL に設定します。このURL は、アプリケーションへのアクセスが許可されたことを示すトークンを伴ってユーザーが戻ってくるWeb 上の場所です。
Assignments セクションで、Limit access to selected groups を選択してグループを追加するか、グループの割り当ては一旦スキップします。
OAuth アプリケーションを保存します。
アプリケーションのGeneral タブに、アプリケーションのClient Id とClient Secret が表示されます。後で使用できるように、これらを記録してください。Client Id はOAuthClientId の設定に使用し、Client Secret はOAuthClientSecret の設定に使用します。
Assignments タブを確認し、アプリケーションにアクセスする必要のあるすべてのユーザーがアプリケーションに割り当てられていることを確かめます。
Okta API Scopes タブで、OAuth アプリケーションに付与するスコープを選択します。これらのスコープは、アプリが読み取り可能なデータを決定します。そのため、特定のビューに対するスコープを付与しないと、そのビューに対するクエリを実行する権限がドライバーに付与されません。各ビューに必要なスコープを確認するには、ヘルプドキュメントのデータモデル > ビュー のビュー固有のページを参照してください。

OAuth 認証の詳細や、OAuthJWT 認証については、ヘルプドキュメントを参照してください。

Okta をクエリするSQL 文の作成

Okta にはSQL でデータアクセスが可能です。Users エンティティからのデータを読み出します。

sql = "SELECT Id, ProfileFirstName FROM Users WHERE Status = 'Active'"

Okta データのETL 処理

DataFrame に格納されたクエリ結果を使って、petl でETL（抽出・変換・ロード）パイプラインを組みます。この例では、Okta のデータを取得して、ProfileFirstName カラムでデータをソートして、CSV ファイルにデータをロードします。

table1 = etl.fromdb(cnxn,sql)

table2 = etl.sort(table1,'ProfileFirstName')

etl.tocsv(table2,'users_data.csv')

CData Python Connector for Okta を使えば、データベースを扱う場合と同感覚で、Okta のデータを扱うことができ、petl のようなETL パッケージから直接データにアクセスが可能になります。

おわりに

Okta Python Connector の30日の無償トライアルをぜひダウンロードして、Okta のデータへの接続をPython アプリやスクリプトから簡単に作成しましょう。

フルソースコード

import petl as etl
import pandas as pd
import cdata.okta as mod

cnxn = mod.connect("Domain=dev-44876464.okta.com;")

sql = "SELECT Id, ProfileFirstName FROM Users WHERE Status = 'Active'"

table1 = etl.fromdb(cnxn,sql)

table2 = etl.sort(table1,'ProfileFirstName')

etl.tocsv(table2,'users_data.csv')