added more default modules

Author: purajit

src/python/pants/backend/python/dependency_inference/default_module_mapping.py

@@ -84,6 +84,8 @@ def two_groups_hyphens_two_replacements_with_suffix(
 google-cloud-foo will be used.
 """
 DEFAULT_MODULE_PATTERN_MAPPING: Dict[re.Pattern, Iterable] = {
+    re.compile(r"""^airflow-providers-(.+)"""): [all_hyphen_to_dot],
+    re.compile(r"""^apache-(airflow-providers-.+)"""): [first_group_hyphen_to_dot],
     re.compile(r"""^azure-.+"""): [all_hyphen_to_dot],
     re.compile(r"""^django-((.+(-.+)?))"""): [first_group_hyphen_to_underscore],
     # See https://github.com/googleapis/google-cloud-python#libraries for all Google cloud
@@ -113,23 +115,33 @@ def two_groups_hyphens_two_replacements_with_suffix(
     "beautifulsoup4": ("bs4",),
     "bitvector": ("BitVector",),
     "cattrs": ("cattr", "cattrs"),
-    "django-filter": ("django_filters",),
-    "django-postgres-extra": ("psqlextra",),
     "django-cors-headers": ("corsheaders",),
     "django-countries": ("django_countries",),
+    "django-filter": ("django_filters",),
     "django-fsm": ("django_fsm",),
+    "django-migration-linter": ("django_migration_linter",),
+    "django-object-actions": ("django_object_actions",),
+    "django-postgres-extra": ("psqlextra",),
     "django-redis": ("django_redis",),
+    "django-scim2": ("django_scim",),
     "djangorestframework": ("rest_framework",),
+    "djangorestframework-api-key": ("rest_framework_api_key",),
     "djangorestframework-dataclasses": ("rest_framework_dataclasses",),
+    "djangorestframework-queryfields": ("drf_queryfields",),
     "djangorestframework-simplejwt": ("rest_framework_simplejwt",),
     "elastic-apm": ("elasticapm",),
     "enum34": ("enum",),
     "factory-boy": ("factory",),
     "fluent-logger": ("fluent",),
     "gitpython": ("git",),
+    "google-api-core": ("google",),
+    "google-api-python-client": ("googleapiclient",),
+    "google-auth": ("google.auth", "google.oauth2"),
     "graphql-core": ("graphql",),
     "grpcio": ("grpc",),
     "grpcio-health-checking": ("grpc_health",),
+    "grpcio-reflection": ("grpc_reflection",),
+    "honeycomb-beeline": ("beeline",),
     "honeycomb-opentelemetry": ("honeycomb.opentelemetry",),
     "ipython": ("IPython",),
     "jack-client": ("jack",),
@@ -139,13 +151,14 @@ def two_groups_hyphens_two_replacements_with_suffix(
     "mail-parser": ("mailparser",),
     "mysql-connector-python": ("mysql.connector",),
     "opencv-python": ("cv2",),
+    "opencv-python-headless": ("cv2",),
     "opensearch-py": ("opensearchpy",),
     # opentelemetry
     "opentelemetry-api": ("opentelemetry",),
-    "opentelemetry-instrumentation-kafka-python": ("opentelemetry.instrumentation.kafka",),
     "opentelemetry-exporter-otlp": ("opentelemetry.exporter",),
     "opentelemetry-exporter-otlp-proto-grpc": ("opentelemetry.exporter.otlp.proto.grpc",),
     "opentelemetry-exporter-otlp-proto-http": ("opentelemetry.exporter.otlp.proto.http",),
+    "opentelemetry-instrumentation-kafka-python": ("opentelemetry.instrumentation.kafka",),
     "opentelemetry-sdk": ("opentelemetry.sdk",),
     "opentelemetry-test-utils": ("opentelemetry.test",),
     "paho-mqtt": ("paho",),
@@ -156,52 +169,56 @@ def two_groups_hyphens_two_replacements_with_suffix(
     "protobuf": ("google.protobuf",),
     "psycopg2-binary": ("psycopg2",),
     "pycrypto": ("Crypto",),
-    "pykube-ng": ("pykube",),
-    "pyhamcrest": ("hamcrest",),
     "pygithub": ("github",),
     "pygobject": ("gi",),
+    "pyhamcrest": ("hamcrest",),
     "pyjwt": ("jwt",),
+    "pykube-ng": ("pykube",),
+    "pymongo": ("bson", "gridfs", "pymongo"),
+    "pymupdf": ("fitz",),
     "pyopenssl": ("OpenSSL",),
     "pypdf2": ("PyPDF2",),
     "pypi-kenlm": ("kenlm",),
     "pysocks": ("socks",),
     "pytest": ("pytest", "_pytest"),
-    "pyyaml": ("yaml",),
-    "pymongo": ("bson", "gridfs", "pymongo"),
-    "pymupdf": ("fitz",),
     "pytest-runner": ("ptr",),
     "python-json-logger": ("pythonjsonlogger",),
     "python-levenshtein": ("Levenshtein",),
     "python-lsp-jsonrpc": ("pylsp_jsonrpc",),
     "pywinrm": ("winrm",),
+    "pyyaml": ("yaml",),
     "randomwords": ("random_words",),
     "scikit-image": ("skimage",),
     "scikit-learn": ("sklearn",),
     "scikit-video": ("skvideo",),
-    "sseclient-py": ("sseclient",),
     "setuptools": ("easy_install", "pkg_resources", "setuptools"),
     "snowflake-connector-python": ("snowflake.connector",),
     "snowflake-snowpark-python": ("snowflake.snowpark",),
     "snowflake-sqlalchemy": ("snowflake.sqlalchemy",),
+    "sseclient-py": ("sseclient",),
     "strawberry-graphql": ("strawberry",),
     "streamlit-aggrid": ("st_aggrid",),
     "unleashclient": ("UnleashClient",),
     "websocket-client": ("websocket",),
 }
 
-DEFAULT_TYPE_STUB_MODULE_PATTERN_MAPPING: Dict[re.Pattern, Iterable] = {}
+DEFAULT_TYPE_STUB_MODULE_PATTERN_MAPPING: Dict[re.Pattern, Iterable] = {
+    re.compile(r"""^mypy-boto3-.+"""): [all_hyphen_to_underscore],
+}
 
 DEFAULT_TYPE_STUB_MODULE_MAPPING = {
+    "boto3-stubs-lite": ("boto3",),
     "djangorestframework-types": ("rest_framework",),
+    "google-api-python-client-stubs": ("googleapiclient",),
     "lark-stubs": ("lark",),
     "types-beautifulsoup4": ("bs4",),
     "types-enum34": ("enum34",),
     "types-pillow": ("PIL",),
     "types-protobuf": ("google.protobuf",),
     "types-pycrypto": ("Crypto",),
     "types-pyopenssl": ("OpenSSL",),
-    "types-pyyaml": ("yaml",),
     "types-python-dateutil": ("dateutil",),
+    "types-pyyaml": ("yaml",),
     "types-setuptools": ("easy_install", "pkg_resources", "setuptools"),
 }