Use different operator based on condition in Python

Question

In my current project I have some duplicated code and I am looking for a possibility to remove it. I have a boolean called forward. If it's true, I want to add days to a datetime, if not I want to subtract days:

if forward:
    day = today + datetime.timedelta(days=3)
else:
    day = today - datetime.timedelta(days=3)

Is there any possibility to do that in less than these 4 lines?

day = today + datetime.timedelta(days=3 if forward else -3) . # timedelta works with negative numbers as well which is same as using -timedelta — Ehtesham Siddiqui
– Ehtesham Siddiqui, Commented Jun 26, 2020 at 9:52
(operator.add if forward else operator.sub)(today, datetime.timedelta(days=3))… But is that really more readable…?! — deceze
– deceze ♦, Commented Jun 26, 2020 at 9:53
There are shorter ways, but none are clearer than what you already have. — Dash
– Dash, Commented Jun 26, 2020 at 9:56

AKX · Accepted Answer · 2020-06-26 09:53:57Z

3

Assuming it's not statically 3, multiply with -1 or +1 depending on direction?

n_days = 10
day = today + datetime.timedelta(days=(n_days * (1 if forward else -1)))

answered Jun 26, 2020 at 9:53

AKX

171k16 gold badges147 silver badges229 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Use different operator based on condition in Python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related