Python, tipi sottoclasse immutabili

Question

Sono la seguente classe:

class MySet(set): 

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

Questo funziona come previsto (inizializzazione del set con le parole della stringa piuttosto che le lettere). Tuttavia quando voglio fare lo stesso con la versione immutabile di set, il metodo __init__ sembra essere ignorato:

class MySet(frozenset): 

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

Posso realizzare qualcosa di simile con __new__?

Answer 1

Sì, è necessario eseguire l'override __new__ metodo speciale:

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') 

e l'uscita è:

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