はじめに

FirebaseのCloud FirestoreとCloud StorageについてPythonから使ったときの方法についてメモを残しておきます。
CSVファイルを用意しておいてpythonからアップロードを行ったり、ダウンロードを行ったりなどの際にご利用ください。

目次

準備
- SDKの追加
- 認証ファイルの用意
Cloud Firestoreを使う
- ライブラリのimport
- 認証情報の初期化
- データを取得
- データを追加
- データを削除
Cloud Storageを使う
- ライブラリのimport
- 認証情報の初期化
- データを取得
- データを追加
- データを削除
まとめ

準備

まず、SDKと認証ファイルを用意します。

SDKの追加

下記のコマンドを実行して、SDKを追加します。

pip install firebase-admin # firebase firestore
pip install --upgrade google-cloud-storage # cloud storage

認証ファイルの用意

プログラムからfirebaseにアクセスするために使用する認証ファイルです。
以下の手順でjsonファイルを取得します。

https://firebase.google.com/docs/admin/setup?hl=ja より

Firebase コンソールで、[設定] > [サービスアカウント] を開きます。

[新しい秘密鍵の生成] をクリックし、[キーを生成] をクリックして確定します。
キーを含む JSON ファイルを安全に保管します。

Cloud Firestoreを使う

pythonからfirestoreを使う方法についてまとめます。

ライブラリのimport

まず、使用するライブラリ類をimportします。

import

import firebase_admin
from firebase_admin import credentials
from firebase_admin import firestore

認証情報の初期化

先ほど用意したjsonの認証ファイルを用いてプログラムからfirebaseにアクセスできるようにします。
以下のコードで初期化を行います。

initialize

cred = credentials.Certificate('<pass_to_json_file>')
default_app = firebase_admin.initialize_app(cred)

<pass_to_json_file>の部分はjsonファイルへのパス(文字列)に置き換えてください。
初期化はfirebaseのデータにアクセスする前に必ず行いましょう。
- 初期化を行わないとエラーが出てデータを取得できません。
以下のようにcredとdefault_appをグローバル変数として確保しておき、初期化関数を用意することで少し便利になるかもしれません。

initialize2

cred = None
default_app = None

def initializeFirebaseAPP():
	global cred
	global default_app
	cred = credentials.Certificate("< pass_to_json_file >")
	default_app = firebase_admin.initialize_app(cred)

自分は利用しなかったのですが、環境変数を利用してjsonファイルへのパスを通すこともできます。

(ターミナルや端末で)環境変数を通します。
この場合の初期化は引数を指定せずに行います。

export

export GOOGLE_APPLICATION_CREDENTIALS="<pass_to_json_file>"

initialize

default_app = firebase_admin.initialize_app()

Firestoreからデータを取得

単一のドキュメントを指定して、内容を取得する例です。
getを使います。
このとき、doc_refの指すdocumentが存在しないと例外が投げられます。
- この例外はgoogle.cloud.exceptions.NotFoundなので、必要に応じてtryで対処しましょう
- doc_refに対して、existsメソッドを利用することで、documentが存在するかを確認できます。

getFirestoreData

def getFirestoreData(collectionName, documentName):
	db = firestore.client()
	doc_ref = db.collection(collectionName).document(documentName)

	doc = doc_ref.get()
	print(u'Document data: {}'.format(doc.to_dict()))

コレクション直下のドキュメントを全て取得するにはcollectionに対してstreamを使います。

getFirestoreData2

def getFirestoreData2(collectionName):
	db = firestore.client()
	docs = db.collection(collectionName).stream()

	for doc in docs:
		print(u'{} => {}'.format(doc.id, doc.to_dict()))

Firestoreからデータを取得(再帰的利用)

ドキュメント内のコレクションにアクセスするには、documentに対してcollectionメソッドを利用します。
直感通りの操作でできますね。

getFirestoreData3

def getFirestoreData3(col_name, doc_name, col_name2):
	db = firestore.client()
	doc_ref = db.collection(col_name).document(doc_name)
	col_ref2 = doc_ref.collection(col_name2)

現在のところ、pythonではドキュメントのサブコレクション一覧を取得することはできません。
もし、サブコレクション一覧を取得したいのであれば他の言語を併用しましょう。
例えば、rubyで書くのであれば以下の通りです。

getCollectionList.rb

require "google/cloud/firestore"

firestore = Google::Cloud::Firestore.new project_id: "<project_id>", keyfile: "<path_to_json>"
doc_ref = firestore.doc "<col_name>/<doc_name>"

doc_ref.cols do |col|
	puts col.collection_id
end

Firestoreへデータを追加

コレクション名とドキュメント名を指定して辞書型データを登録する例です。

uploadData2Firestore

def uploadData2Firestore(collectionName, documentName, data):
	db = firestore.client()
	doc_ref = db.collection(collectionName).document(documentName)
	doc_ref.set(data)

Firestoreのデータを更新

すでにあるドキュメント上のフィールドを更新する方法です。

update_data

def update_data(collectionName, documentName, key, value):
    db = firestore.client()
    doc_ref = db.collection(collectionName).document(documentName)

    doc_ref.update({key: value})

Firestoreからデータを削除

コレクション名とドキュメント名を指定してデータを削除する例です。

deleteFirestoreDocument

def deleteFirestoreDocument(collectionName, documentName):
	db = firestore.client()
	db.collection(collectionName).document(documentName).delete()

Cloud Storageを使う

ライブラリのimport

まず、使用するライブラリ類をimportします。

import

import firebase_admin
from firebase_admin import credentials
from google.cloud import storage

認証情報の初期化

先ほど用意したjsonの認証ファイルを用いてプログラムからfirebaseにアクセスできるようにします。
以下のコードで初期化を行います。

initialize

storage_client = storage.Client.from_service_account_json('<pass_to_json_file>')

<pass_to_json_file>の部分はjsonファイルへのパスに置き換えてください。
ここで初期化した変数はこの後利用していきます。

自分は利用しなかったのですが、環境変数を利用してjsonファイルへのパスを通すこともできます。

(ターミナルや端末で)環境変数を通します。
この場合の初期化は引数を指定せずに行います。

export

export GOOGLE_APPLICATION_CREDENTIALS="<pass_to_json_file>"

initialize

storage_client = storage.Client()

Cloud Storageからデータを取得

blobのurlを指定して、データをファイルにダウンロードする例です。

download_blob

def download_blob(src_blob_url, dst_file):
	bucket_name = '<bucket_name>'
	bucket = storage_client.get_bucket(bucket_name)

	blob = bucket.blob(src_blob_url)
	storage_client.download_blob_to_file(blob, dst_file)

	print('Blob {} downloaded to {}.'.format(src_blob_url, dst_file))

Cloud Storageへデータを追加

ファイル名とデータを指定して登録する例です。

upload_blob

def upload_blob(file_name, data_sorce):
	bucket_name = '<bucket_name>'
	bucket = storage_client.get_bucket(bucket_name)
	blob = bucket.blob(file_name)
	blob.upload_from_filename(filename=data_sorce)

	if file_name in bucket.list_blobs():
		print("cloud storage: upload success")
	else:
		print("cloud storage: upload failed")

Cloud Storageからデータを削除

blobのurlを指定してデータを削除する例です。

delete_blob

def delete_blob(blob_url):
	bucket_name = '<bucket_name>'
	bucket = storage_client.get_bucket(bucket_name)
	blob = bucket.blob(blob_url)

	blob.delete()

まとめ

環境変数を利用しないやり方があまり情報としてなかったのでまとめてみました。
実際に運用する上では環境変数を利用した方が安全なのかもしれませんが...
以下、よくみたサイトです。
誤植等ありましたらご連絡いただければ幸いです。