Skip to content

Commit 8ac8d17

Browse files
committed
Added middleware management
* Added CORS middleware
1 parent 9f5e323 commit 8ac8d17

File tree

4 files changed

+34
-1
lines changed

4 files changed

+34
-1
lines changed

sample_fastapi/app/app_factory.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from fastapi import FastAPI
66
from tqdm.contrib.logging import logging_redirect_tqdm
77

8-
from . import resources
8+
from . import middleware, resources
99

1010
LOG = logging.getLogger(__name__)
1111

@@ -28,4 +28,6 @@ async def app_lifespan_handler(app: FastAPI):
2828
def init_app():
2929
app = FastAPI(lifespan=app_lifespan_handler)
3030
resources.init_app(app)
31+
middleware.init_middleware(app)
32+
3133
return app
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
from .middleware import init_middleware
2+
3+
__all__ = ["init_middleware"]

sample_fastapi/app/middleware/cors.py

+14
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
2+
from fastapi import FastAPI
3+
from fastapi.middleware.cors import CORSMiddleware
4+
5+
6+
def init_middleware(app: FastAPI):
7+
# TODO: Allow this to be configurable
8+
app.add_middleware(
9+
CORSMiddleware,
10+
allow_credentials=True,
11+
allow_origins=["*"],
12+
allow_methods=["*"],
13+
allow_headers=["*"],
14+
)
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
import logging
2+
3+
from fastapi import FastAPI
4+
5+
LOG = logging.getLogger(__name__)
6+
7+
8+
def init_middleware(app: FastAPI):
9+
from . import cors
10+
11+
# NOTE: Add more middleware here
12+
13+
LOG.info("Adding `CORS` middleware")
14+
cors.init_middleware(app)

0 commit comments

Comments
 (0)