diff options
| -rw-r--r-- | bvggrabber/api/actualdeparture.py | 3 | ||||
| -rw-r--r-- | bvggrabber/api/scheduleddeparture.py | 3 | ||||
| -rwxr-xr-x | setup.py | 4 |
3 files changed, 6 insertions, 4 deletions
diff --git a/bvggrabber/api/actualdeparture.py b/bvggrabber/api/actualdeparture.py index e7c084b..00b6d1d 100644 --- a/bvggrabber/api/actualdeparture.py +++ b/bvggrabber/api/actualdeparture.py @@ -51,12 +51,13 @@ class ActualDepartureQueryApi(QueryApi): departures = [] for row in rows: if row.parent.name == 'tbody': + stro = row.find_all("strong") td = row.find_all('td') if td: dep = Departure(start=self.station, end=td[2].text.strip(), when=td[0].text.strip(), - line=td[1].text.strip()) + line=stro[1].text.strip()) departures.append(dep) return Response(True, self.station, departures) else: diff --git a/bvggrabber/api/scheduleddeparture.py b/bvggrabber/api/scheduleddeparture.py index f316b6e..101f692 100644 --- a/bvggrabber/api/scheduleddeparture.py +++ b/bvggrabber/api/scheduleddeparture.py @@ -70,10 +70,11 @@ class ScheduledDepartureQueryApi(QueryApi): departures = [] for row in rows: tds = row.find_all('td') + stro = row.find_all('strong') dep = Departure(start=self.station, end=tds[2].text.strip(), when=tds[0].text.strip(), - line=tds[1].text.strip()) + line=stro[1].text.strip()) departures.append(dep) return Response(True, self.station, departures) else: @@ -14,7 +14,7 @@ setup( name='bvg-grabber', description='Querying the upcoming public transport departures in Berlin', long_description=description, - version='0.1.1', + version='0.1.2', url='https://github.com/MarkusH/bvg-grabber', author='Christian Struck, Markus Holtermann', author_email='info@markusholtermann.eu', @@ -27,7 +27,7 @@ setup( 'requests>=1.2.3', 'six>=1.3.0', ], - test_suite="tests", +# test_suite="tests", scripts=['bvg-grabber.py'], classifiers=[ 'Development Status :: 4 - Beta', |