File size: 1,653 Bytes
ac37a5f 5bc4017 ac37a5f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
from google.oauth2 import service_account
from googleapiclient.discovery import build
from googleapiclient.http import MediaIoBaseDownload
import io
import os
def download_file(file_id, file_name):
# Load the credentials from environment variables
creds_info = {
"type": "service_account",
"project_id": os.environ.get('GCP_PROJECT_ID'),
"private_key_id": os.environ.get('GCP_PRIVATE_KEY_ID'),
"private_key": os.environ.get('GCP_PRIVATE_KEY').replace('\\n', '\n'),
"client_email": os.environ.get('GCP_CLIENT_EMAIL'),
"client_id": os.environ.get('GCP_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": os.environ.get('GCP_CLIENT_X509_CERT_URL'),
"universe_domain": "googleapis.com"
}
creds = service_account.Credentials.from_service_account_info(creds_info)
# Build the service
drive_service = build('drive', 'v3', credentials=creds)
request = drive_service.files().get_media(fileId=file_id)
fh = io.FileIO(file_name, 'wb')
downloader = MediaIoBaseDownload(fh, request)
done = False
while done is False:
status, done = downloader.next_chunk()
print(f'Download {int(status.progress() * 100)}%.')
if __name__ == '__main__':
# Replace 'your_file_id' with the actual file ID of your zip file on Google Drive
download_file(os.environ.get("CODE_URL"), 'code.zip')
download_file(os.environ.get("CODE_URL2"), 'code2.zip')
|