rolling-pin 0.10.0__tar.gz → 0.10.1__tar.gz

{rolling-pin-0.10.0 → rolling-pin-0.10.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rolling-pin
-Version: 0.10.0
+Version: 0.10.1
 Summary: A library of generic tools for ETL work and visualization of JSON blobs and python repositories
 License: MIT
 Keywords: ETL,blob,dependency,graph,svg,networkx,transform,code metrics,dependency diagram,build system

{rolling-pin-0.10.0 → rolling-pin-0.10.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "rolling-pin"
-version = "0.10.0"
+version = "0.10.1"
 description = "A library of generic tools for ETL work and visualization of JSON blobs and python repositories"
 readme = "README.md"
 license = "MIT"
@@ -33,10 +33,10 @@ dependencies = [
     "ipython",
     "lunchbox",
     "networkx",
-    "numpy>=1.23.4,<=1.24.4",
+    "numpy>=1.23.4",
     "pandas>=1.1.5",
     "plotly>=5.22.0",
-    "pydot<=1.4.2",
+    "pydot>=1.4.2",
     "pyyaml",
     "radon<6.0.0",
     "schematics",

{rolling-pin-0.10.0 → rolling-pin-0.10.1}/rolling_pin/repo_etl.py

@@ -335,17 +335,28 @@ class RepoETL():
         return data
 
     @staticmethod
-    def _to_networkx_graph(data):
-        # (DataFrame) -> networkx.DiGraph
+    def _to_networkx_graph(data, escape_chars=False):
+        # (DataFrame, bool) -> networkx.DiGraph
         '''
         Converts given DataFrame into networkx directed graph.
 
         Args:
-            DataFrame: DataFrame of nodes.
+            data (DataFrame): DataFrame of nodes.
+            escape_chars (bool, optional): Escape special characters. Used to
+                avoid dot file errors. Default: False.
 
         Returns:
             networkx.DiGraph: Graph of nodes.
         '''
+        # escape periods for dot file interpolation
+        if escape_chars:
+            data = data.copy()
+            data.node_name = data.node_name \
+                .fillna('') \
+                .apply(lambda x: re.sub(r'\.', '\\.', x))
+            data.dependencies = data.dependencies \
+                .apply(lambda x: [re.sub(r'\.', '\\.', y) for y in x])
+
         graph = networkx.DiGraph()
         data.apply(
             lambda x: graph.add_node(
@@ -406,7 +417,7 @@ class RepoETL():
             color_scheme = rpt.COLOR_SCHEME
 
         # create dot graph
-        graph = self.to_networkx_graph()
+        graph = self._to_networkx_graph(self._data, escape_chars=True)
         dot = networkx.drawing.nx_pydot.to_pydot(graph)
 
         # set layout orientation