В некоторых случаях требуется внедрить один и тот же метод в несколько классов. Чтобы избежать повторения кода, я могу использовать декоратор класса или оболочку класса для назначения новых атрибутов и методов определенному классу. (Поскольку подсказка типа для оболочки и декоратора одинакова, в качестве примера я возьму декоратор)
class Base:
pass
# decorator
def decorator(cls: Input_Type) -> Output_Type:
def new_method(self):
pass
cls.new_method = new_method
return cls
# wrapper
def wrapper(cls: Input_Type) -> Output_Type:
class WrappedClass(cls):
def new_method(self):
pass
return WrappedClass
Более того, для удобства и лучшей ремонтопригодности я хочу добавить подсказку типа для этого декоратора/обертки (Output_Type
в блоке кода). По типу подсказки IDE должна иметь возможность выполнять статический анализ и подсказки как для исходного класса attr, так и для вновь введенного attr. Как мне сделать подсказку типа?
python==3.8.10
VScode==1.91.1
# VScode Extensions
Pylance==v2024.7.1
Python-Type-Hint==v1.5.1
Я попробовал typing.TypeVar
, лайк
import typing as t
T = t.TypeVar("T", bound=t.Type)
def func(cls: T) -> T:
...
Однако в этом случае тип возвращаемого значения будет точно таким же, как тип ввода, и новые назначенные методы не будут подсказываться.
Я также пробовал использовать наследование от typing.Generic
и определять класс протокола, например
import typing as t
T = t.TypeVar("T", bound=t.Type)
class Injected(t.Generic[T]):
def new_method(self): ...
def func(cls: T) -> Injected[T]:
...
Но при этом IDE ни на что не намекает, ни на новый метод, ни на оригиналы.
Интересно, есть ли способ прикрепить новые методы к существующему типу (классу) в подсказке типа.
это переработано. Просто введите method = Base.method
в определениях классов.
@juanpa.arrivillaga Ситуация такова, что у меня есть ряд классов, в которые необходимо внедрить точно такой же метод. Я подумал, что, возможно, было бы лучше избегать определения группы подклассов, определяющих одни и те же новые методы.
@InSync Спасибо за ответ. Если это новая функция для Python, думаю, моя проблема на данный момент неразрешима?
@cby120 cby120 вам не нужно, это одна дополнительная строка, которая ничего не повторяет, точно так же, как декоратор.
Вам вообще нужны новые подклассы? cls.method = new_method; return cls
.
@chepner Конечно, это тоже работает. Мне нужен подкласс по другим причинам. Некоторые методы требуют, чтобы определенные имена были перехвачены фреймворком torch, и мне нужно обернуть некоторый оригинальный метод и вызвать их в обернутых новых методах, что довольно удобно с помощью super(). Это также можно реализовать путем переименования исходных методов.
Вам не нужен декоратор, чтобы назначить атрибут классу или создать класс-оболочку. Самый простой способ сделать то, что вам нужно, — это просто назначить метод в области класса. Это также имеет то преимущество, что хорошо работает с инструментами статического анализа.
def method_implementation(self, arg: int) -> set[int]:
"""
a great docstring
"""
return set()
class Foo:
method = method_implementation
class Bar:
method = method_implementation
def another_method(self, x:str) -> None:
print(x)
bar = Bar()
bar.method("foo")
И видите, инструменты статического анализа и IDE это поймут.
Вот mypy
правильная ошибка:
(py312) Juans-MBP:test juan$ mypy foo.py
foo.py:17: error: Argument 1 has incompatible type "str"; expected "int" [arg-type]
Found 1 error in 1 file (checked 1 source file)
И IDE intellisense это поймет, по крайней мере, используя pylance с VSCode:
Большое спасибо за ответ. Если мне нужно вызвать метод из родительского класса, я думаю, я могу сделать это в функции ``` def метод(self): super(self.__class__, self).method() ```, я попробую.
@ cby120 cby120 это не совсем правильно и будет работать, если Bar
является подклассом, а метод вызывается из экземпляра подкласса.
Я провел тестирование. Это практическая реализация для ситуаций, когда требуется обработать несколько классов. Однако в ситуациях с очень разными классами было бы неплохо использовать функцию для перебора списка. Еще раз спасибо за ответ на мой вопрос.
Это общий вопрос. Ответ, который вам нужен, — это типы пересечений (
type[T] & SomeProtocol
), но в системе типов Python этого еще нет.