Skip to content

Commit 1949bea

Browse files
committed
add sources and make Timelines a boolean
1 parent aec80ac commit 1949bea

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

app/main.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ async def handle_validation_error(
7676

7777

7878
@V2.get("/latest", response_model=models.Latest)
79-
def get_latest(request: fastapi.Request):
79+
def get_latest(request: fastapi.Request, source: Sources = "jhu"):
8080
"""Getting latest amount of total confirmed cases, deaths, and recoveries."""
8181
locations = request.state.source.get_all()
8282
return {
@@ -94,7 +94,7 @@ def get_latest(request: fastapi.Request):
9494
def get_all_locations(
9595
request: fastapi.Request,
9696
country_code: str = None,
97-
timelines: int = 0,
97+
timelines: bool = False,
9898
source: Sources = "jhu",
9999
):
100100
# Retrieve all the locations.
@@ -120,18 +120,16 @@ def get_all_locations(
120120

121121

122122
@V2.get("/locations/{id}", response_model=models.Location)
123-
def get_location_by_id(request: fastapi.Request, id: int, timelines: int = 1):
123+
def get_location_by_id(request: fastapi.Request, id: int, timelines: bool = True):
124124
return {"location": request.state.source.get(id).serialize(timelines)}
125125

126126

127-
@V2.get('/sources')
127+
@V2.get("/sources")
128128
async def sources():
129129
"""
130130
Retrieves a list of data-sources that are availble to use.
131131
"""
132-
return {
133-
'sources': list(data_sources.keys())
134-
}
132+
return {"sources": list(data_sources.keys())}
135133

136134

137135
APP.include_router(V2, prefix="/v2-beta", tags=["v2"])

0 commit comments

Comments
 (0)