Added tortoise connection

lugvitc · Nov 16, 2023 · 08e4172 · 08e4172
1 parent db9acdb
commit 08e4172
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/pwncore/__init__.py b/src/pwncore/__init__.py
@@ -3,8 +3,18 @@
 import pwncore.docs as docs
 import pwncore.routes as routes
 
+from tortoise.contrib.fastapi import register_tortoise
+
 app = FastAPI(
     title="Pwncore", openapi_tags=docs.tags_metadata, description=docs.description
 )
 
 app.include_router(routes.router)
+
+register_tortoise(
+    app,
+    db_url = "sqlite://:memory:",
+    modules = {"models": ["pwncore.models"]},
+    generate_schemas = True,
+    add_exception_handlers = True,
+)