python3でGoogleDriveからファイルの取得

ホーム
BLOG
技術ブログ
python3でGoogleDriveからファイルの取得…

技術ブログ
2020.04.25

python3でGoogleDriveからファイルの取得

GoogleDriveからファイルの取得を定期的にして処理をする必要があったのでその備忘録で。

OAuth 2.0を使った例は結構あったのですが、認証の有効期限があったりなのでDrive v3 APIを使って実施です。

まずこちらから「サービスアカウント」を登録する必要があります。

ロールはプロジェクトの編集者などを選び設定します。

サービスアカウントキーを後ほど使うのでJSONでダウンロードしておく必要があります。

こちらのライブラリからGoogleDriveを有効にしておく必要もあります。

ライブラリのインストール

python3を使ってなので、pip3を使ってインストールします。


# pip3 install --upgrade google-api-python-client google-auth-httplib2 google-auth-oauthlib

# pip3 install --upgrade google-api-python-client google-auth-httplib2 google-auth-oauthlib

ダウンロードプログラム

まず、「service_account_key.json」という名前などでgoogle developer consoleから取得したJSONを保存します。


{
  "type": "service_account",
  "project_id": "***",
  "private_key_id": "***",
  "private_key": "-----BEGIN PRIVATE KEY-----***-----END PRIVATE KEY-----\n",
  "client_email": "***@***.iam.gserviceaccount.com",
  "client_id": "***",
  "auth_uri": "https://accounts.google.com/o/oauth2/auth",
  "token_uri": "https://oauth2.googleapis.com/token",
  "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
  "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/***%***.iam.gserviceaccount.com"
}

{

"type": "service_account",

"project_id": "***",

"private_key_id": "***",

"private_key": "-----BEGIN PRIVATE KEY-----***-----END PRIVATE KEY-----\n",

"client_email": "***@***.iam.gserviceaccount.com",

"client_id": "***",

"auth_uri": "https://accounts.google.com/o/oauth2/auth",

"token_uri": "https://oauth2.googleapis.com/token",

"auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",

"client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/***%***.iam.gserviceaccount.com"

}

以下がダウンロードを実施するプログラムになります。


from apiclient.discovery import build
from oauth2client.service_account import ServiceAccountCredentials
from httplib2 import Http
from googleapiclient.http import MediaIoBaseDownload
import io


SCOPES = ['https://www.googleapis.com/auth/drive']


credentials = ServiceAccountCredentials.from_json_keyfile_name(
    'service_account_key.json', SCOPES
)
http_auth = credentials.authorize(Http())

drive_service = build('drive', 'v3', http=http_auth)


results = drive_service.files().list(
    pageSize=10, fields="nextPageToken, files(id, name)").execute()
items = results.get('files', [])

if not items:
    print('No files found.')
else:
    for item in items:
        ## 拡張子がある時（ファイルの時）
        if "." in item['name']:
            print(u'{0} ({1})'.format(item['name'], item['id']))

            request = drive_service.files().get_media(fileId=item['id'])
            fh = io.FileIO(item['name'], "wb")
            downloader = MediaIoBaseDownload(fh, request)
            done = False
            while done is False:
                status, done = downloader.next_chunk()