Skip to content

Commit 16e9bcd

Browse files
committed
update Sources enum
* linting fix (invalid-name) * use Sources enum instead of string
1 parent bdc7843 commit 16e9bcd

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

app/routers/v2.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,13 @@ class Sources(str, enum.Enum):
1414
A source available for retrieving data.
1515
"""
1616

17-
jhu = "jhu"
18-
csbs = "csbs"
19-
nyt = "nyt"
17+
JHU = "jhu"
18+
CSBS = "csbs"
19+
NYT = "nyt"
2020

2121

2222
@V2.get("/latest", response_model=LatestResponse)
23-
async def get_latest(request: Request, source: Sources = "jhu"): # pylint: disable=unused-argument
23+
async def get_latest(request: Request, source: Sources = Sources.JHU): # pylint: disable=unused-argument
2424
"""
2525
Getting latest amount of total confirmed cases, deaths, and recoveries.
2626
"""
@@ -91,7 +91,7 @@ async def get_locations(
9191
# pylint: disable=invalid-name
9292
@V2.get("/locations/{id}", response_model=LocationResponse)
9393
async def get_location_by_id(
94-
request: Request, id: int, source: Sources = "jhu", timelines: bool = True
94+
request: Request, id: int, source: Sources = Sources.JHU, timelines: bool = True
9595
):
9696
"""
9797
Getting specific location by id.

0 commit comments

Comments
 (0)