Merge pull request #228 from ibhuiyan17/v1-to-FastAPI

V1 to fast api

Author: ExpDev07

app/main.py

@@ -10,15 +10,15 @@
 import uvicorn
 from fastapi import FastAPI, Request, Response
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.middleware.wsgi import WSGIMiddleware
 from fastapi.responses import JSONResponse
 
 from .core import create_app
 from .data import data_source
 from .models.latest import LatestResponse as Latest
 from .models.location import LocationResponse as Location
 from .models.location import LocationsResponse as Locations
-from .router import router
+from .router.v1 import router as v1router
+from .router.v2 import router as v2router
 
 # ############
 # FastAPI App
@@ -83,11 +83,9 @@ async def handle_validation_error(request: Request, exc: pydantic.error_wrappers
 
 
 # Include routers.
-APP.include_router(router, prefix="/v2", tags=["v2"])
+APP.include_router(v1router, prefix="", tags=["v1"])
+APP.include_router(v2router, prefix="/v2", tags=["v2"])
 
-# mount the existing Flask app
-# v1 @ /
-APP.mount("/", WSGIMiddleware(create_app()))
 
 # Running of app.
 if __name__ == "__main__":

app/router/__init__.py

@@ -1,7 +1,6 @@
 from fastapi import APIRouter
 
-# Create the router.
-router = APIRouter()
+from .v1 import all, confirmed, deaths, recovered
 
 # The routes.
-from . import latest, sources, locations  # isort:skip
+from .v2 import latest, sources, locations  # isort:skip

app/router/v1/__init__.py

@@ -0,0 +1,3 @@
+from fastapi import APIRouter
+
+router = APIRouter()

app/router/v1/all.py

@@ -0,0 +1,19 @@
+from ...services.location.jhu import get_category
+from . import router
+
+
+@router.get("/all")
+def all():
+    # Get all the categories.
+    confirmed = get_category("confirmed")
+    deaths = get_category("deaths")
+    recovered = get_category("recovered")
+
+    return {
+        # Data.
+        "confirmed": confirmed,
+        "deaths": deaths,
+        "recovered": recovered,
+        # Latest.
+        "latest": {"confirmed": confirmed["latest"], "deaths": deaths["latest"], "recovered": recovered["latest"],},
+    }

app/router/v1/confirmed.py

@@ -0,0 +1,9 @@
+from ...services.location.jhu import get_category
+from . import router
+
+
+@router.get("/confirmed")
+def confirmed():
+    confirmed = get_category("confirmed")
+
+    return confirmed

app/router/v1/deaths.py

@@ -0,0 +1,9 @@
+from ...services.location.jhu import get_category
+from . import router
+
+
+@router.get("/deaths")
+def deaths():
+    deaths = get_category("deaths")
+
+    return deaths

app/router/v1/recovered.py

@@ -0,0 +1,9 @@
+from ...services.location.jhu import get_category
+from . import router
+
+
+@router.get("/recovered")
+def recovered():
+    recovered = get_category("recovered")
+
+    return recovered

app/router/v2/__init__.py

@@ -0,0 +1,3 @@
+from fastapi import APIRouter
+
+router = APIRouter()

app/router/latest.py app/router/v2/latest.pyapp/router/latest.py renamed to app/router/v2/latest.py

@@ -1,7 +1,7 @@
 from fastapi import Request
 
-from ..enums.sources import Sources
-from ..models.latest import LatestResponse as Latest
+from ...enums.sources import Sources
+from ...models.latest import LatestResponse as Latest
 from . import router
 
 

app/router/locations.py app/router/v2/locations.pyapp/router/locations.py renamed to app/router/v2/locations.py

@@ -1,8 +1,8 @@
 from fastapi import HTTPException, Request
 
-from ..enums.sources import Sources
-from ..models.location import LocationResponse as Location
-from ..models.location import LocationsResponse as Locations
+from ...enums.sources import Sources
+from ...models.location import LocationResponse as Location
+from ...models.location import LocationsResponse as Locations
 from . import router