Code formatting.

Author: ericmjl

pyjviz/__init__.py

@@ -1,23 +1,29 @@
 from .fstriplestore import set_triple_store__, FSTripleOutputOneShot
-from .wb_stack import get_wb_stack
-from .wb_stack_entries import *
-from .obj_utils import *
-from .viz import *
-from .rdf_node import arrow
+import os
+import sys
 
 from .pf_pandas import enable_pf_pandas__
 
-pyjviz_enabled = not ('PYJVIZ_DISABLED' in os.environ and os.environ.get('PYJVIZ_DISABLED').lower() in ['1', 'true','yes'])
+pyjviz_enabled = not (
+    "PYJVIZ_DISABLED" in os.environ
+    and os.environ.get("PYJVIZ_DISABLED").lower() in ["1", "true", "yes"]
+)
 
 if pyjviz_enabled:
-    if os.path.basename(sys.argv[0]) == 'ipykernel_launcher.py':
-        tstr = FSTripleOutputOneShot(None, None) # -- NB: how can we detect that this is notebook run
-    else:        
-        pyjviz_output_dir = os.environ.get('PYJVIZ_OUTPUT_DIR') if 'PYJVIZ_OUTPUT_DIR' in os.environ else "~/.pyjviz"            
+    if os.path.basename(sys.argv[0]) == "ipykernel_launcher.py":
+        tstr = FSTripleOutputOneShot(
+            None, None
+        )  # -- NB: how can we detect that this is notebook run
+    else:
+        pyjviz_output_dir = (
+            os.environ.get("PYJVIZ_OUTPUT_DIR")
+            if "PYJVIZ_OUTPUT_DIR" in os.environ
+            else "~/.pyjviz"
+        )
         pyjviz_output_dir = os.path.expanduser(pyjviz_output_dir)
-        tstr = FSTripleOutputOneShot(pyjviz_output_dir,
-                                     os.path.basename(sys.argv[0]) + ".ttl")
+        tstr = FSTripleOutputOneShot(
+            pyjviz_output_dir, os.path.basename(sys.argv[0]) + ".ttl"
+        )
 
     set_triple_store__(tstr)
     enable_pf_pandas__()
-

pyjviz/fstriplestore.py

@@ -1,23 +1,33 @@
-#import ipdb
-import os, os.path, textwrap
-import tempfile
+# import ipdb
+import os
+import os.path
+import textwrap
 import io
 import rdflib
 
-base_uri = 'https://github.com/pyjanitor-devs/pyjviz/rdflog.shacl.ttl#'
+base_uri = "https://github.com/pyjanitor-devs/pyjviz/rdflog.shacl.ttl#"
+
 
 class FSTripleOutput:
     def __init__(self, out_fd):
         global base_uri
         self.base_uri = base_uri
         self.out_fd = out_fd
-        
+
     def dump_prefixes__(self):
-        print(f"@base <{self.base_uri}> .", file = self.out_fd)
-        print("@prefix rdf: <http://www.w3.org/1999/02/22-rdf-syntax-ns#> .", file = self.out_fd)
-        print("@prefix rdfs: <http://www.w3.org/2000/01/rdf-schema#> .", file = self.out_fd)
+        print(f"@base <{self.base_uri}> .", file=self.out_fd)
+        print(
+            "@prefix rdf: <http://www.w3.org/1999/02/22-rdf-syntax-ns#> .",
+            file=self.out_fd,
+        )
+        print(
+            "@prefix rdfs: <http://www.w3.org/2000/01/rdf-schema#> .",
+            file=self.out_fd,
+        )
 
-        print(textwrap.dedent("""
+        print(
+            textwrap.dedent(
+                """
         <WithBlock> rdf:type rdfs:Class .
         <CodeBlock> rdf:type rdfs:Class .
         <CodeBlock> rdfs:subClassOf <WithBlock> .
@@ -35,14 +45,18 @@ def dump_prefixes__(self):
         <CCGlance> rdfs:subClassOf <CC> .
         <CCBasicPlot> rdf:type rdfs:Class .
         <CCBasicPlot> rdfs:subClassOf <CC> .
-        """), file = self.out_fd)
-        
+        """
+            ),
+            file=self.out_fd,
+        )
+
     def dump_triple(self, subj, pred, obj):
-        print(f"{subj} {pred} {obj} .", file = self.out_fd)
+        print(f"{subj} {pred} {obj} .", file=self.out_fd)
 
     def flush(self):
         self.out_fd.flush()
 
+
 class FSTripleOutputOneShot(FSTripleOutput):
     def __init__(self, output_dir, output_filename):
         self.output_dir = output_dir
@@ -51,7 +65,7 @@ def __init__(self, output_dir, output_filename):
         if self.output_dir is None:
             out_fd = io.StringIO()
         else:
-            #ipdb.set_trace()
+            # ipdb.set_trace()
             self.output_dir = os.path.expanduser(self.output_dir)
             if not os.path.exists(self.output_dir):
                 os.makedirs(self.output_dir)
@@ -63,19 +77,23 @@ def __init__(self, output_dir, output_filename):
 
         super().__init__(out_fd)
         self.dump_prefixes__()
-        
+
     def get_graph(self):
         g = rdflib.Graph()
         self.out_fd.seek(0)
         g.parse(self.out_fd)
         return g
 
+
 class FSTripleOutputNull:
-    def dump_triple(self, s, p, o): pass
-    
+    def dump_triple(self, s, p, o):
+        pass
+
+
 triple_store = None
+
+
 def set_triple_store__(o):
     print("setting up triple_store:", o)
     global triple_store
     triple_store = o
-

pyjviz/wb_stack.py

@@ -1,22 +1,28 @@
 from . import fstriplestore
 from . import rdf_node
 
+
 class WithBlock(rdf_node.RDFNode):
     """
     Code blocks in python - https://docs.python.org/3/reference/executionmodel.html#:~:text=A%20Python%20program%20is%20constructed,typed%20interactively%20is%20a%20block.
     With PEP use of 'block' - https://peps.python.org/pep-0343/#specification-the-with-statement
 
     Base class for CodeBlock, MethodCall and NestedCall.
       - uri - to identify itself in graph as node
-      - rdf_type_uri - to identify node type
-    """
+      - rdf_type_uri- to identify node type
+    """  # noqa: E501
+
     def __init__(self, *, rdf_type, label):
         super().__init__(rdf_type, label)
         rdfl = fstriplestore.triple_store
         global wb_stack
-        parent_uri = wb_stack.stack_entries__[-1].uri if wb_stack.size() > 0 else "rdf:nil"
+        parent_uri = (
+            wb_stack.stack_entries__[-1].uri
+            if wb_stack.size() > 0
+            else "rdf:nil"
+        )
         rdfl.dump_triple(self.uri, "<part-of>", parent_uri)
-        
+
     def __enter__(self):
         global wb_stack
         wb_stack.push(self)
@@ -26,20 +32,23 @@ def __exit__(self, type, value, traceback):
         fstriplestore.triple_store.flush()
         global wb_stack
         wb_stack.pop()
-        
+
+
 class WithBlockStack:
     def __init__(self):
         self.stack_entries__ = []
 
     def to_string(self):
-        return ":".join([f"{x.label}@{x.rdf_type}" for x in self.stack_entries__])
-    
+        return ":".join(
+            [f"{x.label}@{x.rdf_type}" for x in self.stack_entries__]
+        )
+
     def size(self):
         return len(self.stack_entries__)
 
     def push(self, e):
         self.stack_entries__.append(e)
-        
+
     def pop(self):
         return self.stack_entries__.pop()
 
@@ -61,9 +70,11 @@ def get_latest_method_call(self):
     def get_parent_of_current_entry(self):
         ret = self.stack_entries__[-1] if self.size() > 0 else None
         return ret
-        
-    
+
+
 wb_stack = WithBlockStack()
+
+
 def get_wb_stack():
     global wb_stack
     return wb_stack