Fix logging and remove cruft.
[kiosk.git] / gdata_oauth.py
index d2595f0c1aed43c4f01f5095acb98b1194891986..983ab51255196353dd90d9d8472d901a73adfba7 100644 (file)
@@ -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: