Establece los elementos de dictado de Python de forma recursiva, cuando se le da una clave compuesta ‘foo.bar.baz’

Me gustaría lograr lo siguiente:

foodict['foo.bar.baz'] = 'foo' { 'foo': { 'bar': { 'baz': 'foo' } } } } 

… creando claves recursivamente.

Después de rascarme la cabeza por un tiempo, se me ocurrió esto:

 class Config(dict): def __init__(self, *args, **kwargs): self.super = super(Config, self) self.update(*args, **kwargs) def __setitem__(self, keys, value): keys = keys.split('.') keys.reverse() config = Config() for i, k in enumerate(keys): if i == 0: config = Config(**{ k: value }) else: config = Config(**{ k: config }) self.super.update(config) 

Puede considerar la receta “infinito defaultdict”, del propio Raymond Hettinger:

https://twitter.com/raymondh/status/343823801278140417

 >>> from collections import defaultdict >>> infinite_defaultdict = lambda: defaultdict(infinite_defaultdict) >>> d = infinite_defaultdict() >>> d['foo']['bar']['baz'] = 'foo' >>> d defaultdict( at 0x1040388c8>, {'foo': defaultdict( at 0x1040388c8>, {'bar': defaultdict( at 0x1040388c8>, {'baz': 'foo'})})}) 

Otra opción es implementar __missing__ :

 >>> class InfiniteDict(dict): ... def __missing__(self, val): ... d = InfiniteDict() ... self[val] = d ... return d ... >>> d = InfiniteDict() >>> d['foo']['bar']['baz'] = 'foo' >>> d {'foo': {'bar': {'baz': 'foo'}}} 

Y si debes tener atributo de acceso:

 class InfiniteDict(dict): def __missing__(self, val): d = InfiniteDict() self[val] = d return d def __getattr__(self, item): return self.__getitem__(item) def __setattr__(self, item, value): super().__setitem__(item, value) 

En acción:

 >>> d = InfiniteDict() >>> d.foo.bar.baz = 'foo' >>> d {'foo': {'bar': {'baz': 'foo'}}} >>> 

Aunque, eso es un poco rápido y sucio, así que no hay garantías de que no haya errores. Y hay muy pocos guardias contra, por ejemplo, colisiones con atributos reales:

 >>> d.keys = 'should I be allowed?' >>> d {'foo': {'bar': {'baz': 'foo'}}, 'keys': 'should I be allowed?'} >>> d.keys() dict_keys(['foo', 'keys'])