github/datasette
1
0
Fork 0
mirror of https://github.com/simonw/datasette.git synced 2025-12-10 16:51:24 +01:00
Code Issues Projects Releases Packages Wiki Activity

Backport graphlib to work on Python 3.6, refs #878

Browse source
This commit is contained in:
Simon Willison 2021-11-16 15:46:50 -08:00
commit 8f757da075
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
datasette/utils/vendored_graphlib.py
View file

@ -1,4 +1,5 @@
# Vendored from https://raw.githubusercontent.com/python/cpython/3.10/Lib/graphlib.py # Vendored from https://raw.githubusercontent.com/python/cpython/3.10/Lib/graphlib.py
# Modified to work on Python 3.6 (I removed := operator)
# License: https://github.com/python/cpython/blob/main/LICENSE # License: https://github.com/python/cpython/blob/main/LICENSE
__all__ = ["TopologicalSorter", "CycleError"] __all__ = ["TopologicalSorter", "CycleError"]
@ -53,7 +54,8 @@ class TopologicalSorter:
self.add(node, *predecessors) self.add(node, *predecessors)
def _get_nodeinfo(self, node): def _get_nodeinfo(self, node):
if (result := self._node2info.get(node)) is None: result = self._node2info.get(node)
if result is None:
self._node2info[node] = result = _NodeInfo(node) self._node2info[node] = result = _NodeInfo(node)
return result return result
@ -169,7 +171,8 @@ class TopologicalSorter:
for node in nodes: for node in nodes:
# Check if we know about this node (it was added previously using add() # Check if we know about this node (it was added previously using add()
if (nodeinfo := n2i.get(node)) is None: nodeinfo = n2i.get(node)
if nodeinfo is None:
raise ValueError(f"node {node!r} was not added using add()") raise ValueError(f"node {node!r} was not added using add()")
# If the node has not being returned (marked as ready) previously, inform the user. # If the node has not being returned (marked as ready) previously, inform the user.