From da0fa00531ab39d1d44e2fd3a523ee0caadb8e22 Mon Sep 17 00:00:00 2001 From: Sebastian Hasler Date: Wed, 2 Jun 2021 01:54:47 +0200 Subject: [PATCH] feat(zeitrahmen): Add 59s leeway to bis_uhrzeit --- tools/its.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/tools/its.py b/tools/its.py index 265145df..5e925d3c 100644 --- a/tools/its.py +++ b/tools/its.py @@ -3,7 +3,7 @@ import sys import time from base64 import b64encode -from datetime import datetime, date +from datetime import datetime, date, timedelta from datetime import time as dtime from random import choice, randint @@ -798,7 +798,7 @@ def terminpaar_im_zeitrahmen(terminpaar, zeitrahmen): :param zeitrahmen: Zeitrahmen-Dictionary wie in ImpfterminService.termin_suchen :return: True oder False """ - if not zeitrahmen: # Teste auf leeres dict + if not zeitrahmen: # Teste auf leeres dict return True assert zeitrahmen["einhalten_bei"] in ["1", "2", "beide"] @@ -812,9 +812,10 @@ def terminpaar_im_zeitrahmen(terminpaar, zeitrahmen): von_uhrzeit = datetime.strptime( zeitrahmen["von_uhrzeit"], "%H:%M").time() if "von_uhrzeit" in zeitrahmen else dtime.min - bis_uhrzeit = datetime.strptime( - zeitrahmen["bis_uhrzeit"], - "%H:%M").time() if "bis_uhrzeit" in zeitrahmen else dtime.max + bis_uhrzeit = ( + datetime.strptime(zeitrahmen["bis_uhrzeit"], "%H:%M") + + timedelta(seconds=59) + ).time() if "bis_uhrzeit" in zeitrahmen else dtime.max wochentage = [decode_wochentag(wt) for wt in set( zeitrahmen["wochentage"])] if "wochentage" in zeitrahmen else range(7)