How can I return a default value for an attribute?

alwbtc picture alwbtc · Feb 17, 2013 · Viewed 32.7k times · Source

I have an object myobject, which might return None. If it returns None, it won't return an attribute id:

a = myobject.id

So when myobject is None, the stament above results in a AttributeError:

AttributeError: 'NoneType' object has no attribute 'id'

If myobject is None, then I want a to be equal to None. How do I avoid this exception in one line statement, such as:

a = default(myobject.id, None)

Answer

Inbar Rose picture Inbar Rose · Feb 17, 2013

You should use the getattr wrapper instead of directly retrieving the value of id.

a = getattr(myobject, 'id', None)

This is like saying "I would like to retrieve the attribute id from the object myobject, but if there is no attribute id inside the object myobject, then return None instead." But it does it efficiently.

Some objects also support the following form of getattr access:

a = myobject.getattr('id', None)

As per OP request, 'deep getattr':

def deepgetattr(obj, attr):
    """Recurses through an attribute chain to get the ultimate value."""
    return reduce(getattr, attr.split('.'), obj)
# usage: 
print deepgetattr(universe, 'galaxy.solarsystem.planet.name')

Simple explanation:

Reduce is like an in-place recursive function. What it does in this case is start with the obj (universe) and then recursively get deeper for each attribute you try to access using getattr, so in your question it would be like this:

a = getattr(getattr(myobject, 'id', None), 'number', None)