83 lines
2.4 KiB
Python
83 lines
2.4 KiB
Python
# Project: Telegram PC kill switch bot
|
|
# Author: Stanislav Mykhailenko
|
|
# License: Unlicense
|
|
|
|
# This file contains Google Drive interactions
|
|
|
|
|
|
from __future__ import print_function
|
|
|
|
import os
|
|
from mimetypes import MimeTypes
|
|
from google.auth.transport.requests import Request
|
|
from google.oauth2.credentials import Credentials
|
|
from google_auth_oauthlib.flow import InstalledAppFlow
|
|
from googleapiclient.discovery import build
|
|
from googleapiclient.errors import HttpError
|
|
from googleapiclient.http import MediaFileUpload
|
|
|
|
|
|
creds = None
|
|
|
|
def authenticate():
|
|
global creds
|
|
# The file token.json stores the user's access and refresh tokens, and is
|
|
# created automatically when the authorization flow completes for the first
|
|
# time.
|
|
if os.path.exists(os.path.join(os.path.dirname(__file__), 'token.json')):
|
|
creds = Credentials.from_authorized_user_file(os.path.join(os.path.dirname(__file__), 'token.json'), ['https://www.googleapis.com/auth/drive'])
|
|
# If there are no (valid) credentials available, let the user log in.
|
|
if not creds or not creds.valid:
|
|
if creds and creds.expired and creds.refresh_token:
|
|
creds.refresh(Request())
|
|
else:
|
|
flow = InstalledAppFlow.from_client_secrets_file(
|
|
os.path.join(os.path.dirname(__file__), 'credentials.json'), ['https://www.googleapis.com/auth/drive'])
|
|
creds = flow.run_local_server(port=0)
|
|
# Save the credentials for the next run
|
|
with open(os.path.join(os.path.dirname(__file__), 'token.json'), 'w') as token:
|
|
token.write(creds.to_json())
|
|
|
|
|
|
|
|
def createFolder(name, parent):
|
|
try:
|
|
service = build('drive', 'v3', credentials=creds)
|
|
file_metadata = {
|
|
'name': [name],
|
|
'mimeType': 'application/vnd.google-apps.folder',
|
|
'parents': [parent]
|
|
}
|
|
|
|
file = service.files().create(body=file_metadata, fields='id'
|
|
).execute()
|
|
return file.get('id')
|
|
|
|
except HttpError as error:
|
|
print(F'An error occurred: {error}')
|
|
return None
|
|
|
|
|
|
def uploadFile(path, folder):
|
|
try:
|
|
service = build('drive', 'v3', credentials=creds)
|
|
|
|
name = os.path.basename(os.path.normpath(path))
|
|
|
|
file_metadata = {
|
|
'name': [name],
|
|
'parents': [folder]
|
|
}
|
|
|
|
mimetype = MimeTypes().guess_type(path)[0]
|
|
|
|
media = MediaFileUpload(path,
|
|
mimetype=mimetype)
|
|
|
|
file = service.files().create(body=file_metadata, media_body=media,
|
|
fields='id').execute()
|
|
return file.get('id')
|
|
|
|
except HttpError as error:
|
|
print(F'An error occurred: {error}')
|
|
file = None
|