mirror of
https://github.com/anasty17/mirror-leech-telegram-bot.git
synced 2025-01-08 12:07:33 +08:00
26 lines
874 B
Python
26 lines
874 B
Python
import pickle
|
|
import os
|
|
from google_auth_oauthlib.flow import InstalledAppFlow
|
|
from google.auth.transport.requests import Request
|
|
|
|
credentials = None
|
|
__G_DRIVE_TOKEN_FILE = "token.pickle"
|
|
__OAUTH_SCOPE = ["https://www.googleapis.com/auth/drive"]
|
|
if os.path.exists(__G_DRIVE_TOKEN_FILE):
|
|
with open(__G_DRIVE_TOKEN_FILE, 'rb') as f:
|
|
credentials = pickle.load(f)
|
|
if (
|
|
(credentials is None or not credentials.valid)
|
|
and credentials
|
|
and credentials.expired
|
|
and credentials.refresh_token
|
|
):
|
|
credentials.refresh(Request())
|
|
else:
|
|
flow = InstalledAppFlow.from_client_secrets_file(
|
|
'credentials.json', __OAUTH_SCOPE)
|
|
credentials = flow.run_console(port=0)
|
|
|
|
# Save the credentials for the next run
|
|
with open(__G_DRIVE_TOKEN_FILE, 'wb') as token:
|
|
pickle.dump(credentials, token) |