|
import pickle |
|
import os |
|
import datetime |
|
from google_auth_oauthlib.flow import Flow, InstalledAppFlow |
|
from googleapiclient.discovery import build |
|
from googleapiclient.http import MediaFileUpload, MediaIoBaseDownload |
|
from google.auth.transport.requests import Request |
|
|
|
|
|
def token_file_exists(prefix="token_"): |
|
files = os.listdir() |
|
for file in files: |
|
if file.startswith(prefix): |
|
return True |
|
|
|
return False |
|
|
|
def Create_Service(client_secret_file, api_name, api_version, *scopes): |
|
CLIENT_SECRET_FILE = client_secret_file |
|
API_SERVICE_NAME = api_name |
|
API_VERSION = api_version |
|
SCOPES = [scope for scope in scopes[0]] |
|
|
|
cred = None |
|
pickle_file = f'token_{API_SERVICE_NAME}_{API_VERSION}.pickle' |
|
|
|
if os.path.exists(pickle_file): |
|
with open(pickle_file, 'rb') as token: |
|
cred = pickle.load(token) |
|
|
|
if not cred or not cred.valid: |
|
if cred and cred.expired and cred.refresh_token: |
|
cred.refresh(Request()) |
|
else: |
|
flow = InstalledAppFlow.from_client_secrets_file(CLIENT_SECRET_FILE, SCOPES) |
|
cred = flow.run_local_server() |
|
|
|
with open(pickle_file, 'wb') as token: |
|
pickle.dump(cred, token) |
|
|
|
try: |
|
service = build(API_SERVICE_NAME, API_VERSION, credentials=cred) |
|
return service |
|
except Exception as e: |
|
print('Unable to connect.') |
|
print(e) |
|
return None |
|
|
|
def convert_to_RFC_datetime(year=1900, month=1, day=1, hour=0, minute=0): |
|
dt = datetime.datetime(year, month, day, hour, minute, 0).isoformat() + 'Z' |
|
return dt |
|
|
|
import os |
|
def removeAccount(): |
|
directory = os.getcwd() |
|
files = os.listdir(directory) |
|
for file in files: |
|
if "token" in file: |
|
try: |
|
os.remove(os.path.join(directory, file)) |
|
print(f"File '{file}' deleted successfully.") |
|
except Exception as e: |
|
print(f"Error deleting file '{file}': {e}") |
|
|