Python, подклассы неизменяемых типов - PullRequest
Новая
       20

Python, подклассы неизменяемых типов

13 голосов
/ 25 июня 2010

У меня есть следующий класс: 

class MySet(set):

    def __init__(self, arg=None):
        if isinstance(arg, basestring):
            arg = arg.split()
        set.__init__(self, arg)

Это работает как ожидалось (инициализация набора со словами строки, а не с буквами).Однако, когда я хочу сделать то же самое с неизменяемой версией набора, метод __init__, кажется, игнорируется: 

class MySet(frozenset):

    def __init__(self, arg=None):
        if isinstance(arg, basestring):
            arg = arg.split()
        frozenset.__init__(self, arg)

Можно ли добиться чего-то подобного с __new__?

1 Ответ

12 голосов
/ 25 июня 2010

Да, вам нужно переопределить __new__ специальный метод: 

class MySet(frozenset):

    def __new__(cls, *args):
        if args and isinstance (args[0], basestring):
            args = (args[0].split (),) + args[1:]
        return super (MySet, cls).__new__(cls, *args)

print MySet ('foo bar baz')

И вывод: 

MySet(['baz', 'foo', 'bar'])
...