Is there no elif / else logic here?  It's confusing to read.  If not, not--but if so, please use.