CLN: Use Black formatter on codebase (#18)

Author: achabotl

tests/test_zkviz.py

@@ -6,12 +6,11 @@
 
 
 class TestListZettels(TestCase):
-
     def test_list_zettels_with_md_extension(self):
         # Create a temporary folder and write files in it
         with TemporaryDirectory() as tmpdirname:
-            ext = '.md'
-            basename = '201906242157'
+            ext = ".md"
+            basename = "201906242157"
             filepaths = []
             for i in range(3):
                 path = Path(tmpdirname, basename + str(i) + ext)
@@ -24,49 +23,48 @@ def test_list_zettels_with_md_extension(self):
     def test_list_zettels_with_txt_extension(self):
         # Create a temporary folder and write files in it
         with TemporaryDirectory() as tmpdirname:
-            ext = '.txt'
-            basename = '201906242157'
+            ext = ".txt"
+            basename = "201906242157"
             filepaths = []
             for i in range(3):
                 path = Path(tmpdirname, basename + str(i) + ext)
                 path.touch()
                 filepaths.append(str(path))
 
-            files_found = zkviz.list_zettels(tmpdirname, '*.txt')
+            files_found = zkviz.list_zettels(tmpdirname, "*.txt")
             self.assertEqual(filepaths, files_found)
 
     def test_list_zettels_with_mixed_extensions(self):
         # Create a temporary folder and write files in it
         with TemporaryDirectory() as tmpdirname:
             filepaths = []
-            basename = '201906242157'
+            basename = "201906242157"
 
-            ext = '.txt'
+            ext = ".txt"
             for i in range(5):
                 path = Path(tmpdirname, basename + str(i) + ext)
                 path.touch()
                 filepaths.append(str(path))
 
-            ext = '.md'
+            ext = ".md"
             for i in range(5, 10):
                 path = Path(tmpdirname, basename + str(i) + ext)
                 path.touch()
                 filepaths.append(str(path))
 
-            files_found = zkviz.list_zettels(tmpdirname, '*.txt|*.md')
+            files_found = zkviz.list_zettels(tmpdirname, "*.txt|*.md")
             self.assertEqual(filepaths, files_found)
 
 
 class TestParseArgs(TestCase):
-
     def test_default_extension(self):
-        args = zkviz.parse_args('')
-        self.assertEqual(['*.md'], args.pattern)
+        args = zkviz.parse_args("")
+        self.assertEqual(["*.md"], args.pattern)
 
     def test_overwrite_extension(self):
-        args = zkviz.parse_args(["--pattern","*.txt"])
-        self.assertEqual(['*.txt'], args.pattern)
+        args = zkviz.parse_args(["--pattern", "*.txt"])
+        self.assertEqual(["*.txt"], args.pattern)
 
     def test_multiple_extensions(self):
-        args = zkviz.parse_args(["--pattern","*.txt", "--pattern", '*.md'])
-        self.assertEqual(['*.txt', '*.md'], args.pattern)
+        args = zkviz.parse_args(["--pattern", "*.txt", "--pattern", "*.md"])
+        self.assertEqual(["*.txt", "*.md"], args.pattern)

zkviz/__init__.py

@@ -1 +1 @@
-__version__ = '1.2.0'
+__version__ = "1.2.0"

zkviz/graphviz.py

@@ -4,8 +4,7 @@
 
 
 class NetworkGraphviz:
-
-    def __init__(self, name='Zettelkasten', engine='sfdp', shape='record'):
+    def __init__(self, name="Zettelkasten", engine="sfdp", shape="record"):
         """
         Build network to visualize with Graphviz.
 
@@ -50,11 +49,11 @@ def add_node(self, node_id, title):
             The text label for each node, typically the zettel title.
 
         """
-        if self.shape == 'plaintext':
+        if self.shape == "plaintext":
             label = self.wrap_title("{} {}".format(node_id, title)).strip()
-        elif self.shape == 'record':
+        elif self.shape == "record":
             # Wrap in {} so the elements are stacked vertically
-            label = "{" + '|'.join([node_id, self.wrap_title(title)]) + "}"
+            label = "{" + "|".join([node_id, self.wrap_title(title)]) + "}"
 
         self.graph.node(node_id, label, shape=self.shape)
 

zkviz/plotly.py

@@ -3,8 +3,7 @@
 
 
 class NetworkPlotly:
-
-    def __init__(self, name='Zettelkasten'):
+    def __init__(self, name="Zettelkasten"):
         """
         Build network to visualize with Plotly
 
@@ -75,64 +74,71 @@ def build_plotly_figure(self, pos=None):
             x=[],
             y=[],
             text=[],
-            mode='markers',
-            hoverinfo='text',
+            mode="markers",
+            hoverinfo="text",
             marker=dict(
                 showscale=True,
                 # colorscale options
                 #'Greys' | 'YlGnBu' | 'Greens' | 'YlOrRd' | 'Bluered' | 'RdBu' |
                 #'Reds' | 'Blues' | 'Picnic' | 'Rainbow' | 'Portland' | 'Jet' |
                 #'Hot' | 'Blackbody' | 'Earth' | 'Electric' | 'Viridis' |
-                colorscale='YlGnBu',
+                colorscale="YlGnBu",
                 reversescale=True,
                 color=[],
                 size=10,
                 colorbar=dict(
-                    thickness=15,
-                    title='Centrality',
-                    xanchor='left',
-                    titleside='right'
+                    thickness=15, title="Centrality", xanchor="left", titleside="right"
                 ),
-                line=dict(width=2)))
+                line=dict(width=2),
+            ),
+        )
 
         for node in self.graph.nodes():
             x, y = pos[node]
-            text = '<br>'.join([node, self.graph.node[node].get('title', '')])
-            node_trace['x'] += tuple([x])
-            node_trace['y'] += tuple([y])
-            node_trace['text'] += tuple([text])
+            text = "<br>".join([node, self.graph.node[node].get("title", "")])
+            node_trace["x"] += tuple([x])
+            node_trace["y"] += tuple([y])
+            node_trace["text"] += tuple([text])
 
         # Color nodes based on the centrality
         for node, centrality in nx.degree_centrality(self.graph).items():
-            node_trace['marker']['color']+=tuple([centrality])
+            node_trace["marker"]["color"] += tuple([centrality])
 
         # Draw the edges as annotations because it's only sane way to draw arrows.
         edges = []
         for from_node, to_node in self.graph.edges():
             edges.append(
                 dict(
                     # Tail coordinates
-                    ax=pos[from_node][0], ay=pos[from_node][1], axref='x', ayref='y',
+                    ax=pos[from_node][0],
+                    ay=pos[from_node][1],
+                    axref="x",
+                    ayref="y",
                     # Head coordinates
-                    x=pos[to_node][0], y=pos[to_node][1], xref='x', yref='y',
+                    x=pos[to_node][0],
+                    y=pos[to_node][1],
+                    xref="x",
+                    yref="y",
                     # Aesthetics
-                    arrowwidth=2, arrowcolor='#666', arrowhead=2,
+                    arrowwidth=2,
+                    arrowcolor="#666",
+                    arrowhead=2,
                     # Have the head stop short 5 px for the center point,
                     # i.e., depends on the node marker size.
                     standoff=5,
-                    )
                 )
+            )
 
         fig = go.Figure(
             data=[node_trace],
             layout=go.Layout(
                 showlegend=False,
-                hovermode='closest',
+                hovermode="closest",
                 margin=dict(b=20, l=5, r=5, t=40),
                 annotations=edges,
                 xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
                 yaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
-            )
+            ),
         )
         return fig
 
@@ -150,6 +156,6 @@ def render(self, output, view=True):
 
         """
         fig = self.build_plotly_figure()
-        if not output.endswith('.html'):
-            output += '.html'
+        if not output.endswith(".html"):
+            output += ".html"
         fig.write_html(output, auto_open=view)