python walrus operator if statement and