Создайте параметризованный тип в Python, но сделайте все экземпляры потомками «супертипа» - PullRequest
Новая
       44

Создайте параметризованный тип в Python, но сделайте все экземпляры потомками «супертипа»

0 голосов
/ 03 марта 2019

Предположим, у меня есть тип Python со свойством t.Я хочу создать «параметризованный метатип» таким образом, чтобы работало следующее: 

class MySuperClass(type):
    pass

class MySubClass(MySuperClass):
    # Here is the problem -- How do I define types that contain stuff,
    # independent of an object?
    def __init__(self, t): # Or __getitem__
        self.t = t

    def __instancecheck__(self, instance):
        return isinstance(instance, MySubClass) and instance.t == self.t

    def __subclasscheck__(self, subclass):
        return MySubClass in subclass.__mro__ and subclass.t == self.t

class MyObject(metaclass=MySubClass):
    def __init__(self, t):
        self.t = t

# Test code:
## Both of these, square brackets work too
assert isinstance(MyObject(0), MySubClass(0))
assert not isinstance(MyObject(0), MySubClass(1))

## Ideally
assert isinstance(MyObject(0), MySuperClass) or isinstance(MyObject(0), MySubClass)

В настоящее время я получаю следующую ошибку: 

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-4-99ad08881526> in <module>
     14         return MySubClass in subclass.__mro__ and subclass.t == self.t
     15 
---> 16 class MyObject(metaclass=MySubClass):
     17     def __init__(self, t):
     18         self.t = t

TypeError: __init__() takes 2 positional arguments but 4 were given

Ответы [ 2 ]

0 голосов
/ 04 марта 2019

Кажется, я нашел решение, и оно намного, намного чище, чем то, которое я изначально имел в виду. 

class MyMetaSuper(type):
    pass

class MyObject:
    def __init__(self, t):
        self.t = t

    def __class_getitem__(cls, key):        
        class MyMeta(MyMetaSuper):
            t = key
            def __instancecheck__(self, instance):
                return isinstance(instance, cls) and self.t == instance.t

            def __subclasscheck__(self, subclass):
                return isinstance(subclass, MyMetaSuper) and self.t == subclass.t

        class MyObjectSpecific(MyObject, metaclass=MyMeta):
            pass

        return MyObjectSpecific

# Check for specific condition
assert isinstance(MyObject(0), MyObject[0])
# Make sure isinstance fails when condition fails
assert not isinstance(MyObject(0), MyObject[1])

# Test the generic object
assert isinstance(MyObject(0), MyObject)
0 голосов
/ 03 марта 2019

Есть возможность удовлетворить первую часть или требование.Но для этого потребуется вспомогательный класс .MySubClass является потомком типа, MySubClass(0) должен быть классом.Достаточно создать внутренний класс InstanceChecker класс в MySubClass и поставить __instancecheck__ переопределить их.

Код может быть: 

class MySubClass(MySuperClass):
    def __new__(cls, name, bases=None, namespace=None, *args, **kwargs):
        if bases is not None:
            return super().__new__(cls, name, bases, namespace, **kwargs)
        return cls.InstanceChecker(name)

    class InstanceChecker:
        def __init__(self, t):
            self.t = t
        def __instancecheck__(self, instance):
            return isinstance(instance.__class__, MySubClass) and instance.t == self.t            

class MyObject(metaclass=MySubClass):
    def __init__(self, t):
        self.t = t

# Test code:
## Both of these, square brackets work too
assert isinstance(MyObject(0), MySubClass(0))
assert not isinstance(MyObject(0), MySubClass(1))

Кстати, я удалилпереопределение __subclasscheck__, поскольку t только в атрибуте instance в MyObject

Кроме того, метакласс может автоматически добавлять суперкласс в параметре bases,В следующем коде MySuperClass больше не является суперклассом MySubClass, а MyObject: 

class MySuperClass():
    pass


class MySubClass(type):
    def __new__(cls, name, bases=None, namespace=None, *args, **kwargs):
        if bases is not None:
            return super().__new__(cls, name, bases + (MySuperClass,), namespace, **kwargs)
        return cls.InstanceChecker(name)
    class InstanceChecker:
        def __init__(self, t):
            self.t = t
        def __instancecheck__(self, instance):
            return isinstance(instance.__class__, MySubClass) and instance.t == self.t

class MyObject(metaclass=MySubClass):
    def __init__(self, t):
        self.t = t

# Test code:
## Both of these, square brackets work too
assert isinstance(MyObject(0), MySubClass(0))
assert not isinstance(MyObject(0), MySubClass(1))

## Ideally
assert isinstance(MyObject(0), MySuperClass)
...