Comment résoudre le problème de la détection de type pylance des décorateurs personnalisés dans Python?

Conflit et solution entre pylance et invites de type décorateur personnalisé de python

Les décorateurs de Python sont des outils de réutilisation de code puissants, mais lors de l'utilisation de décorateurs personnalisés, les vérificateurs de type statique (tels que Pylance) peuvent ressentir des erreurs d'invite de type, en particulier lorsque le décorateur modifie le type de retour de la fonction. Cet article démontrera un problème et une solution communs.

Problème: Pylance ne peut pas identifier correctement le type de retour de la fonction modifié par un décorateur personnalisé. Par exemple, un décorateur modifie le type de retour d'une fonction, mais Pylance affiche toujours le type de retour de la fonction d'origine, ce qui donne un avertissement de type.

Exemple de code:

 Def Execute (FUNC):
    def inner_wrapper (* args, ** kwargs) -> résultat [any]: # le problème de pylance réside dans session.begin () comme session:
            result = session.execute (func (* args, ** kwargs)))
            Résultat de retour
    retourner inner_wrapper

@exécuter
def query_data_source (start_id: int = 1, max_results_amount: int = 10) -> select:
    stmt = select (
        dataSource.id,
        dataSource.name,
        dataSource.source_url,
        dataSource.author,
        DataSource.Description,
        dataSource.cover_image_url,
        dataSource.start_date,
        dataSource.end_date,
    .
    retourner STMT

La fonction query_data_source renvoie en fait result[any] type, mais Pylance le reconnaît toujours comme select , augmentant un avertissement de type.

Solution: Utilisez typing.Callable pour déclarer le type de retour du décorateur plus précisément, aidant ainsi Pylance à comprendre correctement le comportement du décorateur.

Code modifié:

 de la frappe d'import

def execute (func: callable [..., any]) -> callable [..., résultat [any]]: # use typing. callable
    def inner_wrapper (* args, ** kwargs) -> résultat [any]:
        avec session.begin () comme session:
            result = session.execute (func (* args, ** kwargs)))
            Résultat de retour
    retourner inner_wrapper

@exécuter
def query_data_source (start_id: int = 1, max_results_amount: int = 10) -> select:
    stmt = select (
        dataSource.id,
        dataSource.name,
        dataSource.source_url,
        dataSource.author,
        DataSource.Description,
        dataSource.cover_image_url,
        dataSource.start_date,
        dataSource.end_date,
    .
    retourner STMT

En utilisant Callable[..., Result[Any]] Comme l'invite de type de retour dans le décorateur execute , Pylance peut déduire avec précision le type de retour réel de la fonction query_data_source , éliminant ainsi l'avertissement de type. ... signifie que le nombre de paramètres est variable Any signifie que le type de paramètre est variable. Assurez-vous que Result et les types select sont correctement définis.

Cette approche résout efficacement les limites de l'inférence du type de retour de Pylance lors de la gestion des décorateurs personnalisés, améliorant ainsi la lisibilité et la maintenabilité du code.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!