Allow passing a fallback now_dt to time_machine_now

This commit is contained in:
Mira Weller
2024-04-30 21:49:58 +02:00
parent b29c7fc11d
commit b61880fb5b
+3 -3
View File
@@ -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()