async def create_app(gdrive):
""" - Create google drive service app - """
creds = helper.get_credentials(str(gdrive.from_id))
if creds is not None:
""" - Repack credential objects from strings - """
creds = pickle.loads(base64.b64decode(creds.encode()))
if not creds or not creds.valid:
if creds and creds.expired and creds.refresh_token:
await gdrive.edit("`Refreshing credentials...`")
""" - Refresh credentials - """
creds.refresh(Request())
try:
helper.save_credentials(
str(gdrive.from_id),
base64.b64encode(pickle.dumps(creds)).decode())
except Exception:
for name in ['creds', 'gdrive']:
delete_table(name)
helper.save_credentials(
str(gdrive.from_id),
base64.b64encode(pickle.dumps(creds)).decode())
else:
await gdrive.edit("`Credentials is empty, please generate it...`")
return False
service = build('drive', 'v3', credentials=creds, cache_discovery=False)
return service
async def generate_credentials(gdrive):
""" - Only generate once for long run - """
if helper.get_credentials(str(gdrive.from_id)) is not None:
await gdrive.edit("`You already authorized token...`")
await asyncio.sleep(1.5)
return await gdrive.delete()
""" - Generate credentials - """
if G_DRIVE_DATA is not None:
configs = json.loads(G_DRIVE_DATA)
else:
""" - Only for old user - """
configs = {
"installed": {
"client_id": G_DRIVE_CLIENT_ID,
"client_secret": G_DRIVE_CLIENT_SECRET,
"auth_uri": GOOGLE_AUTH_URI,
"token_uri": GOOGLE_TOKEN_URI,
}
}
await gdrive.edit("`Creating credentials...`")
flow = InstalledAppFlow.from_client_config(
configs, SCOPES, redirect_uri=REDIRECT_URI)
auth_url, _ = flow.authorization_url(
access_type='offline', prompt='consent')
msg = await gdrive.respond(
"`Go to your BOTLOG group to authenticate token...`"
)
async with gdrive.client.conversation(BOTLOG_CHATID) as conv:
url_msg = await conv.send_message(
"Please go to this URL:\n"
f"{auth_url}\nauthorize then reply the code"
)
r = conv.wait_event(
events.NewMessage(outgoing=True, chats=BOTLOG_CHATID))
r = await r
code = r.message.message.strip()
flow.fetch_token(code=code)
creds = flow.credentials
await asyncio.sleep(3.5)
await gdrive.client.delete_messages(gdrive.chat_id, msg.id)
await gdrive.client.delete_messages(BOTLOG_CHATID, [url_msg.id, r.id])
""" - Unpack credential objects into strings - """
creds = base64.b64encode(pickle.dumps(creds)).decode()
await gdrive.edit("`Credentials created...`")
try:
helper.save_credentials(str(gdrive.from_id), creds)
except Exception:
for name in ['creds', 'gdrive']:
delete_table(name)
helper.save_credentials(str(gdrive.from_id), creds)
await gdrive.delete()
return