Как ввести подсказку для оболочки класса или класса, метод которого назначен не в определении в Python3?

В некоторых случаях требуется внедрить один и тот же метод в несколько классов. Чтобы избежать повторения кода, я могу использовать декоратор класса или оболочку класса для назначения новых атрибутов и методов определенному классу. (Поскольку подсказка типа для оболочки и декоратора одинакова, в качестве примера я возьму декоратор)

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 ни на что не намекает, ни на новый метод, ни на оригиналы.

Интересно, есть ли способ прикрепить новые методы к существующему типу (классу) в подсказке типа.

Это общий вопрос. Ответ, который вам нужен, — это типы пересечений (type[T] & SomeProtocol), но в системе типов Python этого еще нет.

InSync 03.08.2024 06:45

это переработано. Просто введите method = Base.method в определениях классов.

juanpa.arrivillaga 03.08.2024 06:58

@juanpa.arrivillaga Ситуация такова, что у меня есть ряд классов, в которые необходимо внедрить точно такой же метод. Я подумал, что, возможно, было бы лучше избегать определения группы подклассов, определяющих одни и те же новые методы.

cby120 03.08.2024 07:42

@InSync Спасибо за ответ. Если это новая функция для Python, думаю, моя проблема на данный момент неразрешима?

cby120 03.08.2024 07:44

@cby120 cby120 вам не нужно, это одна дополнительная строка, которая ничего не повторяет, точно так же, как декоратор.

juanpa.arrivillaga 03.08.2024 07:54

Вам вообще нужны новые подклассы? cls.method = new_method; return cls.

chepner 03.08.2024 17:03

@chepner Конечно, это тоже работает. Мне нужен подкласс по другим причинам. Некоторые методы требуют, чтобы определенные имена были перехвачены фреймворком torch, и мне нужно обернуть некоторый оригинальный метод и вызвать их в обернутых новых методах, что довольно удобно с помощью super(). Это также можно реализовать путем переименования исходных методов.

cby120 04.08.2024 18:05
Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
0
7
50
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Вам не нужен декоратор, чтобы назначить атрибут классу или создать класс-оболочку. Самый простой способ сделать то, что вам нужно, — это просто назначить метод в области класса. Это также имеет то преимущество, что хорошо работает с инструментами статического анализа.

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 03.08.2024 08:30

@ cby120 cby120 это не совсем правильно и будет работать, если Bar является подклассом, а метод вызывается из экземпляра подкласса.

juanpa.arrivillaga 03.08.2024 09:03

Я провел тестирование. Это практическая реализация для ситуаций, когда требуется обработать несколько классов. Однако в ситуациях с очень разными классами было бы неплохо использовать функцию для перебора списка. Еще раз спасибо за ответ на мой вопрос.

cby120 03.08.2024 09:22

Другие вопросы по теме