Format codebase w/ black.
[kiosk.git] / gdocs_renderer.py
index d734a2deb90c4051db50d9d1fb97af2492be9f9b..44203e539094f12468189aba8bde52e520edcdf0 100644 (file)
@@ -5,6 +5,7 @@ import sets
 import gdata_oauth
 import secrets
 
+
 class gdocs_renderer(renderer.debuggable_abstaining_renderer):
     """A renderer to fetches and munge docs from drive.google.com"""
 
@@ -25,13 +26,13 @@ class gdocs_renderer(renderer.debuggable_abstaining_renderer):
             try:
                 param = {}
                 if page_token:
-                    param['pageToken'] = page_token
-                param['q'] = self.query
-                print("QUERY: %s" % param['q'])
+                    param["pageToken"] = page_token
+                param["q"] = self.query
+                print("QUERY: %s" % param["q"])
 
                 files = self.client.files().list(**param).execute()
-                result.extend(files['items'])
-                page_token = files.get('nextPageToken')
+                result.extend(files["items"])
+                page_token = files.get("nextPageToken")
                 if not page_token:
                     break
             except:
@@ -47,11 +48,11 @@ class gdocs_renderer(renderer.debuggable_abstaining_renderer):
             return "font-size:%dpt" % (x)
 
         for f in result:
-            print(f['title'])
-            print(f['id'])
-            self.debug_print("%s (%s)\n" % (f['title'], f['id']))
-            title = f['title']
-            url = f['exportLinks']['text/html']
+            print(f["title"])
+            print(f["id"])
+            self.debug_print("%s (%s)\n" % (f["title"], f["id"]))
+            title = f["title"]
+            url = f["exportLinks"]["text/html"]
             print(f)
             print("Fetching %s..." % url)
             resp, contents = self.client._http.request(url)
@@ -59,18 +60,21 @@ class gdocs_renderer(renderer.debuggable_abstaining_renderer):
             print(contents)
             if resp.status == 200:
                 print("Got contents.")
-                contents = re.sub('<body class="..">', '', contents)
-                contents = contents.replace('</body>', '')
-                contents = re.sub('font-size:([0-9]+)pt', boost_font_size, contents)
-                f = file_writer.file_writer('%s_2_3600.html' % title)
+                contents = re.sub('<body class="..">', "", contents)
+                contents = contents.replace("</body>", "")
+                contents = re.sub("font-size:([0-9]+)pt", boost_font_size, contents)
+                f = file_writer.file_writer("%s_2_3600.html" % title)
                 now = datetime.datetime.now()
-                f.write("""
+                f.write(
+                    """
 <H1>%s</H1>
 <!-- Last updated at %s -->
 <HR>
 <DIV STYLE="-webkit-column-count: 2; -moz-column-count: 2; column-count: 2;">
 %s
-</DIV>""" % (title, now, contents))
+</DIV>"""
+                    % (title, now, contents)
+                )
                 f.close()
             else:
                 self.debug_print("error: %s" % resp)
@@ -78,8 +82,8 @@ class gdocs_renderer(renderer.debuggable_abstaining_renderer):
         return True
 
 
-#oauth = gdata_oauth.OAuth(secrets.google_client_id,
+# oauth = gdata_oauth.OAuth(secrets.google_client_id,
 #                          secrets.google_client_secret)
-#x = gdocs_renderer({"Testing", 12345},
+# x = gdocs_renderer({"Testing", 12345},
 #                   oauth)
-#x.periodic_render("Test")
+# x.periodic_render("Test")