From 951a2d325c92111083c73a237f8ac4c39c4adcea Mon Sep 17 00:00:00 2001
From: Jan Losinski <losinski@wh2.tu-dresden.de>
Date: Mon, 7 Jan 2019 10:49:23 +0100
Subject: [PATCH] Improve multi-appointment booking

---
 punkow/scraper.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/punkow/scraper.py b/punkow/scraper.py
index 8c9f368..1eb0fb7 100644
--- a/punkow/scraper.py
+++ b/punkow/scraper.py
@@ -24,6 +24,7 @@ def print_url(r, *args, **kwargs):
 class BookingData(typing.NamedTuple):
     name: str
     email: str
+    id: typing.Optional[int]
 
 
 class BookingService(object):
@@ -190,7 +191,8 @@ class BookingService(object):
             with self._local_referrer():
                 for slot_url in self._iter_bookable_times(day_url):
                     if self._book_appointment(slot_url, cur_data.name, cur_data.email):
-                        cur_data = next(data_iter)
+                        yield cur_data
+                        cur_data = next(data_iter, None)
                         if cur_data is None:
                             return
         return