rolling-pin 0.10.0__py3-none-any.whl → 0.10.1__py3-none-any.whl

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

{rolling_pin-0.10.0.dist-info → rolling_pin-0.10.1.dist-info}/METADATA

@@ -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
@@ -19,10 +19,10 @@ Requires-Dist: graphviz
 Requires-Dist: ipython
 Requires-Dist: lunchbox
 Requires-Dist: networkx
-Requires-Dist: numpy<=1.24.4,>=1.23.4
+Requires-Dist: numpy>=1.23.4
 Requires-Dist: pandas>=1.1.5
 Requires-Dist: plotly>=5.22.0
-Requires-Dist: pydot<=1.4.2
+Requires-Dist: pydot>=1.4.2
 Requires-Dist: pyyaml
 Requires-Dist: radon<6.0.0
 Requires-Dist: schematics

{rolling_pin-0.10.0.dist-info → rolling_pin-0.10.1.dist-info}/RECORD

@@ -5,11 +5,11 @@ rolling_pin/conform_config.py,sha256=W-Ps6FPuZ7SOxFOfEjIo2MZac1vCRuVky6samSZPQkw
 rolling_pin/conform_etl.py,sha256=VyHAKVvdOtjx6Fy8dWn6XMfNh9OFgBbidyDUnSl3Zl4,9594
 rolling_pin/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rolling_pin/radon_etl.py,sha256=0rxFSnv2z41vKFe35XqCHSnnjT9qNNjouaCcIYkcHgI,18252
-rolling_pin/repo_etl.py,sha256=W3ms6zdK22A6d9kVzHLP3HE1h0fTPMUeXww0c57Nq-k,20479
+rolling_pin/repo_etl.py,sha256=Q6W-_4hTNh3nj5o9eTz5d7HuH5Kf9Uew8W9UD52dCJ0,21041
 rolling_pin/toml_etl.py,sha256=fyioQTfqFW1e15qZ8F64sX379V5JzNlBKBuC5IPIgIo,4350
 rolling_pin/tools.py,sha256=bLZGXDg5DuV59RxiVjLIwTAr83X5nMBElvR7w0mEnx8,19179
-rolling_pin-0.10.0.dist-info/entry_points.txt,sha256=nbMiJnGLhp49xrtx1pjK8sJiBrwkpj2kq-ezp8Zfyyk,56
-rolling_pin-0.10.0.dist-info/WHEEL,sha256=B19PGBCYhWaz2p_UjAoRVh767nYQfk14Sn4TpIZ-nfU,87
-rolling_pin-0.10.0.dist-info/METADATA,sha256=Oh1-rm5lU1LsZg4ufs40VoEQd9FR7M2MBJ5GINJfpJw,20063
-rolling_pin-0.10.0.dist-info/licenses/LICENSE,sha256=Z61vsRc9ZyRu0Tm3TkYgIeKDjUJCR3dT0hJonw9Q3i4,1067
-rolling_pin-0.10.0.dist-info/RECORD,,
+rolling_pin-0.10.1.dist-info/entry_points.txt,sha256=nbMiJnGLhp49xrtx1pjK8sJiBrwkpj2kq-ezp8Zfyyk,56
+rolling_pin-0.10.1.dist-info/WHEEL,sha256=B19PGBCYhWaz2p_UjAoRVh767nYQfk14Sn4TpIZ-nfU,87
+rolling_pin-0.10.1.dist-info/METADATA,sha256=B7s0WHFa3vdPUKujCpbyGuA45FAmlphifDzmRDonhSM,20054
+rolling_pin-0.10.1.dist-info/licenses/LICENSE,sha256=Z61vsRc9ZyRu0Tm3TkYgIeKDjUJCR3dT0hJonw9Q3i4,1067
+rolling_pin-0.10.1.dist-info/RECORD,,