|
7 | 7 |
|
8 | 8 | import pydot |
9 | 9 | import rdflib |
| 10 | +from packaging.version import Version |
| 11 | + |
| 12 | +if Version(pydot.__version__) > Version("3.0"): |
| 13 | + quote_id_if_necessary = pydot.quote_id_if_necessary |
| 14 | +else: |
| 15 | + quote_id_if_necessary = pydot.quote_if_necessary |
10 | 16 |
|
11 | 17 |
|
12 | 18 | def _get_inner_edges_query() -> str: |
@@ -99,8 +105,8 @@ def _set_inner_edges(self) -> None: |
99 | 105 | self._dot_graph.add_node(n) |
100 | 106 | self._dot_graph.add_edge( |
101 | 107 | pydot.Edge( |
102 | | - pydot.quote_id_if_necessary(str(inner_edge_row["source_step"])), |
103 | | - pydot.quote_id_if_necessary(str(inner_edge_row["target_step"])), |
| 108 | + quote_id_if_necessary(str(inner_edge_row["source_step"])), |
| 109 | + quote_id_if_necessary(str(inner_edge_row["target_step"])), |
104 | 110 | ) |
105 | 111 | ) |
106 | 112 |
|
@@ -130,8 +136,8 @@ def _set_input_edges(self) -> None: |
130 | 136 | inputs_subgraph.add_node(n) |
131 | 137 | self._dot_graph.add_edge( |
132 | 138 | pydot.Edge( |
133 | | - pydot.quote_id_if_necessary(str(input_row["input"])), |
134 | | - pydot.quote_id_if_necessary(str(input_row["step"])), |
| 139 | + quote_id_if_necessary(str(input_row["input"])), |
| 140 | + quote_id_if_necessary(str(input_row["step"])), |
135 | 141 | ) |
136 | 142 | ) |
137 | 143 |
|
@@ -161,8 +167,8 @@ def _set_output_edges(self) -> None: |
161 | 167 | outputs_graph.add_node(n) |
162 | 168 | self._dot_graph.add_edge( |
163 | 169 | pydot.Edge( |
164 | | - pydot.quote_id_if_necessary(output_edge_row["step"]), |
165 | | - pydot.quote_id_if_necessary(output_edge_row["output"]), |
| 170 | + quote_id_if_necessary(output_edge_row["step"]), |
| 171 | + quote_id_if_necessary(output_edge_row["output"]), |
166 | 172 | ) |
167 | 173 | ) |
168 | 174 |
|
|
0 commit comments