diff options
| author | Gabriel Pérez-Cerezo <gabriel@gpcf.eu> | 2018-10-11 16:22:31 +0200 |
|---|---|---|
| committer | Gabriel Pérez-Cerezo <gabriel@gpcf.eu> | 2018-10-11 16:22:31 +0200 |
| commit | 12a004ad3bcc2150dca5ae45c963161e7eedc25c (patch) | |
| tree | 818c6a6b7d8f13eed9ad0ca2a77593caf9a16820 | |
| parent | 7664fb26df0c4ba0f03504b55e15022769bab861 (diff) | |
| download | bvg-grabber-12a004ad3bcc2150dca5ae45c963161e7eedc25c.tar.gz bvg-grabber-12a004ad3bcc2150dca5ae45c963161e7eedc25c.tar.bz2 bvg-grabber-12a004ad3bcc2150dca5ae45c963161e7eedc25c.zip | |
Fix Issue #11 from github
| -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', |