bazel-contrib / bazel-contrib/rules_python
Allow specifying custom stubs for Gazelle
- Langage dominant
- Starlark
- Étoiles
- 688
- Forks
- 721
- Merge moyen
- 15 h 7 min
- PR mergées (30 j)
- 76
Description
# 🚀 feature request
### Description
If a library does not provide type stubs, and if there isn't already a type stub library, you might need to write your own stubs (e.g. [kafka-python](https://pypi.org/project/kafka-python/)). Doing so is easy enough:
```py
py_library(
name = "kafka",
imports = ["."],
pyi_srcs = [
"kafka/__init__.pyi",
"kafka/consumer/fetcher.pyi",
"kafka/errors.pyi",
],
visibility = ["//visibility:public"],
)
```
Gazelle currently has a feature to automatically add stubs for a dependency (e.g. automatically add `types-request` whenever `request` is a dependency), but there's no way to register this target as being the type stubs for the `@pypi//kafka` library. Furthermore, adding it manually will remove it without `# keep`
### Describe the solution you'd like
First, in `gazelle_python.yaml`, add a `stub_mapping` section that explicitly maps package to stubs
```yaml
stub_mapping:
requests: types_requests
```
Then instead of [guessing which type packages to include](https://github.com/bazel-contrib/rules_python/blob/bb2aad2d1e3f883c9cdc2264e0b4a2815233db57/gazelle/python/resolve.go#L258-L269), refer to this explicit mapping.
After this refactor, update `modules_mapping` to register additional stubs:
```py
modules_mapping(
name = "modules_map",
include_stub_packages = True,
wheels = all_whl_requirements,
extra_stubs = {
"kafka-python": "//my_stubs:kafka",
},
)
```
This would get added to the new `stub_mapping` section in `gazelle_python.yaml` and everything should Just Work.
### Describe alternatives you've considered
Guide de contribution
Ouvrir le guide de contribution
Piste de recherche
Commencez par gazelle_python.yaml et la logique de sélection des packages de types dans gazelle/python/resolve.go, en particulier les lignes liées. Suivez ensuite modules_mapping et ses options include_stub_packages et extra_stubs. Le travail est terminé lorsqu’un stub_mapping explicite peut enregistrer des stubs personnalisés et que Gazelle conserve et utilise ce mapping au lieu de deviner les packages de types.
Rédigé par le modèle d'indexation à partir du texte de l'issue.
Évaluation
- Stack technique
- python
- Domaine
- build-system
- Type d'issue
- Fonctionnalité
- Difficulté
- 5/5
- Temps estimé
- Plus d'une semaine
- Activité
- À l'abandon
- Clarté
- Plutôt claire
- Accessibilité débutants
- 35/100