Início / coding / Perguntas / 78208034
Accepted
Mark
Asked: 2024-03-23 01:48:34 +0800 CST

Como obter anotação de tipo de retorno de propriedade?

  • 772

Estou procurando uma maneira de extrair a anotação do tipo de retorno de uma propriedade de uma classe, por exemplo:

class Test:
    @property
    def foo(self) -> int:
        return 1

print(return_type_extract(Test, 'foo')) # <class 'int'>

Para métodos padrão, isso pode ser feito assim:

import inspect
class Test:
    def foo(self) -> int:
        return 1

def return_type_extract(cls: type, method_name: str) -> type:
    method = getattr(cls, method_name)
    return inspect.signature(method).return_annotation

print(return_type_extract(Test, 'foo')) # <class 'int'>

Este método, entretanto, não funciona para o @propertydecorador, pois gera um erro interno inspect.signature. Procurei um equivalente para propriedades na inspectbiblioteca, mas até agora não tive sorte.

python

1 respostas

  1. Best Answer

    Se o atributo recuperado for um objeto de propriedade, podemos acessar seu método getter ( .fget) para obter a função subjacente. Em seguida, usamos get_type_hints()para extrair a anotação do tipo de retorno da função subjacente.

    Assim :

    from typing import get_type_hints

class Test:
    @property
    def foo(self) -> int:
        return 1

def return_type_extract(cls: object, property_name: str) -> type:
    property_obj = getattr(cls, property_name)
    if isinstance(property_obj, property):
        property_obj = property_obj.fget
    return get_type_hints(property_obj).get('return')

print(return_type_extract(Test, 'foo'))  # <class 'int'>
    • 2

relate perguntas