SD_Hackathon / src /google_drive.py
com3dian's picture
Create google_drive.py
6612d28 verified
raw
history blame
754 Bytes
from googleapiclient.discovery import build
from googleapiclient.http import MediaFileUpload
from google.oauth2 import service_account
import json
import os
def upload_to_drive(local_path, filename):
creds_dict = json.loads(os.environ["GOOGLE_CREDS_JSON"])
creds = service_account.Credentials.from_service_account_info(creds_dict)
service = build("drive", "v3", credentials=creds)
file_metadata = {
"name": filename,
# "parents": ["your-folder-id"] # optional: target folder
}
media = MediaFileUpload(local_path, mimetype="application/zip")
uploaded_file = service.files().create(
body=file_metadata,
media_body=media,
fields="id"
).execute()
return uploaded_file.get("id")