Skip to content

Commit c15631b

Browse files
committed
add more module mappings for popular packages
1 parent 882ffe6 commit c15631b

File tree

1 file changed

+31
-10
lines changed

1 file changed

+31
-10
lines changed

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

+31-10
Original file line numberDiff line numberDiff line change
@@ -83,23 +83,33 @@ def two_groups_hyphens_two_replacements_with_suffix(
8383
"beautifulsoup4": ("bs4",),
8484
"bitvector": ("BitVector",),
8585
"cattrs": ("cattr", "cattrs"),
86-
"django-filter": ("django_filters",),
87-
"django-postgres-extra": ("psqlextra",),
8886
"django-cors-headers": ("corsheaders",),
8987
"django-countries": ("django_countries",),
88+
"django-filter": ("django_filters",),
9089
"django-fsm": ("django_fsm",),
90+
"django-object-actions": ("django_object_actions",),
91+
"django-postgres-extra": ("psqlextra",),
9192
"django-redis": ("django_redis",),
93+
"django-scim2": ("django_scim",),
9294
"djangorestframework": ("rest_framework",),
95+
"djangorestframework-api-key": ("rest_framework_api_key",),
9396
"djangorestframework-dataclasses": ("rest_framework_dataclasses",),
97+
"djangorestframework-queryfields": ("drf_queryfields",),
9498
"djangorestframework-simplejwt": ("rest_framework_simplejwt",),
9599
"elastic-apm": ("elasticapm",),
96100
"enum34": ("enum",),
97101
"factory-boy": ("factory",),
98102
"fluent-logger": ("fluent",),
99103
"gitpython": ("git",),
104+
"google-api-python-client": ("googleapiclient",),
105+
"google-auth": (
106+
"google.auth",
107+
"google.oauth2",
108+
),
100109
"graphql-core": ("graphql",),
101110
"grpcio": ("grpc",),
102111
"grpcio-health-checking": ("grpc_health",),
112+
"grpcio-reflection": ("grpc_reflection",),
103113
"honeycomb-opentelemetry": ("honeycomb.opentelemetry",),
104114
"ipython": ("IPython",),
105115
"jack-client": ("jack",),
@@ -109,14 +119,25 @@ def two_groups_hyphens_two_replacements_with_suffix(
109119
"mail-parser": ("mailparser",),
110120
"mysql-connector-python": ("mysql.connector",),
111121
"opencv-python": ("cv2",),
122+
"opencv-python-headless": ("cv2",),
112123
"opensearch-py": ("opensearchpy",),
113124
# opentelemetry
114125
"opentelemetry-api": ("opentelemetry",),
115-
"opentelemetry-instrumentation-kafka-python": ("opentelemetry.instrumentation.kafka",),
116126
"opentelemetry-exporter-otlp": ("opentelemetry.exporter",),
117127
"opentelemetry-exporter-otlp-proto-grpc": ("opentelemetry.exporter.otlp.proto.grpc",),
118128
"opentelemetry-exporter-otlp-proto-http": ("opentelemetry.exporter.otlp.proto.http",),
119-
"opentelemetry-sdk": ("opentelemetry.sdk",),
129+
"opentelemetry-instrumentation-kafka-python": ("opentelemetry.instrumentation.kafka",),
130+
"opentelemetry-sdk": (
131+
"opentelemetry._logs",
132+
"opentelemetry.baggage",
133+
"opentelemetry.context",
134+
"opentelemetry.environment_variables",
135+
"opentelemetry.metrics",
136+
"opentelemetry.propagate",
137+
"opentelemetry.propagators",
138+
"opentelemetry.sdk",
139+
"opentelemetry.trace",
140+
),
120141
"opentelemetry-test-utils": ("opentelemetry.test",),
121142
"paho-mqtt": ("paho",),
122143
"phonenumberslite": ("phonenumbers",),
@@ -126,33 +147,33 @@ def two_groups_hyphens_two_replacements_with_suffix(
126147
"protobuf": ("google.protobuf",),
127148
"psycopg2-binary": ("psycopg2",),
128149
"pycrypto": ("Crypto",),
129-
"pykube-ng": ("pykube",),
130-
"pyhamcrest": ("hamcrest",),
131150
"pygithub": ("github",),
132151
"pygobject": ("gi",),
152+
"pyhamcrest": ("hamcrest",),
133153
"pyjwt": ("jwt",),
154+
"pykube-ng": ("pykube",),
155+
"pymongo": ("bson", "gridfs", "pymongo"),
156+
"pymupdf": ("fitz",),
134157
"pyopenssl": ("OpenSSL",),
135158
"pypdf2": ("PyPDF2",),
136159
"pypi-kenlm": ("kenlm",),
137160
"pysocks": ("socks",),
138161
"pytest": ("pytest", "_pytest"),
139-
"pyyaml": ("yaml",),
140-
"pymongo": ("bson", "gridfs", "pymongo"),
141-
"pymupdf": ("fitz",),
142162
"pytest-runner": ("ptr",),
143163
"python-json-logger": ("pythonjsonlogger",),
144164
"python-levenshtein": ("Levenshtein",),
145165
"python-lsp-jsonrpc": ("pylsp_jsonrpc",),
146166
"pywinrm": ("winrm",),
167+
"pyyaml": ("yaml",),
147168
"randomwords": ("random_words",),
148169
"scikit-image": ("skimage",),
149170
"scikit-learn": ("sklearn",),
150171
"scikit-video": ("skvideo",),
151-
"sseclient-py": ("sseclient",),
152172
"setuptools": ("easy_install", "pkg_resources", "setuptools"),
153173
"snowflake-connector-python": ("snowflake.connector",),
154174
"snowflake-snowpark-python": ("snowflake.snowpark",),
155175
"snowflake-sqlalchemy": ("snowflake.sqlalchemy",),
176+
"sseclient-py": ("sseclient",),
156177
"strawberry-graphql": ("strawberry",),
157178
"streamlit-aggrid": ("st_aggrid",),
158179
"unleashclient": ("UnleashClient",),

0 commit comments

Comments
 (0)