Comment résoudre les conflits entre l'erreur de détection de type pylance et le décorateur personnalisé?

Solutions au conflit entre la vérification du type de pylance et le décorateur personnalisé

Dans le développement de Python, les outils de vérification de type statique (tels que Pylance) génèrent souvent des erreurs de vérification de type ou des avertissements avec des décorateurs personnalisés. Cet article utilisera un exemple pour démontrer comment résoudre le problème de décalage du type rapporté par Pylance dans un décorateur personnalisé.

Exemple de question:

L'extrait de code suivant montre un décorateur personnalisé execute et la fonction décorée query_data_source :

 Def Execute (FUNC):
    def inner_wrapper (* args, ** kwargs) -> résultat [any]: # Le problème 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: # Pylance estime que le type de retour ici est sélectionné
    stmt = (
        sélectionner(
            dataSource.id,
            dataSource.name,
            dataSource.source_url,
            dataSource.author,
            DataSource.Description,
            dataSource.cover_image_url,
            dataSource.start_date,
            dataSource.end_date,
        )
        .where (dataSource.id> = start_id)
        .limit (max_results_amount)
        .Order_By (DataSource.id)
    )
    retourner STMT

Il n'y a aucun problème avec le code lors de l'exécution, mais Pylance avertit l'inadéquation de type retour de query_data_source , car il infère que le décorateur execute modifie le type de retour pour result[any] .

Solution:

Pour résoudre ce problème, nous devons ajouter des conseils de type plus précis à la définition du décorateur pour clarifier le type de retour de la fonction de décoration. Le code modifié est le suivant:

 de la frappe d'import

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

    retourner inner_wrapper

En ajoutant l'invite de type Callable[..., Result] dans le décorateur execute , nous disons à Pylance de renvoyer Result . ... signifie que le nombre et le type de paramètres sont incertains, et Result suppose qu'il s'agit d'un type personnalisé ou d'un type importé. Cela résout l'erreur de vérification du type de Pylance et garantit que les informations de type sont exactes.

Cette méthode convient aux problèmes de vérification de type avec de nombreux décorateurs personnalisés. En ajoutant des invites de type appropriées, vous pouvez travailler efficacement avec des outils de vérification de type pour améliorer la qualité et la maintenabilité du code. N'oubliez pas de remplacer Result et session par les types et objets réels utilisés dans votre code en fonction de votre situation réelle.

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!