assign the condition of the if statement to a variable

Question

I have a long snippet of code and i am trying to shorten it. This is what I have now:

statuses = cleaned_data.get("statuses")
if statuses:
    for status in statuses:
        ...

it would be nice if i can assign the value of cleaned_data.get("statuses") inside the if condition, like this:
if cleaned_data.get("statuses") as statuses
and carry on.
Is it doable in python?

@deceze nice idea. but what if i want to do something if the condition fails? using for else will not work as the else is reached in the last iteration — Ahmad Hussian, Mar 13 '23 at 00:44

score 1 · Accepted Answer · answered Mar 08 '23 at 10:28

1

You can use Assignment Expression (a.k.a walrus operator):

if statuses := cleaned_data.get("statuses"):
    for status in statuses:

answered Mar 08 '23 at 10:28

matszwecja

6,357
2
10
17

Can be shortened to `for status in statuses := cleaned_data.get("statuses", []):` – B Remmelzwaal Mar 08 '23 at 10:30
You are probably right, but without knowing what `cleaned_data.get("statuses")` is, I am hesitant to make this statement, as the object can be Falsey but still iterable upon. – matszwecja Mar 08 '23 at 10:35
Good point, best to keep bugs like those out. – B Remmelzwaal Mar 08 '23 at 10:38

assign the condition of the if statement to a variable

1 Answers1