Python, types immuables

Question

sous-classement

J'ai la classe suivante:

class MySet(set): 

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

Cela fonctionne comme prévu (initialisant l'ensemble avec les mots de la chaîne plutôt que les lettres). Cependant, quand je veux faire la même chose avec la version immuable de jeu, la méthode __init__ semble être ignoré:

class MySet(frozenset): 

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

Puis-je obtenir quelque chose de similaire avec __new__?

Answer 1

Oui, vous devez remplacer __new__ méthode spéciale:

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

Et la sortie est:

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