black.py

import os

from pkg_resources import get_distribution


def check(cmd, mf):
    m = mf.findNode("black")
    if m is None or m.filename is None:
        return None

    egg = get_distribution("black").egg_info
    top = os.path.join(egg, "top_level.txt")

    # These cannot be in zip
    packages = ["black", "blib2to3"]

    # black may include optimized platform specific C extension which has
    # unusual name, e.g. 610faff656c4cfcbb4a3__mypyc; best to determine it from
    # the egg-info/top_level.txt
    with open(top, "r") as f:
        includes = set(f.read().strip().split("\n"))
    includes = list(includes.difference(packages))

    # Missed dependency
    includes.append("pathspec")

    return {"includes": includes, "packages": packages}