python assign if not none one line