Fixed issue with the alerte feature

nbr23 · Jun 5, 2015 · 0817530 · 0817530
1 parent 7e550a5
commit 0817530
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/nextstop.py b/src/nextstop.py
@@ -74,7 +74,8 @@ def main():
     if cause == "" or type_transp == "":
       printUsage(sys.argv[0])
       return 1
-    return ratp.getDisturbance(cause, type_transp)
+    print(ratp.getDisturbance(cause, type_transp))
+    return 0
   if type_transp == "" or line == "":
     printUsage(sys.argv[0])
     return 1

diff --git a/src/ratp.py b/src/ratp.py
@@ -108,6 +108,10 @@ def getDisturbance(cause, transport):
     page = getPage('/siv/perturbation?cause=%s&reseau=%s' % (cause, transport))
     soup = bs4.BeautifulSoup(page)
     content = soup.find_all('div', { "class" : "bg1" })
+    disturbances = ""
     for item in content:
+        if disturbances != "":
+            disturbances += "\n"
         item = str(item).replace('<br/>', ' ').replace('  ', ' ')
-        print(bs4.BeautifulSoup(item).get_text())
+        disturbances += (bs4.BeautifulSoup(item).get_text())
+    return disturbances