add source query params

Author: Kilo59

app/main.py

@@ -114,7 +114,7 @@ async def handle_validation_error(
 
 
 @APP.get("/latest", response_model=Latest)
-def get_latest(request: fastapi.Request):
+def get_latest(request: fastapi.Request, source: str = "jhu"):
     """Getting latest amount of total confirmed cases, deaths, and recoveries."""
     locations = request.state.source.get_all()
     return {
@@ -128,7 +128,10 @@ def get_latest(request: fastapi.Request):
 
 @APP.get("/locations", response_model=AllLocations)
 def get_all_locations(
-    request: fastapi.Request, country_code: str = None, timelines: int = 0
+    request: fastapi.Request,
+    country_code: str = None,
+    timelines: int = 0,
+    source: str = "jhu",
 ):
     # Retrieve all the locations.
     locations = request.state.source.get_all()
@@ -153,9 +156,12 @@ def get_all_locations(
 
 
 @APP.get("/locations/{id}", response_model=Location)
-def get_location_by_id(request: fastapi.Request, id: int, timelines: int = 1):
+def get_location_by_id(
+    request: fastapi.Request, id: int, timelines: int = 1, source: str = "jhu"
+):
     return {"location": request.state.source.get(id).serialize(timelines)}
 
+
 # mount the existing Flask app to /v2
 APP.mount("/v2", WSGIMiddleware(create_app()))