Catching DBConnectionError on register_tortoise() in try/except

[PancakePuncher]

Hey guys, I am hoping for some help.

In attempt to harden up my script a little I wanted to try and gracefully handle the situation where my Database might not exist. Currently I am using FastAPI, MariaDB, and TortoiseORM. I'm using Tortoise's convenient "register_tortoise()" method to handle the initial connecting of my API to MariaDB. Today I attempting to catch the failure of that process in a try/except block and it seems that I am unable to actual catch DBConnectionError. Here is some sudo example coding.

import uvicorn
import sys
from tortoise.contrib.fastapi import register_tortoise
from fastapi.middleware.cors import CORSMiddleware
from fastapi import FastAPI

app = FastAPI()

origins = [
    "http://localhost:3000",
]

app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

try:
    register_tortoise(
        app,
        db_url=f"mysql://user:password@localhost:3306/mariadb",
        modules={"models": ["models.database"]},
        generate_schemas=False,
    )
except Exception as e:
    print(f"Database Error {e}: An error occurred between the API and the database.", e)
    x = str(input("Would you like to conintue?")).lower()

    if x == "no" or x == "n":
        sys.exit("API stopped by request.")

@app.get(
    "/brew_coffee",
    status_code=418,
)
async def brew_coffee():
    return {"message": "I can't brew coffee because..."}

if __name__ == "__main__":
    uvicorn.run("main:app", host="127.0.0.1", port=8000, reload=True)

Anyone know how I might go about catching when register_tortoise() fails? Right now it just hard stops the API.

My guess is that there's a raise occurring much deeper in to the register_tortoise() that's never making it to the higher scoped try/except block.

My current design this is working on is much larger then what I am showing here and I'm hoping I do not have to ditch the register_tortoise() method as the re-design could be much more intense. I really just want my API to move on and still function, just with no database capabilities.

Any help is much appreciated!