tilt-listdir.py

def listdir_relative(path):
    result = []
    cwd = os.getcwd() + '/'
    for entry in listdir(path):
        if not entry.startswith(cwd):
            fail('Unexpected entry ' + entry)
        entry = entry[len(cwd):]
        result.append(entry)
    return result

def listdir_relative_with_dirs(path):
    result = {}
    root = os.path.join(os.getcwd(), path) + '/'
    for entry in listdir(path, recursive=True):
        # Make the entry relative
        if not entry.startswith(root):
            fail('Unexpected entry ' + entry)
        entry = entry[len(root):]
        # Is it in a subdirectory?
        slash = entry.find('/')
        if slash == -1:
            # Top-level, add it
            result[entry] = True
        else:
            # Subdirectory, add the subdirectory itself
            result[entry[:slash]] = True

    if path == '.':
        return result.keys()
    else:
        return [os.path.join(path, entry) for entry in result.keys()]