From b61880fb5b097d0210fb185784717baee2cd6c43 Mon Sep 17 00:00:00 2001 From: Mira Weller Date: Thu, 4 Apr 2024 14:01:37 +0200 Subject: [PATCH] Allow passing a fallback now_dt to time_machine_now --- src/pretix/base/timemachine.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/pretix/base/timemachine.py b/src/pretix/base/timemachine.py index 2960850cf..c1f12b6cd 100644 --- a/src/pretix/base/timemachine.py +++ b/src/pretix/base/timemachine.py @@ -57,12 +57,12 @@ class TimeMachineMiddleware: request.now_dt_is_fake = False try: - TimeMachineMiddleware.tls.now_dt = request.now_dt + TimeMachineMiddleware.tls.now_dt = request.now_dt if request.now_dt_is_fake else None return self.get_response(request) finally: TimeMachineMiddleware.tls.now_dt = None -def time_machine_now(): - return getattr(TimeMachineMiddleware.tls, 'now_dt', None) or now() +def time_machine_now(now_dt=None): + return getattr(TimeMachineMiddleware.tls, 'now_dt', None) or now_dt or now()