Plugin not working for decorated BaseModel classes

[OfekEngelClaroty]

Describe the bug Plugin looses signature when a BaseModel class is decorated.

To Reproduce

from typing import TypeVar

from pydantic import BaseModel

_T = TypeVar("_T")

def deco(obj: _T) -> _T:
    return obj

@deco
class Model(BaseModel):
    foo: int

if __name__ == '__main__':
    a = Model()

Expected behavior Plugin to infer type correctly and add auto completions to pycharm

Screenshots

Environments (please complete the following information):

• IDE: PyCharm 2022.3.2 (Professional Edition)
• OS: [e.g. macOS 12.6.3 ]
• Pydantic Version [e.g. 1.10.2 ]
• Plugin version [e.g. 0.4.0 ]

Additional context Without decorator, plugin works as inteneded

[koxudaxi]

Thank you for creating the issue. I will fix it.

[OfekEngelClaroty]

@koxudaxi is there an eta for fix?

[koxudaxi]

Last week, I checked the code. But I couldn't find a good solution at the time. I will investigate again.

[koxudaxi]

Your example doesn't work with @dataclass.dataclass too. It may be a hard fix without changing PyCharm side :thinking:

[OfekEngelClaroty]

perhaps its related to this issue