Python dictionary get with default value
Yesterday I spent way more time than neccessary debugging a piece of python code. It has something to do with how the python dictionary’s
get method works with
default arguments. TLDR: it is ok when default is a simple value, not recommended if default is a function call.
Here is what the problematic code looks like:
def get_option(option_name): option = # some logic return option x = d.get("option_1", default=get_option("option_1"))
The problem here is that the
default argument is a pointer to the return value of a function call, thus the function will be evaluated before the body of the
get method. So no matter whether the key exists or not, the function will always be evaluated. It can break if:
- the function is not defined.
- the function call raises an Error or Exception.
The fix is change it to:
x = d.get("option_1", default=None) or get_option("option_1")
So that the function call will be used as a last resort when key is not in the dictionary.
But one drawback is that, if the value is logically
False, like a int value
0, it will still fallback to the function call.
So a better rewrite is:
x = d["option_1"] if "option_1" in d else get_option("option_1")
This is slightly different to
collections.defaultdict. In which, the fallback
default_factory callable is only called after the key check. But ofcourse,
default_factory does not accept any argument, thus does not suit the usecase here.