How To Define A Variable In A Function Python Syntax If Else Statement