import sys
from slither.slither import Slither


if len(sys.argv) != 2:
    print('python export_dominator_tree_to_dot.py contract.sol')
    exit(-1)

# Init slither
slither = Slither(sys.argv[1])

for contract in slither.contracts:
    for function in contract.functions + contract.modifiers:
        filename = "{}-{}-{}_dom.dot".format(sys.argv[1], contract.name, function.full_name)
        print('Export {}'.format(filename))
        function.dominator_tree_to_dot(filename)