X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=gdata_oauth.py;h=983ab51255196353dd90d9d8472d901a73adfba7;hb=5ea88ab72e175e2d4f57ae8645ca6f825549a7a9;hp=d2595f0c1aed43c4f01f5095acb98b1194891986;hpb=5c39d86ebc075ccb7be98b1dfab8040b72ff9134;p=kiosk.git diff --git a/gdata_oauth.py b/gdata_oauth.py index d2595f0..983ab51 100644 --- a/gdata_oauth.py +++ b/gdata_oauth.py @@ -8,40 +8,36 @@ import logging import os import pickle -import sys -import urllib.request, urllib.parse, urllib.error from apiclient.discovery import build from google_auth_oauthlib.flow import InstalledAppFlow -logger = logging.getLogger(__file__) +logger = logging.getLogger(__name__) class OAuth: - def __init__(self, client_secret_file='client_secret.json'): + def __init__(self, client_secret_file="client_secret.json"): self.credentials = None - self.credentials_pickle = './credentials.pickle' + self.credentials_pickle = "./credentials.pickle" if os.path.exists(self.credentials_pickle): logger.debug( - f'Refreshing credentials from disk pickle file {self.credentials_pickle}' + f"Refreshing credentials from disk pickle file {self.credentials_pickle}" ) - self.credentials = pickle.load(open(self.credentials_pickle, 'rb')) + self.credentials = pickle.load(open(self.credentials_pickle, "rb")) else: - logger.debug( - f'{self.credentials_pickle} does not exist; calling Google.' - ) + logger.debug(f"{self.credentials_pickle} does not exist; calling Google.") self.refresh_credentials(client_secret_file) self.save() assert self.credentials is not None def refresh_credentials(self, client_secret_file): scopes = [ - 'https://www.googleapis.com/auth/calendar.events.readonly', - 'https://www.googleapis.com/auth/calendar.readonly', - 'https://www.googleapis.com/auth/drive.readonly', - 'https://www.googleapis.com/auth/drive.photos.readonly', - 'https://www.googleapis.com/auth/photoslibrary.readonly', + "https://www.googleapis.com/auth/calendar.events.readonly", + "https://www.googleapis.com/auth/calendar.readonly", + "https://www.googleapis.com/auth/drive.readonly", + "https://www.googleapis.com/auth/drive.photos.readonly", + "https://www.googleapis.com/auth/photoslibrary.readonly", # 'https://www.googleapis.com/auth/keep.readonly', ] flow = InstalledAppFlow.from_client_secrets_file( @@ -50,16 +46,19 @@ class OAuth: self.credentials = flow.run_console() def save(self): - pickle.dump(self.credentials, open(self.credentials_pickle, 'wb')) + pickle.dump(self.credentials, open(self.credentials_pickle, "wb")) def calendar_service(self): return build("calendar", "v3", credentials=self.credentials) def keep_service(self): - return build('keep', 'v1', - discoveryServiceUrl='https://keep.googleapis.com/$discovery/rest?version=v1', - credentials=self.credentials) - #print(gkeep_service.notes().list().execute()) + return build( + "keep", + "v1", + discoveryServiceUrl="https://keep.googleapis.com/$discovery/rest?version=v1", + credentials=self.credentials, + ) + # print(gkeep_service.notes().list().execute()) # class OAuth: