summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGabriel Pérez-Cerezo <gabriel@gpcf.eu>2018-10-11 16:22:31 +0200
committerGabriel Pérez-Cerezo <gabriel@gpcf.eu>2018-10-11 16:22:31 +0200
commit12a004ad3bcc2150dca5ae45c963161e7eedc25c (patch)
tree818c6a6b7d8f13eed9ad0ca2a77593caf9a16820
parent7664fb26df0c4ba0f03504b55e15022769bab861 (diff)
downloadbvg-grabber-12a004ad3bcc2150dca5ae45c963161e7eedc25c.zip
bvg-grabber-12a004ad3bcc2150dca5ae45c963161e7eedc25c.tar.gz
bvg-grabber-12a004ad3bcc2150dca5ae45c963161e7eedc25c.tar.bz2
Fix Issue #11 from github
-rw-r--r--bvggrabber/api/actualdeparture.py3
-rw-r--r--bvggrabber/api/scheduleddeparture.py3
-rwxr-xr-xsetup.py4
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:
diff --git a/setup.py b/setup.py
index 8661e64..4b73677 100755
--- a/setup.py
+++ b/setup.py
@@ -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',