@@ -97,6 +97,8 @@ def only_preserve_group(m: Match[str]) -> str:
97
97
google-cloud-foo will be used.
98
98
"""
99
99
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 ],
100
102
re .compile (r"""^azure-.+""" ): [all_hyphen_to_dot ],
101
103
re .compile (r"""^django-((.+(-.+)?))""" ): [first_group_hyphen_to_underscore ],
102
104
# 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:
126
128
"beautifulsoup4" : ("bs4" ,),
127
129
"bitvector" : ("BitVector" ,),
128
130
"cattrs" : ("cattr" , "cattrs" ),
129
- "django-filter" : ("django_filters" ,),
130
- "django-postgres-extra" : ("psqlextra" ,),
131
131
"django-cors-headers" : ("corsheaders" ,),
132
132
"django-countries" : ("django_countries" ,),
133
+ "django-filter" : ("django_filters" ,),
133
134
"django-fsm" : ("django_fsm" ,),
135
+ "django-migration-linter" : ("django_migration_linter" ,),
136
+ "django-object-actions" : ("django_object_actions" ,),
137
+ "django-postgres-extra" : ("psqlextra" ,),
134
138
"django-redis" : ("django_redis" ,),
139
+ "django-scim2" : ("django_scim" ,),
135
140
"djangorestframework" : ("rest_framework" ,),
141
+ "djangorestframework-api-key" : ("rest_framework_api_key" ,),
136
142
"djangorestframework-dataclasses" : ("rest_framework_dataclasses" ,),
143
+ "djangorestframework-queryfields" : ("drf_queryfields" ,),
137
144
"djangorestframework-simplejwt" : ("rest_framework_simplejwt" ,),
138
145
"elastic-apm" : ("elasticapm" ,),
139
146
"enum34" : ("enum" ,),
140
147
"factory-boy" : ("factory" ,),
141
148
"fluent-logger" : ("fluent" ,),
142
149
"gitpython" : ("git" ,),
150
+ "google-api-core" : ("google" ,),
151
+ "google-api-python-client" : ("googleapiclient" ,),
152
+ "google-auth" : ("google.auth" , "google.oauth2" ),
143
153
"graphql-core" : ("graphql" ,),
144
154
"grpcio" : ("grpc" ,),
145
155
"grpcio-health-checking" : ("grpc_health" ,),
156
+ "grpcio-reflection" : ("grpc_reflection" ,),
157
+ "honeycomb-beeline" : ("beeline" ,),
146
158
"honeycomb-opentelemetry" : ("honeycomb.opentelemetry" ,),
147
159
"ipython" : ("IPython" ,),
148
160
"jack-client" : ("jack" ,),
@@ -152,13 +164,14 @@ def only_preserve_group(m: Match[str]) -> str:
152
164
"mail-parser" : ("mailparser" ,),
153
165
"mysql-connector-python" : ("mysql.connector" ,),
154
166
"opencv-python" : ("cv2" ,),
167
+ "opencv-python-headless" : ("cv2" ,),
155
168
"opensearch-py" : ("opensearchpy" ,),
156
169
# opentelemetry
157
170
"opentelemetry-api" : ("opentelemetry" ,),
158
- "opentelemetry-instrumentation-kafka-python" : ("opentelemetry.instrumentation.kafka" ,),
159
171
"opentelemetry-exporter-otlp" : ("opentelemetry.exporter" ,),
160
172
"opentelemetry-exporter-otlp-proto-grpc" : ("opentelemetry.exporter.otlp.proto.grpc" ,),
161
173
"opentelemetry-exporter-otlp-proto-http" : ("opentelemetry.exporter.otlp.proto.http" ,),
174
+ "opentelemetry-instrumentation-kafka-python" : ("opentelemetry.instrumentation.kafka" ,),
162
175
"opentelemetry-sdk" : ("opentelemetry.sdk" ,),
163
176
"opentelemetry-test-utils" : ("opentelemetry.test" ,),
164
177
"paho-mqtt" : ("paho" ,),
@@ -169,33 +182,33 @@ def only_preserve_group(m: Match[str]) -> str:
169
182
"protobuf" : ("google.protobuf" ,),
170
183
"psycopg2-binary" : ("psycopg2" ,),
171
184
"pycrypto" : ("Crypto" ,),
172
- "pykube-ng" : ("pykube" ,),
173
- "pyhamcrest" : ("hamcrest" ,),
174
185
"pygithub" : ("github" ,),
175
186
"pygobject" : ("gi" ,),
187
+ "pyhamcrest" : ("hamcrest" ,),
176
188
"pyjwt" : ("jwt" ,),
189
+ "pykube-ng" : ("pykube" ,),
190
+ "pymongo" : ("bson" , "gridfs" , "pymongo" ),
191
+ "pymupdf" : ("fitz" ,),
177
192
"pyopenssl" : ("OpenSSL" ,),
178
193
"pypdf2" : ("PyPDF2" ,),
179
194
"pypi-kenlm" : ("kenlm" ,),
180
195
"pysocks" : ("socks" ,),
181
196
"pytest" : ("pytest" , "_pytest" ),
182
- "pyyaml" : ("yaml" ,),
183
- "pymongo" : ("bson" , "gridfs" , "pymongo" ),
184
- "pymupdf" : ("fitz" ,),
185
197
"pytest-runner" : ("ptr" ,),
186
198
"python-json-logger" : ("pythonjsonlogger" ,),
187
199
"python-levenshtein" : ("Levenshtein" ,),
188
200
"python-lsp-jsonrpc" : ("pylsp_jsonrpc" ,),
189
201
"pywinrm" : ("winrm" ,),
202
+ "pyyaml" : ("yaml" ,),
190
203
"randomwords" : ("random_words" ,),
191
204
"scikit-image" : ("skimage" ,),
192
205
"scikit-learn" : ("sklearn" ,),
193
206
"scikit-video" : ("skvideo" ,),
194
- "sseclient-py" : ("sseclient" ,),
195
207
"setuptools" : ("easy_install" , "pkg_resources" , "setuptools" ),
196
208
"snowflake-connector-python" : ("snowflake.connector" ,),
197
209
"snowflake-snowpark-python" : ("snowflake.snowpark" ,),
198
210
"snowflake-sqlalchemy" : ("snowflake.sqlalchemy" ,),
211
+ "sseclient-py" : ("sseclient" ,),
199
212
"strawberry-graphql" : ("strawberry" ,),
200
213
"streamlit-aggrid" : ("st_aggrid" ,),
201
214
"unleashclient" : ("UnleashClient" ,),
@@ -211,16 +224,18 @@ def only_preserve_group(m: Match[str]) -> str:
211
224
}
212
225
213
226
DEFAULT_TYPE_STUB_MODULE_MAPPING = {
227
+ "boto3-stubs-lite" : ("boto3" ,),
214
228
"djangorestframework-types" : ("rest_framework" ,),
229
+ "google-api-python-client-stubs" : ("googleapiclient" ,),
215
230
"lark-stubs" : ("lark" ,),
216
231
"types-beautifulsoup4" : ("bs4" ,),
217
232
"types-enum34" : ("enum34" ,),
218
233
"types-pillow" : ("PIL" ,),
219
234
"types-protobuf" : ("google.protobuf" ,),
220
235
"types-pycrypto" : ("Crypto" ,),
221
236
"types-pyopenssl" : ("OpenSSL" ,),
222
- "types-pyyaml" : ("yaml" ,),
223
237
"types-python-dateutil" : ("dateutil" ,),
238
+ "types-pyyaml" : ("yaml" ,),
224
239
"types-setuptools" : ("easy_install" , "pkg_resources" , "setuptools" ),
225
240
}
226
241
0 commit comments