Типизация: как рассматривать аргументы класса, обернутые партиалом? ⇐ Python

Программы на Python
Anonymous
Типизация: как рассматривать аргументы класса, обернутые партиалом?

Сообщение Anonymous »

У меня есть класс MyClass, который ожидает класс Parent_cls в качестве аргумента, который заполняет интерфейс ParentInterface.
ChildA реализует/расширяет ParentInterface. Поскольку внутри MyClass родительский_cls создается с аргументами a и b, дополнительный аргумент c ChildA частично создается снаружи.
В принципе, это работает в Python. Однако я получаю предупреждение типа от Pycharm:
Изображение

Есть идеи, как исправить это предупреждение?
from abc import ABC, abstractmethod
from functools import partial
from typing import Type, Optional, Callable, cast

class ParentInterface(ABC):
def __init__(self, a: int, b: int):
self.a = a
self.b = b

@abstractmethod
def do_something(self):
pass

class ChildA(ParentInterface):
def __init__(self, a: int, b: int, c: str):
super().__init__(a, b)
self.c = c

def do_something(self):
print('I am ChildA')

# update 1
class ChildB(ParentInterface):
def __init__(self, a: int, b: int):
super().__init__(a, b)

def do_something(self):
print('I am ChildB')

class MyClass:
def __init__(self, parent_cls: Type[ParentInterface]):
self.parent = parent_cls(3, 4)

# alternative
# class MyClass:
# def __init__(self, parent_cls: Callable[[int, int], ParentInterface]):
# self.parent = parent_cls(3, 4)

def typed_partial(cls, *args, **kwargs):
return cast(Type[cls], partial(cls, *args, **kwargs))

# original code
# child_a_cls = partial(ChildA, c='some string')
# solution
child_a_cls = typed_partial(ChildA, c='some string')

my_class_with_childa = MyClass(parent_cls=child_a_cls)
my_class_with_childb = MyClass(parent_cls=ChildB)


Подробнее здесь: https://stackoverflow.com/questions/611 ... th-partial

Вернуться в «Python»