Skip to content

Commit f05afe1

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

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

app/routers/v2.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,15 @@ 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(
24+
request: Request, source: Sources = Sources.JHU
25+
): # pylint: disable=unused-argument
2426
"""
2527
Getting latest amount of total confirmed cases, deaths, and recoveries.
2628
"""
@@ -91,7 +93,7 @@ async def get_locations(
9193
# pylint: disable=invalid-name
9294
@V2.get("/locations/{id}", response_model=LocationResponse)
9395
async def get_location_by_id(
94-
request: Request, id: int, source: Sources = "jhu", timelines: bool = True
96+
request: Request, id: int, source: Sources = Sources.JHU, timelines: bool = True
9597
):
9698
"""
9799
Getting specific location by id.

0 commit comments

Comments
 (0)