jebus/enviPy-bayer
1
0
Fork 0
forked from enviPath/enviPy
Code Issues Pull Requests Actions Releases Activity

clean up d3_json

Browse Source
This commit is contained in:
Tim Lorsbach
2025-10-31 09:38:41 +01:00
parent 13ed86a780
commit c2d45917ce
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
epdb/models.py
View File

@ -1573,14 +1573,12 @@ class Pathway(EnviPathModel, AliasMixin, ScenarioMixin):
while len(queue): while len(queue):
current = queue.pop() current = queue.pop()
processed.add(current) processed.add(current)
nodes.append(current.d3_json()) nodes.append(current.d3_json())
for e in self.edges: for e in self.edges.filter(start_nodes=current).distinct():
if current in e.start_nodes.all(): for prod in e.end_nodes.all():
for prod in e.end_nodes.all(): if prod not in queue and prod not in processed:
if prod not in queue and prod not in processed: queue.append(prod)
queue.append(prod)
# We shouldn't lose or make up nodes... # We shouldn't lose or make up nodes...
assert len(nodes) == len(self.nodes) assert len(nodes) == len(self.nodes)