0x90/ticketfrei4
12
1
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

create haltestellen if they don't exist

Browse source
This commit is contained in:
0x90.space 2021-12-29 04:32:44 +01:00
parent ab79366ec5
commit a5ef683e32
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/ticketfrei3/application.py
View file

@ -62,7 +62,7 @@ async def fetch_haltestellen(client: ClientSession) -> None:
async def fetch_fahrten(client: ClientSession) -> None: async def fetch_fahrten(client: ClientSession) -> None:
for zweig in ("bus", "tram", "ubahn"): for zweig in ("bus", "tram", "ubahn"):
response = await client.get( response = await client.get(
"https://start.vag.de/dm/api/v1/fahrten.json/%s?timespan=120" % zweig "https://start.vag.de/dm/api/v1/fahrten.json/%s?timespan=70" % zweig
) )
fahrten = from_json(await response.text()) fahrten = from_json(await response.text())
for fahrt in fahrten["Fahrten"]: for fahrt in fahrten["Fahrten"]:
@ -82,6 +82,14 @@ async def fetch_fahrten(client: ClientSession) -> None:
.using_db(connection) .using_db(connection)
.first() .first()
) )
if haltestelle is None:
haltestelle = Haltestelle(
id=halt["VGNKennung"],
name=halt["Haltestellenname"],
longitude=halt.get("Longitude"),
latitude=halt.get("Latitude"),
)
await haltestelle.save(using_db=connection)
await Halt( await Halt(
fahrt=fahrt, fahrt=fahrt,
haltestelle=haltestelle, haltestelle=haltestelle,