there is already a bit of code almost doing this here: https://github.com/pantsbuild/pants/blob/master/src/python/pants/engine/legacy/graph.py#L320-L348