gaugendre/tcl-filtrage
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Reformat

Browse source
This commit is contained in:
Gabriel Augendre 2021-11-09 12:33:52 +01:00
parent ce7eecfd3d
commit 8d0ca77be1
3 changed files with 20 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
main.py
View file

@ -1,15 +1,19 @@
import enum
from typing import List, Optional from typing import List, Optional
import httpx
from fastapi import FastAPI, HTTPException from fastapi import FastAPI, HTTPException
from fastapi.params import Header from fastapi.params import Header
from pydantic import BaseModel from pydantic import BaseModel
import enum, httpx
app = FastAPI() app = FastAPI()
class PassageType(enum.Enum): class PassageType(enum.Enum):
E = "E" E = "E"
T = "T" T = "T"
class Passage(BaseModel): class Passage(BaseModel):
coursetheorique: str coursetheorique: str
delaipassage: str delaipassage: str
@ -22,9 +26,11 @@ class Passage(BaseModel):
ligne: str ligne: str
type: PassageType type: PassageType
class Passages(BaseModel): class Passages(BaseModel):
passages: List[Passage] passages: List[Passage]
# Stop id can be optained using # Stop id can be optained using
# https://data.grandlyon.com/jeux-de-donnees/points-arret-reseau-transports-commun-lyonnais/donnees # https://data.grandlyon.com/jeux-de-donnees/points-arret-reseau-transports-commun-lyonnais/donnees
@app.get("/stop/{stop_id}", response_model=Passages) @app.get("/stop/{stop_id}", response_model=Passages)
@ -32,15 +38,19 @@ async def stop(stop_id: int, authorization: Optional[str] = Header(None)):
if authorization is None: if authorization is None:
raise HTTPException(status_code=401, detail="Not authenticated") raise HTTPException(status_code=401, detail="Not authenticated")
headers = {'Authorization': authorization} headers = {"Authorization": authorization}
async with httpx.AsyncClient(headers=headers) as client: async with httpx.AsyncClient(headers=headers) as client:
res = await client.get('https://download.data.grandlyon.com/ws/rdata/tcl_sytral.tclpassagearret/all.json?maxfeatures=-1') res = await client.get(
"https://download.data.grandlyon.com/ws/rdata/tcl_sytral.tclpassagearret/all.json?maxfeatures=-1"
)
if res.status_code != 200: if res.status_code != 200:
raise HTTPException(status_code=res.status_code, detail="HTTP error during call to remote API") raise HTTPException(
status_code=res.status_code,
detail="HTTP error during call to remote API",
)
passages: List[Passage] = [] passages: List[Passage] = []
for passage in res.json().get("values"): for passage in res.json().get("values"):
if passage.get("id") == stop_id: if passage.get("id") == stop_id:
passages.append(passage) passages.append(passage)
return Passages(passages=passages) return Passages(passages=passages)