openide/python/helpers/syspath.py

import os.path
import sys

_helpers_root = os.path.dirname(os.path.abspath(__file__))
_working_dir = os.getcwd()


def get_pep660_editable_roots_from_metadata():
    try:
        # importlib.metadata was added in 3.8
        from importlib.metadata import distributions, Distribution
    except ImportError:
        return []

    def read_from_direct_url_json(dist):
        # type: (Distribution) -> str | None

        import json
        direct_url_file = dist._path / 'direct_url.json'
        if direct_url_file.exists():
            with direct_url_file.open() as f:
                data = json.load(f)
                if data.get('dir_info', {}).get('editable'):
                    return data.get('url', '')
        return None

    editable_roots = []
    for dist in distributions():
        url = None  # type: str | None
        try:
            # 'origin' was added in Python 3.13
            if dist.origin.dir_info.editable:
                url = dist.origin.url
        except AttributeError:
            url = read_from_direct_url_json(dist)

        FILE_PREFIX = "file://"
        if url and url.startswith(FILE_PREFIX):
            editable_roots.append(url[len(FILE_PREFIX):])
    return editable_roots


if __name__ == "__main__":
    for root in sys.path:
        # The current working dir is not automatically included but can be if PYTHONPATH
        # contains an empty entry.
        if root != _helpers_root and root != os.curdir and root != _working_dir:
            print(root)

    for editable_root in get_pep660_editable_roots_from_metadata():
        print(editable_root)