Skip to content

Commit aec80ac

Browse files
committed
use separate v2 router
1 parent ece30f3 commit aec80ac

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

app/main.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,10 @@ async def handle_validation_error(
7272
# Routes
7373
# ################
7474

75+
V2 = fastapi.APIRouter()
7576

76-
@APP.get("/latest", response_model=models.Latest)
77+
78+
@V2.get("/latest", response_model=models.Latest)
7779
def get_latest(request: fastapi.Request):
7880
"""Getting latest amount of total confirmed cases, deaths, and recoveries."""
7981
locations = request.state.source.get_all()
@@ -86,7 +88,7 @@ def get_latest(request: fastapi.Request):
8688
}
8789

8890

89-
@APP.get(
91+
@V2.get(
9092
"/locations", response_model=models.AllLocations, response_model_exclude_unset=True
9193
)
9294
def get_all_locations(
@@ -117,12 +119,12 @@ def get_all_locations(
117119
}
118120

119121

120-
@APP.get("/locations/{id}", response_model=models.Location)
122+
@V2.get("/locations/{id}", response_model=models.Location)
121123
def get_location_by_id(request: fastapi.Request, id: int, timelines: int = 1):
122124
return {"location": request.state.source.get(id).serialize(timelines)}
123125

124126

125-
@APP.get('/sources')
127+
@V2.get('/sources')
126128
async def sources():
127129
"""
128130
Retrieves a list of data-sources that are availble to use.
@@ -132,6 +134,7 @@ async def sources():
132134
}
133135

134136

137+
APP.include_router(V2, prefix="/v2-beta", tags=["v2"])
135138
# mount the existing Flask app
136139
# v1 @ /
137140
# v2 @ /v2

0 commit comments

Comments
 (0)