migrate to fastapi

Author: mohamed040406

Pipfile

@@ -6,15 +6,16 @@ verify_ssl = true
 [dev-packages]
 black = "*"
 flake8 = "*"
+requests = "*"
 pre-commit = "*"
 pytest-asyncio = "*"
 
 [packages]
-quart = "*"
-quart-cors = "*"
-postdb = "*"
 pyjwt = "*"
+postdb = "*"
 aiohttp = "*"
+fastapi = "*"
+uvicorn = {extras = ["standard"], version = "*"}
 uvloop = {markers = "platform_system == 'linux'", version = "*"}
 
 [requires]
@@ -24,4 +25,5 @@ python_version = "3.8"
 allow_prereleases = true
 
 [scripts]
+test = "pytest"
 lint = "pre-commit run --all-files"

Pipfile.lock

@@ -1,45 +1,22 @@
-from quart import Quart, Response, exceptions, jsonify
-from datetime import datetime, date
-from aiohttp import ClientSession
-from typing import Any, Optional
-from quart_cors import cors
+from fastapi import FastAPI, HTTPException, Request
+from utils.response import JSONResponse
 import logging
-import json
-
-import utils
-
-from api.blueprints import auth, guilds, users
 
 
 log = logging.getLogger()
 
 
-class JSONEncoder(json.JSONEncoder):
-    def default(self, o: Any) -> Any:
-        if isinstance(o, (datetime, date)):
-            o.replace(microsecond=0)
-            return o.isoformat()
-
-        return super().default(o)
-
-
-class API(Quart):
-    """Quart subclass to implement more API like handling."""
-
-    http_session: Optional[ClientSession] = None
-    request_class = utils.Request
-    json_encoder = JSONEncoder
-
+class API(FastAPI):
+    """FastAPI subclass to implement more API like handling."""
     def __init__(self, *args, **kwargs):
-        kwargs.setdefault("static_folder", None)
         super().__init__(*args, **kwargs)
 
-    async def handle_request(self, request: utils.Request) -> Response:
+    async def handle_request(self, request: Request):
         response = await super().handle_request(request)
         log.info(f"{request.method} @ {request.base_url} -> {response.status_code}")
         return response
 
-    async def handle_http_exception(self, error: exceptions.HTTPException):
+    async def handle_http_exception(self, error: HTTPException):
         """
         Returns errors as JSON instead of default HTML
         Uses custom error handler if one exists.
@@ -53,35 +30,26 @@ async def handle_http_exception(self, error: exceptions.HTTPException):
         headers = error.get_headers()
         headers["Content-Type"] = "application/json"
 
-        return (
-            jsonify(error=error.name, message=error.description),
-            error.status_code,
-            headers,
+        return JSONResponse(
+            headers=headers,
+            status_code=error.status_code,
+            content={"error": error.name, "message": error.description}
         )
 
-    async def startup(self) -> None:
-        self.http_session = ClientSession()
-        return await super().startup()
-
 
-# Set up app
-app = API(__name__)
-app.asgi_app = utils.TokenAuthMiddleware(app.asgi_app, app)
-app = cors(app, allow_origin="*")  # TODO: Restrict the origin(s) in production.
-# Set up blueprints
-auth.setup(app=app, url_prefix="/auth")
-users.setup(app=app, url_prefix="/users")
-guilds.setup(app=app, url_prefix="/guilds")
+app = API()
+app.router.default_response_class = JSONResponse
+app.add_exception_handler(HTTPException, app.handle_http_exception)
 
 
-@app.route("/")
+@app.get("/")
 async def index():
     """Index endpoint used for testing."""
-    return jsonify(status="OK")
+    return {"status": "ok"}
 
 
-@app.errorhandler(500)
-async def error_500(error: BaseException):
+@app.exception_handler(500)
+async def error_500(request, error: HTTPException):
     """
     TODO: Handle the error with our own error handling system.
     """
@@ -90,7 +58,7 @@ async def error_500(error: BaseException):
         exc_info=(type(error), error, error.__traceback__),
     )
 
-    return (
-        jsonify(error="Internal Server Error", message="Server got itself in trouble"),
-        500,
+    return JSONResponse(
+        status_code=500,
+        content={"error": "Internal Server Error", "message": "Server got itself in trouble"}
     )