Function strip_py

mypy/find_sources.py:249–257

Strip a trailing .py or .pyi suffix. Return None if no such suffix is found.

(arg: str)

def strip_py(arg: str) -> str | None:
    """Strip a trailing .py or .pyi suffix.

    Return None if no such suffix is found.
    """
    for ext in PY_EXTENSIONS:
        if arg.endswith(ext):
            return arg[: -len(ext)]
    return None