def foo(arg):
    return arg if arg else None