Merge pull request #36 from dexplo/format_pep

Format code with black and flake8 to meet PEP

Author: mjam03

jupyter_to_medium/__init__.py

@@ -1,9 +1,15 @@
 from ._publish_to_medium import publish
-__version__ = '0.2.5'
+
+__version__ = "0.2.5"
+
 
 def _jupyter_nbextension_paths():
-    return [dict(
-        section="notebook",
-        src="nbextension",
-        dest="jupyter_to_medium",
-        require="jupyter_to_medium/index")]
+    return [
+        dict(
+            section="notebook",
+            src="nbextension",
+            dest="jupyter_to_medium",
+            require="jupyter_to_medium/index",
+        )
+    ]
+

jupyter_to_medium/_bundler.py

@@ -1,94 +1,107 @@
 from pathlib import Path
 import json
-import urllib.parse
-
-from tornado import gen
 
 from ._publish_to_medium import publish
 
 # https://testnb.readthedocs.io/en/latest/examples/Notebook/Distributing%20Jupyter%20Extensions%20as%20Python%20Packages.html#
-# as explained here this is required to tell Jupyter how we want the bundle to show
+# as explained here, this is required to tell Jupyter how wto show bundle
 
 
 def _jupyter_bundlerextension_paths():
-    return [{
-        # the name of the bundle
-        "name": "jupyter_to_medium_bundler",
-        # where the bundler.py file sits to bundle the code
-        "module_name": "jupyter_to_medium._bundler",
-        # what we want it to show as in the menu
-        "label" : "Medium Post",
-        # which menu we want it as (download vs deploy)
-        "group" : "deploy",
-    }]
+    return [
+        {
+            # the name of the bundle
+            "name": "jupyter_to_medium_bundler",
+            # where the bundler.py file sits to bundle the code
+            "module_name": "jupyter_to_medium._bundler",
+            # what we want it to show as in the menu
+            "label": "Medium Post",
+            # which menu we want it as (download vs deploy)
+            "group": "deploy",
+        }
+    ]
 
 
 def upload(model, handler):
-    arguments = ['title', 'integration_token', 'pub_name', 'tags',
-                 'publish_status', 'notify_followers', 'license', 'canonical_url',
-                 'chrome_path', 'save_markdown', 'table_conversion', 'gistify',
-                 'gist_threshold']
+    arguments = [
+        "title",
+        "integration_token",
+        "pub_name",
+        "tags",
+        "publish_status",
+        "notify_followers",
+        "license",
+        "canonical_url",
+        "chrome_path",
+        "save_markdown",
+        "table_conversion",
+        "gistify",
+        "gist_threshold",
+    ]
 
     kwargs = {arg: handler.get_query_argument(arg, None) for arg in arguments}
-    path = model['path']
-    kwargs['filename'] = path
-    kwargs['integration_token'] = kwargs['integration_token'].strip() or None
-    kwargs['pub_name'] == kwargs['pub_name'].strip() or None
-    kwargs['tags'] = [tag.strip() for tag in kwargs['tags'].split(',')[:5]]
-    kwargs['notify_followers'] = kwargs['notify_followers'] == "True"
-    kwargs['canonical_url'] = kwargs['canonical_url'].strip() or None
-    kwargs['save_markdown'] = kwargs['save_markdown'] == "True"
-    kwargs['gistify'] = kwargs['gistify'] == "True"
-    if kwargs['gist_threshold'] == '':
-        kwargs['gist_threshold'] = 5
+    path = model["path"]
+    kwargs["filename"] = path
+    kwargs["integration_token"] = kwargs["integration_token"].strip() or None
+    kwargs["pub_name"] == kwargs["pub_name"].strip() or None
+    kwargs["tags"] = [tag.strip() for tag in kwargs["tags"].split(",")[:5]]
+    kwargs["notify_followers"] = kwargs["notify_followers"] == "True"
+    kwargs["canonical_url"] = kwargs["canonical_url"].strip() or None
+    kwargs["save_markdown"] = kwargs["save_markdown"] == "True"
+    kwargs["gistify"] = kwargs["gistify"] == "True"
+    if kwargs["gist_threshold"] == "":
+        kwargs["gist_threshold"] = 5
     else:
-        kwargs['gist_threshold'] = int(kwargs['gist_threshold'])
+        kwargs["gist_threshold"] = int(kwargs["gist_threshold"])
 
     # add these options in the future to html form
     # kwargs['chrome_path'] = kwargs['chrome_path'].strip() or None
-   
+
     try:
         data = publish(**kwargs)
     except Exception as e:
         import traceback
+
         error_name = type(e).__name__
-        error = f'{error_name}: {str(e)}'
+        error = f"{error_name}: {str(e)}"
         tb = traceback.format_exc()
-        msg = error + f'\n\n{tb}'
+        msg = error + f"\n\n{tb}"
         print(msg)
-        msg = msg.replace('\n', '<br>')
-        data = {'app_status': 'fail', 
-                'error_data': msg}
+        msg = msg.replace("\n", "<br>")
+        data = {"app_status": "fail", "error_data": msg}
     else:
-        if 'data' in data:
-            data = data['data']
-            data['app_status'] = "success"
+        if "data" in data:
+            data = data["data"]
+            data["app_status"] = "success"
         else:
-            data = {'app_status': 'fail', 
-                    'error_data': 'Error: \n' + str(data)}
+            data = {
+                "app_status": "fail",
+                "error_data": "Error: \n" + str(data),
+            }
 
     return data
 
+
 def read_html(name):
     mod_path = Path(__file__).parent
-    html_path = mod_path / 'static' / f'{name}.html'
+    html_path = mod_path / "static" / f"{name}.html"
     return open(html_path).read()
 
 
 def get_html_form(xsrf_input, title):
-    html = read_html('form')
+    html = read_html("form")
     return html.format(xsrf_input=xsrf_input, title=title)
 
 
 def get_html_success(data):
-    html = read_html('success')
+    html = read_html("success")
     return html.format(**data)
 
 
 def get_html_fail(data):
-    error_data = data['error_data']
+    error_data = data["error_data"]
     error_message = json.dumps(error_data)
-    html = read_html('fail')
+    html = read_html("fail")
     return html.format(error_message=error_message)
 
 
@@ -105,34 +118,33 @@ def bundle(handler, model):
         Notebook model from the configured ContentManager
     """
     # check on the app_status in the form.html tag
-    app_status = handler.get_query_argument('app_status', None)
-    
+    app_status = handler.get_query_argument("app_status", None)
+
     # if we've never initialised the form e.g. when clicked
     if app_status is None:
         # create the input element to be included in our form
         xsrf_input = handler.xsrf_form_html()
         # get the base notebook filename to populate the form
-        notebook_filename = Path(model['name']).stem
+        notebook_filename = Path(model["name"]).stem
         # create the html we will push to the user (the input form)
         html = get_html_form(xsrf_input, notebook_filename)
         # write this html to the tornado handler so user will see
         handler.write(html)
     # else if we've already initialised the form
-    elif app_status == 'waiting':
+    elif app_status == "waiting":
         # read the html if we've filled in more of the user fields
         # this is required so we continuously serve the correct html
-        html = read_html('waiting')
+        html = read_html("waiting")
         # write updated html
         handler.write(html)
         handler.flush()
 
         data = upload(model, handler)
-        if data['app_status'] == 'success':
+        if data["app_status"] == "success":
             html = get_html_success(data)
         else:
             html = get_html_fail(data)
-        
+
         handler.write(html)
 
     handler.finish()
-