Skip to content

Commit 6a1e8ad

Browse files
committed
added more default modules
1 parent 1550337 commit 6a1e8ad

File tree

1 file changed

+25
-10
lines changed

1 file changed

+25
-10
lines changed

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

+25-10
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,8 @@ def only_preserve_group(m: Match[str]) -> str:
9797
google-cloud-foo will be used.
9898
"""
9999
DEFAULT_MODULE_PATTERN_MAPPING: Dict[re.Pattern, Iterable] = {
100+
re.compile(r"""^airflow-providers-(.+)"""): [all_hyphen_to_dot],
101+
re.compile(r"""^apache-(airflow-providers-.+)"""): [first_group_hyphen_to_dot],
100102
re.compile(r"""^azure-.+"""): [all_hyphen_to_dot],
101103
re.compile(r"""^django-((.+(-.+)?))"""): [first_group_hyphen_to_underscore],
102104
# See https://github.com/googleapis/google-cloud-python#libraries for all Google cloud
@@ -126,23 +128,33 @@ def only_preserve_group(m: Match[str]) -> str:
126128
"beautifulsoup4": ("bs4",),
127129
"bitvector": ("BitVector",),
128130
"cattrs": ("cattr", "cattrs"),
129-
"django-filter": ("django_filters",),
130-
"django-postgres-extra": ("psqlextra",),
131131
"django-cors-headers": ("corsheaders",),
132132
"django-countries": ("django_countries",),
133+
"django-filter": ("django_filters",),
133134
"django-fsm": ("django_fsm",),
135+
"django-migration-linter": ("django_migration_linter",),
136+
"django-object-actions": ("django_object_actions",),
137+
"django-postgres-extra": ("psqlextra",),
134138
"django-redis": ("django_redis",),
139+
"django-scim2": ("django_scim",),
135140
"djangorestframework": ("rest_framework",),
141+
"djangorestframework-api-key": ("rest_framework_api_key",),
136142
"djangorestframework-dataclasses": ("rest_framework_dataclasses",),
143+
"djangorestframework-queryfields": ("drf_queryfields",),
137144
"djangorestframework-simplejwt": ("rest_framework_simplejwt",),
138145
"elastic-apm": ("elasticapm",),
139146
"enum34": ("enum",),
140147
"factory-boy": ("factory",),
141148
"fluent-logger": ("fluent",),
142149
"gitpython": ("git",),
150+
"google-api-core": ("google",),
151+
"google-api-python-client": ("googleapiclient",),
152+
"google-auth": ("google.auth", "google.oauth2"),
143153
"graphql-core": ("graphql",),
144154
"grpcio": ("grpc",),
145155
"grpcio-health-checking": ("grpc_health",),
156+
"grpcio-reflection": ("grpc_reflection",),
157+
"honeycomb-beeline": ("beeline",),
146158
"honeycomb-opentelemetry": ("honeycomb.opentelemetry",),
147159
"ipython": ("IPython",),
148160
"jack-client": ("jack",),
@@ -152,13 +164,14 @@ def only_preserve_group(m: Match[str]) -> str:
152164
"mail-parser": ("mailparser",),
153165
"mysql-connector-python": ("mysql.connector",),
154166
"opencv-python": ("cv2",),
167+
"opencv-python-headless": ("cv2",),
155168
"opensearch-py": ("opensearchpy",),
156169
# opentelemetry
157170
"opentelemetry-api": ("opentelemetry",),
158-
"opentelemetry-instrumentation-kafka-python": ("opentelemetry.instrumentation.kafka",),
159171
"opentelemetry-exporter-otlp": ("opentelemetry.exporter",),
160172
"opentelemetry-exporter-otlp-proto-grpc": ("opentelemetry.exporter.otlp.proto.grpc",),
161173
"opentelemetry-exporter-otlp-proto-http": ("opentelemetry.exporter.otlp.proto.http",),
174+
"opentelemetry-instrumentation-kafka-python": ("opentelemetry.instrumentation.kafka",),
162175
"opentelemetry-sdk": ("opentelemetry.sdk",),
163176
"opentelemetry-test-utils": ("opentelemetry.test",),
164177
"paho-mqtt": ("paho",),
@@ -169,33 +182,33 @@ def only_preserve_group(m: Match[str]) -> str:
169182
"protobuf": ("google.protobuf",),
170183
"psycopg2-binary": ("psycopg2",),
171184
"pycrypto": ("Crypto",),
172-
"pykube-ng": ("pykube",),
173-
"pyhamcrest": ("hamcrest",),
174185
"pygithub": ("github",),
175186
"pygobject": ("gi",),
187+
"pyhamcrest": ("hamcrest",),
176188
"pyjwt": ("jwt",),
189+
"pykube-ng": ("pykube",),
190+
"pymongo": ("bson", "gridfs", "pymongo"),
191+
"pymupdf": ("fitz",),
177192
"pyopenssl": ("OpenSSL",),
178193
"pypdf2": ("PyPDF2",),
179194
"pypi-kenlm": ("kenlm",),
180195
"pysocks": ("socks",),
181196
"pytest": ("pytest", "_pytest"),
182-
"pyyaml": ("yaml",),
183-
"pymongo": ("bson", "gridfs", "pymongo"),
184-
"pymupdf": ("fitz",),
185197
"pytest-runner": ("ptr",),
186198
"python-json-logger": ("pythonjsonlogger",),
187199
"python-levenshtein": ("Levenshtein",),
188200
"python-lsp-jsonrpc": ("pylsp_jsonrpc",),
189201
"pywinrm": ("winrm",),
202+
"pyyaml": ("yaml",),
190203
"randomwords": ("random_words",),
191204
"scikit-image": ("skimage",),
192205
"scikit-learn": ("sklearn",),
193206
"scikit-video": ("skvideo",),
194-
"sseclient-py": ("sseclient",),
195207
"setuptools": ("easy_install", "pkg_resources", "setuptools"),
196208
"snowflake-connector-python": ("snowflake.connector",),
197209
"snowflake-snowpark-python": ("snowflake.snowpark",),
198210
"snowflake-sqlalchemy": ("snowflake.sqlalchemy",),
211+
"sseclient-py": ("sseclient",),
199212
"strawberry-graphql": ("strawberry",),
200213
"streamlit-aggrid": ("st_aggrid",),
201214
"unleashclient": ("UnleashClient",),
@@ -211,16 +224,18 @@ def only_preserve_group(m: Match[str]) -> str:
211224
}
212225

213226
DEFAULT_TYPE_STUB_MODULE_MAPPING = {
227+
"boto3-stubs-lite": ("boto3",),
214228
"djangorestframework-types": ("rest_framework",),
229+
"google-api-python-client-stubs": ("googleapiclient",),
215230
"lark-stubs": ("lark",),
216231
"types-beautifulsoup4": ("bs4",),
217232
"types-enum34": ("enum34",),
218233
"types-pillow": ("PIL",),
219234
"types-protobuf": ("google.protobuf",),
220235
"types-pycrypto": ("Crypto",),
221236
"types-pyopenssl": ("OpenSSL",),
222-
"types-pyyaml": ("yaml",),
223237
"types-python-dateutil": ("dateutil",),
238+
"types-pyyaml": ("yaml",),
224239
"types-setuptools": ("easy_install", "pkg_resources", "setuptools"),
225240
}
226241

0 commit comments

Comments
 (0)