diccionario Python retorno pidió clave si el valor no existe

Question

Busco una manera fácil de ser capaz de obtener un valor de un diccionario, y si no está allí, devolver la llave que el usuario se ha pasado.

por ejemplo:

>>> lookup = defaultdict(magic) 
>>> print lookup['DNE'] 
'DNE' 
>>> print lookup.get('DNE') 
'DNE' 
>>> print lookup['exists'] 
'some other value' 
>>> print lookup.get('exists') 
'some other value' 

Estos siempre será cuerdas, pero básicamente estoy creando un mapa idioma y necesitan una manera fácil de obtener un valor, si es que existe regresar en otro devolver la llave.

¿Hay alguna manera fácil de hacerlo? ¿O debería extender el dict y hacerlo manualmente?

Answer 1

debería ser posible con una función lambda

from collections import defaultdict 
a = defaultdict((lambda : 'DNE')) 

Edit: Lo siento leí mal la pregunta. Como el comentario anterior ya dijo. El camino a seguir es extender la clase dict.

>>> class mydict(dict): 
...  def __missing__(self,key): 
...   return key 
... 
>>> a = mydict() 
>>> a['asd'] 
'asd' 

Answer 2

No creo que defaultdict lo ayude aquí porque la función que genera el valor predeterminado no tiene acceso a la clave solicitada.

Sin embargo utiliza un diccionario ordinario y utiliza get con un valor por defecto:

>>> lookup = {} 
>>> key = 'DNE' 
>>> lookup.get(key, key) 
'DNE' 

Answer 3

Esto funciona bajo Python 2.7 como mínimo:

from collections import defaultdict 

class KeyAwareDefaultDict(defaultdict): 
    def __missing__(self, key): 
     if self.default_factory is None: 
      raise KeyError(key) 
     self[key] = value = self.default_factory(key) 
     return value 

lookup = KeyAwareDefaultDict((lambda key: key)) 

A diferencia de la norma dict, defaultdict.get() parece llamar __missing__ también, así que tiene una mejor base para la extensión.