Script to compare just directory names

comparedirs.py

#!/usr/bin/env python

import os
import sys


def remove_prefix(text, prefix):
    if text.startswith(prefix):
        return text[len(prefix):]
    return text


path1 = sys.argv[1]
path2 = sys.argv[2]

dirs1 = set([remove_prefix(dp, path1) for dp, dn, fn in os.walk(path1)])
dirs2 = set([remove_prefix(dp, path2) for dp, dn, fn in os.walk(path2)])

print('\n'.join(dirs1 - dirs2))