How can I do a conditional yield with Python?

Question

I have to refactor some code. The original code was something like:

 while (yield _requires_payment(state)):
        did_pass_limit = yield _did_pass_limit(state)
        if not did_pass_limit:
            if existing_count is None:
                yield send_info_log("some stuff")
            yield send_info_log(f"more stuff")

So I refactored to:

    if yield _requires_payment(state):
        yield send_info_log(f"stuff")

If it matters, the function definition is:

@dialog(version="1.0")
async def _requires_payment(state):
    return await apply_payment_status_check(state, check_types=["window"])

But this creates an issue. What am I doing wrong?

Answer 1

If you want to use yield as an expression there, surround it with parentheses . Just like it was in your while condition.

(In assignment statements , like x = yield y , it's not needed, because that directly uses yield_expression instead of yield_atom .)

Answer 2

if evaluates Boolean values Try:

 if (yield _requires_payment(state)):
        yield send_info_log(f"stuff")