Всем привет! Ищу способ динамически описать аттрибутом инстанса класса возвращаемое

Question

Всем привет! Ищу способ динамически описать аттрибутом инстанса класса возвращаемое

значение у его метода

Например:

class Foo:
def __init__(model: BaseModel):
self.model = model

def find_model(query) -> self.model:
return db.get(query)

А потом использовать это вот так:

1. bar = Foo(model=MyModel)
2. obj = bar.find_model('my_query')
3. obj.

1. Передаю класс (не объект класса), который наследуется от BaseModel
2. Получаю объект класса MyModel
3. получаю список всех методов и аттрибутов объекта класса MyModel в пайчарме

В вопросе на стаковерфлоу я ещё показал как это делается на джаве

Я видел ответы с TypeVar и ClassVar, но вообще не понял, нафига это нужно, потому что пайчарм всё равно не индексировал аттрибуты с методами

Может, кто-то пытался сделать что-то подобное?