nemunaire/epaper
1
0
Fork 0
Code Issues Pull Requests Releases Activity

ratp: Add message to disruptions

Browse Source
This commit is contained in:
nemunaire 2023-01-19 12:04:08 +01:00
parent 4bf17e2e39
commit 2bd7fafaa5
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
modules/ratp.py
View File

@ -3,6 +3,7 @@ import hashlib
import json import json
import logging import logging
import os import os
import re
import urllib.parse import urllib.parse
import urllib.request import urllib.request
@ -64,7 +65,9 @@ class IDFMAPI:
self.cache_timeout = config.cache_timeout self.cache_timeout = config.cache_timeout
self.max_cache_timeout = config.max_cache_timeout self.max_cache_timeout = config.max_cache_timeout
def fromIVtoPRIM(src): def fromIVtoPRIM(src, line):
cleanr = re.compile('<.*?>')
cleanrA = re.compile('<a.*?>.*?</a>')
return { return {
"InfoChannelRef": { "InfoChannelRef": {
"value": "Perturbation" if src["severity"] >= 2 else "Travaux", "value": "Perturbation" if src["severity"] >= 2 else "Travaux",
@ -73,7 +76,9 @@ class IDFMAPI:
"Message": [{ "Message": [{
"MessageType": "SHORT_MESSAGE", "MessageType": "SHORT_MESSAGE",
"MessageText": { "MessageText": {
"value": src["title"], "id": src["id"],
"title": src["title"].replace("Métro " + line + " : ", '').replace("Ligne " + line + " : ", ''),
"value": re.sub(cleanr, '', re.sub(cleanrA, '', src["message"].replace('&nbsp;', ' ').replace('&#8217;', "'").replace('&#224;', 'à').replace('&#233;', 'é').replace('&#232;', 'è').replace('<br>', ' ').replace('</p>', ' ').replace('Information Ile de France Mobilités :', ''))).strip(),
} }
}], }],
} }
@ -167,7 +172,7 @@ class IDFMAPI:
if "id" in l: if "id" in l:
if l["id"] == "line:IDFM:" + IDFMAPI.lines[mode][line]: if l["id"] == "line:IDFM:" + IDFMAPI.lines[mode][line]:
for d in l["disruptions"]: for d in l["disruptions"]:
yield IDFMAPI.fromIVtoPRIM(d) yield IDFMAPI.fromIVtoPRIM(d, line)
return None return None
@ -204,6 +209,7 @@ class RATPWeatherModule:
alerts = [] alerts = []
weather = IDFMAPI(config).get_weather() weather = IDFMAPI(config).get_weather()
id_seens = []
for mode in weather: for mode in weather:
for line in weather[mode]: for line in weather[mode]:
if mode[0].upper() + line not in self.major_lines: if mode[0].upper() + line not in self.major_lines:
@ -224,7 +230,7 @@ class RATPWeatherModule:
return icon return icon
for info in weather[mode][line]: for info in weather[mode][line]:
if "InfoChannelRef" not in info or info["InfoChannelRef"]["value"] != "Perturbation": if "InfoChannelRef" not in info or (info["InfoChannelRef"]["value"] != "Perturbation" and info["InfoChannelRef"]["value"] != "Travaux"):
continue continue
if "Message" not in info["Content"]: if "Message" not in info["Content"]:
@ -234,7 +240,12 @@ class RATPWeatherModule:
if "MessageType" not in msg or msg["MessageType"] != "SHORT_MESSAGE": if "MessageType" not in msg or msg["MessageType"] != "SHORT_MESSAGE":
continue continue
if "id" in msg["MessageText"]:
if msg["MessageText"]["id"] in id_seens:
continue
id_seens.append(msg["MessageText"]["id"])
yield { yield {
"title": msg["MessageText"]["title"] if "title" in msg["MessageText"] else "",
"description": msg["MessageText"]["value"].replace(" ", " "), "description": msg["MessageText"]["value"].replace(" ", " "),
"icon": alert_icon(mode, line), "icon": alert_icon(mode, line),
} }