#232: Make all endpoints async

.gitignore

@@ -66,4 +66,7 @@ docs/_build/
 target/
 
 # OSX Stuff
-.DS_Store
+.DS_Store
+
+# IntelliJ/Pycharm
+.idea/

Pipfile

@@ -4,19 +4,28 @@ url = "https://pypi.org/simple"
 verify_ssl = true
 
 [dev-packages]
+async-asgi-testclient = "*"
+async_generator = "*"
+asyncmock = "*"
 bandit = "*"
 black = "==19.10b0"
 coveralls = "*"
+importlib-metadata = "*"
 invoke = "*"
 isort = "*"
 pylint = "*"
 pytest = "*"
+pytest-asyncio = "*"
 pytest-cov = "*"
 
 [packages]
+aiohttp = "*"
+asyncache = "*"
 cachetools = "*"
+dataclasses = "*"
 fastapi = "*"
 gunicorn = "*"
+idna_ssl = "*"
 python-dateutil = "*"
 python-dotenv = "*"
 requests = "*"

app/main.py

@@ -13,6 +13,7 @@
 from .data import data_source
 from .router.v1 import V1
 from .router.v2 import V2
+from .utils.httputils import setup_client_session, teardown_client_session
 
 # ############
 # FastAPI App
@@ -28,6 +29,8 @@
     version="2.0.1",
     docs_url="/",
     redoc_url="/docs",
+    on_startup=[setup_client_session],
+    on_shutdown=[teardown_client_session],
 )
 
 # #####################

app/router/v1/all.py

@@ -4,11 +4,11 @@
 
 
 @V1.get("/all")
-def all():  # pylint: disable=redefined-builtin
+async def all():  # pylint: disable=redefined-builtin
     """Get all the categories."""
-    confirmed = get_category("confirmed")
-    deaths = get_category("deaths")
-    recovered = get_category("recovered")
+    confirmed = await get_category("confirmed")
+    deaths = await get_category("deaths")
+    recovered = await get_category("recovered")
 
     return {
         # Data.

app/router/v1/confirmed.py

@@ -4,6 +4,8 @@
 
 
 @V1.get("/confirmed")
-def confirmed():
+async def confirmed():
     """Confirmed cases."""
-    return get_category("confirmed")
+    confirmed_data = await get_category("confirmed")
+
+    return confirmed_data

app/router/v1/deaths.py

@@ -4,6 +4,8 @@
 
 
 @V1.get("/deaths")
-def deaths():
+async def deaths():
     """Total deaths."""
-    return get_category("deaths")
+    deaths_data = await get_category("deaths")
+
+    return deaths_data

app/router/v1/recovered.py

@@ -4,6 +4,8 @@
 
 
 @V1.get("/recovered")
-def recovered():
+async def recovered():
     """Recovered cases."""
-    return get_category("recovered")
+    recovered_data = await get_category("recovered")
+
+    return recovered_data

app/router/v2/latest.py

@@ -7,11 +7,11 @@
 
 
 @V2.get("/latest", response_model=Latest)
-def get_latest(request: Request, source: Sources = "jhu"):  # pylint: disable=unused-argument
+async def get_latest(request: Request, source: Sources = "jhu"):  # pylint: disable=unused-argument
     """
     Getting latest amount of total confirmed cases, deaths, and recoveries.
     """
-    locations = request.state.source.get_all()
+    locations = await request.state.source.get_all()
     return {
         "latest": {
             "confirmed": sum(map(lambda location: location.confirmed, locations)),