| from __future__ import annotations |
| |
| |
| def is_module_from_legacy_bundled_package(module: str) -> bool: |
| top_level = module.split(".", 1)[0] |
| return top_level in legacy_bundled_packages |
| |
| |
| def stub_distribution_name(module: str) -> str | None: |
| top_level = module.split(".", 1)[0] |
| |
| dist = legacy_bundled_packages.get(top_level) |
| if dist: |
| return dist |
| dist = non_bundled_packages_flat.get(top_level) |
| if dist: |
| return dist |
| |
| if top_level in non_bundled_packages_namespace: |
| namespace = non_bundled_packages_namespace[top_level] |
| components = module.split(".") |
| for i in range(len(components), 0, -1): |
| module = ".".join(components[:i]) |
| dist = namespace.get(module) |
| if dist: |
| return dist |
| |
| return None |
| |
| |
| # Stubs for these third-party packages used to be shipped with mypy. |
| # |
| # Map package name to PyPI stub distribution name. |
| legacy_bundled_packages: dict[str, str] = { |
| "aiofiles": "types-aiofiles", |
| "bleach": "types-bleach", |
| "cachetools": "types-cachetools", |
| "click_spinner": "types-click-spinner", |
| "croniter": "types-croniter", |
| "dateparser": "types-dateparser", |
| "dateutil": "types-python-dateutil", |
| "decorator": "types-decorator", |
| "deprecated": "types-Deprecated", |
| "docutils": "types-docutils", |
| "first": "types-first", |
| "markdown": "types-Markdown", |
| "mock": "types-mock", |
| "OpenSSL": "types-pyOpenSSL", |
| "paramiko": "types-paramiko", |
| "polib": "types-polib", |
| "pycurl": "types-pycurl", |
| "pymysql": "types-PyMySQL", |
| "pyrfc3339": "types-pyRFC3339", |
| "pytz": "types-pytz", |
| "requests": "types-requests", |
| "retry": "types-retry", |
| "simplejson": "types-simplejson", |
| "singledispatch": "types-singledispatch", |
| "six": "types-six", |
| "tabulate": "types-tabulate", |
| "toml": "types-toml", |
| "ujson": "types-ujson", |
| "waitress": "types-waitress", |
| "yaml": "types-PyYAML", |
| } |
| |
| # Map package name to PyPI stub distribution name from typeshed. |
| # Stubs for these packages were never bundled with mypy. Don't |
| # include packages that have a release that includes PEP 561 type |
| # information. |
| # |
| # Note that these packages are omitted for now: |
| # pika: typeshed's stubs are on PyPI as types-pika-ts. |
| # types-pika already exists on PyPI, and is more complete in many ways, |
| # but is a non-typeshed stubs package. |
| non_bundled_packages_flat: dict[str, str] = { |
| "_cffi_backend": "types-cffi", |
| "_win32typing": "types-pywin32", |
| "antlr4": "types-antlr4-python3-runtime", |
| "assertpy": "types-assertpy", |
| "atheris": "types-atheris", |
| "authlib": "types-Authlib", |
| "aws_xray_sdk": "types-aws-xray-sdk", |
| "boltons": "types-boltons", |
| "braintree": "types-braintree", |
| "bs4": "types-beautifulsoup4", |
| "bugbear": "types-flake8-bugbear", |
| "caldav": "types-caldav", |
| "capturer": "types-capturer", |
| "cffi": "types-cffi", |
| "chevron": "types-chevron", |
| "click_default_group": "types-click-default-group", |
| "click_log": "types-click-log", |
| "click_web": "types-click-web", |
| "colorama": "types-colorama", |
| "commctrl": "types-pywin32", |
| "commonmark": "types-commonmark", |
| "consolemenu": "types-console-menu", |
| "corus": "types-corus", # codespell:ignore corus |
| "cronlog": "types-python-crontab", |
| "crontab": "types-python-crontab", |
| "crontabs": "types-python-crontab", |
| "datemath": "types-python-datemath", |
| "dateparser_data": "types-dateparser", |
| "dde": "types-pywin32", |
| "defusedxml": "types-defusedxml", |
| "docker": "types-docker", |
| "dockerfile_parse": "types-dockerfile-parse", |
| "editdistance": "types-editdistance", |
| "entrypoints": "types-entrypoints", |
| "exifread": "types-ExifRead", |
| "fanstatic": "types-fanstatic", |
| "farmhash": "types-pyfarmhash", |
| "flake8_builtins": "types-flake8-builtins", |
| "flake8_docstrings": "types-flake8-docstrings", |
| "flake8_rst_docstrings": "types-flake8-rst-docstrings", |
| "flake8_simplify": "types-flake8-simplify", |
| "flake8_typing_imports": "types-flake8-typing-imports", |
| "flake8": "types-flake8", |
| "flask_cors": "types-Flask-Cors", |
| "flask_migrate": "types-Flask-Migrate", |
| "flask_socketio": "types-Flask-SocketIO", |
| "fpdf": "types-fpdf2", |
| "gdb": "types-gdb", |
| "gevent": "types-gevent", |
| "greenlet": "types-greenlet", |
| "hdbcli": "types-hdbcli", |
| "html5lib": "types-html5lib", |
| "httplib2": "types-httplib2", |
| "humanfriendly": "types-humanfriendly", |
| "hvac": "types-hvac", |
| "ibm_db": "types-ibm-db", |
| "icalendar": "types-icalendar", |
| "import_export": "types-django-import-export", |
| "influxdb_client": "types-influxdb-client", |
| "inifile": "types-inifile", |
| "isapi": "types-pywin32", |
| "jack": "types-JACK-Client", |
| "jenkins": "types-python-jenkins", |
| "Jetson": "types-Jetson.GPIO", |
| "jks": "types-pyjks", |
| "jmespath": "types-jmespath", |
| "jose": "types-python-jose", |
| "jsonschema": "types-jsonschema", |
| "jwcrypto": "types-jwcrypto", |
| "keyboard": "types-keyboard", |
| "ldap3": "types-ldap3", |
| "lupa": "types-lupa", |
| "lzstring": "types-lzstring", |
| "m3u8": "types-m3u8", |
| "mmapfile": "types-pywin32", |
| "mmsystem": "types-pywin32", |
| "mypy_extensions": "types-mypy-extensions", |
| "MySQLdb": "types-mysqlclient", |
| "nanoid": "types-nanoid", |
| "nanoleafapi": "types-nanoleafapi", |
| "netaddr": "types-netaddr", |
| "netifaces": "types-netifaces", |
| "networkx": "types-networkx", |
| "nmap": "types-python-nmap", |
| "ntsecuritycon": "types-pywin32", |
| "oauthlib": "types-oauthlib", |
| "objgraph": "types-objgraph", |
| "odbc": "types-pywin32", |
| "olefile": "types-olefile", |
| "openpyxl": "types-openpyxl", |
| "opentracing": "types-opentracing", |
| "parsimonious": "types-parsimonious", |
| "passlib": "types-passlib", |
| "passpy": "types-passpy", |
| "peewee": "types-peewee", |
| "pep8ext_naming": "types-pep8-naming", |
| "perfmon": "types-pywin32", |
| "pexpect": "types-pexpect", |
| "playhouse": "types-peewee", |
| "portpicker": "types-portpicker", |
| "psutil": "types-psutil", |
| "psycopg2": "types-psycopg2", |
| "pyasn1": "types-pyasn1", |
| "pyaudio": "types-pyaudio", |
| "pyautogui": "types-PyAutoGUI", |
| "pycocotools": "types-pycocotools", |
| "pyflakes": "types-pyflakes", |
| "pygit2": "types-pygit2", |
| "pygments": "types-Pygments", |
| "pyi_splash": "types-pyinstaller", |
| "PyInstaller": "types-pyinstaller", |
| "pynput": "types-pynput", |
| "pyscreeze": "types-PyScreeze", |
| "pysftp": "types-pysftp", |
| "pytest_lazyfixture": "types-pytest-lazy-fixture", |
| "python_http_client": "types-python-http-client", |
| "pythoncom": "types-pywin32", |
| "pythonwin": "types-pywin32", |
| "pywintypes": "types-pywin32", |
| "qrbill": "types-qrbill", |
| "qrcode": "types-qrcode", |
| "regex": "types-regex", |
| "regutil": "types-pywin32", |
| "reportlab": "types-reportlab", |
| "requests_oauthlib": "types-requests-oauthlib", |
| "RPi": "types-RPi.GPIO", |
| "s2clientprotocol": "types-s2clientprotocol", |
| "sass": "types-libsass", |
| "sassutils": "types-libsass", |
| "seaborn": "types-seaborn", |
| "send2trash": "types-Send2Trash", |
| "serial": "types-pyserial", |
| "servicemanager": "types-pywin32", |
| "setuptools": "types-setuptools", |
| "shapely": "types-shapely", |
| "slumber": "types-slumber", |
| "sspicon": "types-pywin32", |
| "str2bool": "types-str2bool", |
| "tensorflow": "types-tensorflow", |
| "tgcrypto": "types-TgCrypto", |
| "timer": "types-pywin32", |
| "toposort": "types-toposort", |
| "tqdm": "types-tqdm", |
| "translationstring": "types-translationstring", |
| "tree_sitter_languages": "types-tree-sitter-languages", |
| "ttkthemes": "types-ttkthemes", |
| "unidiff": "types-unidiff", |
| "untangle": "types-untangle", |
| "usersettings": "types-usersettings", |
| "uwsgi": "types-uWSGI", |
| "uwsgidecorators": "types-uWSGI", |
| "vobject": "types-vobject", |
| "webob": "types-WebOb", |
| "whatthepatch": "types-whatthepatch", |
| "win2kras": "types-pywin32", |
| "win32": "types-pywin32", |
| "win32api": "types-pywin32", |
| "win32clipboard": "types-pywin32", |
| "win32com": "types-pywin32", |
| "win32comext": "types-pywin32", |
| "win32con": "types-pywin32", |
| "win32console": "types-pywin32", |
| "win32cred": "types-pywin32", |
| "win32crypt": "types-pywin32", |
| "win32cryptcon": "types-pywin32", |
| "win32event": "types-pywin32", |
| "win32evtlog": "types-pywin32", |
| "win32evtlogutil": "types-pywin32", |
| "win32file": "types-pywin32", |
| "win32gui_struct": "types-pywin32", |
| "win32gui": "types-pywin32", |
| "win32help": "types-pywin32", |
| "win32inet": "types-pywin32", |
| "win32inetcon": "types-pywin32", |
| "win32job": "types-pywin32", |
| "win32lz": "types-pywin32", |
| "win32net": "types-pywin32", |
| "win32netcon": "types-pywin32", |
| "win32pdh": "types-pywin32", |
| "win32pdhquery": "types-pywin32", |
| "win32pipe": "types-pywin32", |
| "win32print": "types-pywin32", |
| "win32process": "types-pywin32", |
| "win32profile": "types-pywin32", |
| "win32ras": "types-pywin32", |
| "win32security": "types-pywin32", |
| "win32service": "types-pywin32", |
| "win32serviceutil": "types-pywin32", |
| "win32timezone": "types-pywin32", |
| "win32trace": "types-pywin32", |
| "win32transaction": "types-pywin32", |
| "win32ts": "types-pywin32", |
| "win32ui": "types-pywin32", |
| "win32uiole": "types-pywin32", |
| "win32verstamp": "types-pywin32", |
| "win32wnet": "types-pywin32", |
| "winerror": "types-pywin32", |
| "winioctlcon": "types-pywin32", |
| "winnt": "types-pywin32", |
| "winperf": "types-pywin32", |
| "winxpgui": "types-pywin32", |
| "winxptheme": "types-pywin32", |
| "workalendar": "types-workalendar", |
| "wtforms": "types-WTForms", |
| "wurlitzer": "types-wurlitzer", |
| "xdg": "types-pyxdg", |
| "xdgenvpy": "types-xdgenvpy", |
| "Xlib": "types-python-xlib", |
| "xmltodict": "types-xmltodict", |
| "zstd": "types-zstd", |
| "zxcvbn": "types-zxcvbn", |
| # Stub packages that are not from typeshed |
| # Since these can be installed automatically via --install-types, we have a high trust bar |
| # for additions here |
| "pandas": "pandas-stubs", # https://github.com/pandas-dev/pandas-stubs |
| "lxml": "lxml-stubs", # https://github.com/lxml/lxml-stubs |
| "scipy": "scipy-stubs", # https://github.com/scipy/scipy-stubs |
| } |
| |
| |
| non_bundled_packages_namespace: dict[str, dict[str, str]] = { |
| "backports": {"backports.ssl_match_hostname": "types-backports.ssl_match_hostname"}, |
| "google": {"google.cloud.ndb": "types-google-cloud-ndb", "google.protobuf": "types-protobuf"}, |
| "paho": {"paho.mqtt": "types-paho-mqtt"}, |
| } |
