31 lines
925 B
Python
31 lines
925 B
Python
'''
|
|
Overall, I like types.SimpleNamespace. But I decided to make dotdict because:
|
|
1. SimpleNamespace is a cumbersome name to type and to look at.
|
|
2. I wanted my class to support default values.
|
|
'''
|
|
from voussoirkit import sentinel
|
|
|
|
NO_DEFAULT = sentinel.Sentinel('NO_DEFAULT')
|
|
|
|
class DotDict:
|
|
def __init__(self, __dict=None, *, default=NO_DEFAULT, **kwargs):
|
|
self.__default = default
|
|
if __dict:
|
|
self.__dict__.update(__dict)
|
|
self.__dict__.update(**kwargs)
|
|
|
|
def __getattr__(self, key):
|
|
try:
|
|
return self.__dict__[key]
|
|
except KeyError:
|
|
if self.__default is not NO_DEFAULT:
|
|
return self.__default
|
|
raise
|
|
|
|
def __setattr__(self, key, value):
|
|
self.__dict__[key] = value
|
|
|
|
def __repr__(self):
|
|
display = self.__dict__.copy()
|
|
display.pop('_DotDict__default')
|
|
return f'DotDict {display}'
|