import requests
from datetime import datetime
import urllib.parse
from flask import Flask
app = Flask(__name__)
def roman_to_arabic(value):
roman_map = {
'I': 1, 'V': 5, 'X': 10, 'L': 50,
'C': 100, 'D': 500, 'M': 1000
}
if not value or not all(ch in roman_map for ch in value):
return value
total = 0
prev_value = 0
for ch in reversed(value):
current = roman_map[ch]
if current < prev_value:
total -= current
else:
total += current
prev_value = current
return str(total)
def abbrevia_destinazione(dest):
dest = dest.upper()
dest = dest.replace("VITERBO PORTA FIORENTINA", "VITERBO P. F.")
dest = dest.replace("VITERBO PORTA ROMANA", "VITERBO P. R.")
return dest
@app.route("/")
def partenze():
station_id = "S08327" # Roma Balduina
now = datetime.now()
formatted = now.strftime("%a %b %d %Y %H:%M:%S +0200")
encoded_date = urllib.parse.quote(formatted, safe="")
url = f"http://www.viaggiatreno.it/infomobilita/resteasy/viaggiatreno/partenze/{station_id}/{encoded_date}"
response = requests.get(url)
if response.status_code != 200:
table_rows = "
ERRORE
"
else:
data = response.json()
table_rows = ""
for treno in data[:8]:
ora = datetime.fromtimestamp(treno['orarioPartenza']/1000).strftime('%H:%M').upper()
destinazione = abbrevia_destinazione(treno.get('destinazione', 'SCONOSCIUTA'))
categoria = treno.get('categoriaDescrizione', '').upper()
numero = str(treno.get('numeroTreno', '')).upper()
ritardo = treno.get('ritardo', 0)
binario = treno.get('binarioEffettivoPartenzaDescrizione') or treno.get('binarioProgrammatoPartenzaDescrizione') or '-'
binario = roman_to_arabic(binario)
if ritardo > 0:
stato = f"+{ritardo} MIN"
elif ritardo < 0:
stato = f"{ritardo} MIN"
else:
stato = " "
table_rows += f"""
{ora}
{categoria} {numero}
{destinazione}
{binario}
{stato}
"""
html = f"""
ORARIO
TRENO
DESTINAZIONE
BIN.
RITARDO
{table_rows}
"""
return html
if __name__ == "__main__":
app.run(host="0.0.0.0", port=9999)