Files
gork/googledoccy.py
2025-05-22 22:16:10 -07:00

61 lines
2.1 KiB
Python

import os.path
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
# If modifying these scopes, delete the file token.json.
SCOPES = ["https://www.googleapis.com/auth/documents.readonly"]
# The ID of a sample document.
DOCUMENT_ID = "11kMFDtTfACx7oCkxc0knTtXe8vPcddw8sNNvTb1u0_g"
def jackblack():
"""Shows basic usage of the Docs API.
Prints the title of a sample document.
"""
creds = None
# 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("token.json"):
creds = Credentials.from_authorized_user_file("token.json", SCOPES)
# 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(
"credentials.json", SCOPES
)
creds = flow.run_local_server(port=0)
# Save the credentials for the next run
with open("token.json", "w") as token:
token.write(creds.to_json())
try:
service = build("docs", "v1", credentials=creds)
# Retrieve the documents contents from the Docs service.
document = service.documents().get(documentId=DOCUMENT_ID).execute()
#print(document.get('body').get('content'))
resps = []
doc_content = document.get('body').get('content')
for element in doc_content:
if 'paragraph' in element:
for text_run in element['paragraph']['elements']:
resps.append(text_run['textRun']['content'].strip("\n"))
#print(resps)
except HttpError as err:
resps = ['Your free trial for "gork" has expired! Purchase a subscription to our discord bot AI hosting services [here](<https://traps.moe/koa2.png>) to continue.']
print(err)
pass
return resps