mirror of https://github.com/zulip/zulip.git
93 lines
2.7 KiB
Python
93 lines
2.7 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
|
|
from collections import defaultdict
|
|
|
|
from typing import DefaultDict, List, Set, Tuple
|
|
|
|
class Graph(object):
|
|
def __init__(self, *tuples):
|
|
# type: (Tuple[str, str]) -> None
|
|
self.children = defaultdict(list) # type: DefaultDict[str, List[str]]
|
|
self.parents = defaultdict(list) # type: DefaultDict[str, List[str]]
|
|
self.nodes = set() # type: Set[str]
|
|
|
|
for parent, child in tuples:
|
|
self.parents[child].append(parent)
|
|
self.children[parent].append(child)
|
|
self.nodes.add(parent)
|
|
self.nodes.add(child)
|
|
|
|
def remove_exterior_nodes(self):
|
|
# type: () -> None
|
|
still_work_to_do = True
|
|
while still_work_to_do:
|
|
still_work_to_do = False # for now
|
|
for node in self.nodes:
|
|
if self.is_exterior_node(node):
|
|
self.remove(node)
|
|
still_work_to_do = True
|
|
break
|
|
|
|
def is_exterior_node(self, node):
|
|
# type: (str) -> bool
|
|
parents = self.parents[node]
|
|
children = self.children[node]
|
|
if not parents:
|
|
return True
|
|
if not children:
|
|
return True
|
|
if len(parents) > 1 or len(children) > 1:
|
|
return False
|
|
|
|
# If our only parent and child are the same node, then we could
|
|
# effectively be collapsed into the parent, so don't add clutter.
|
|
return parents[0] == children[0]
|
|
|
|
def remove(self, node):
|
|
# type: (str) -> None
|
|
for parent in self.parents[node]:
|
|
self.children[parent].remove(node)
|
|
for child in self.children[node]:
|
|
self.parents[child].remove(node)
|
|
self.nodes.remove(node)
|
|
|
|
def report(self):
|
|
# type: () -> None
|
|
print('parents/children/module')
|
|
tups = sorted([
|
|
(len(self.parents[node]), len(self.children[node]), node)
|
|
for node in self.nodes])
|
|
for tup in tups:
|
|
print(tup)
|
|
|
|
def make_dot_file(graph):
|
|
# type: (Graph) -> str
|
|
buffer = 'digraph G {\n'
|
|
for node in graph.nodes:
|
|
buffer += node + ';\n'
|
|
for child in graph.children[node]:
|
|
buffer += '{} -> {};\n'.format(node, child)
|
|
buffer += '}'
|
|
return buffer
|
|
|
|
def test():
|
|
# type: () -> None
|
|
graph = Graph(
|
|
('x', 'a'),
|
|
('a', 'b'),
|
|
('b', 'c'),
|
|
('c', 'a'),
|
|
('c', 'd'),
|
|
('d', 'e'),
|
|
('e', 'f'),
|
|
('e', 'g'),
|
|
)
|
|
graph.remove_exterior_nodes()
|
|
|
|
s = make_dot_file(graph)
|
|
open('zulip-deps.dot', 'w').write(s)
|
|
|
|
if __name__ == '__main__':
|
|
test()
|