diff --git a/utils/db.py b/utils/db.py index 52e1e1c..d0d0beb 100644 --- a/utils/db.py +++ b/utils/db.py @@ -36,8 +36,10 @@ __all__ = [ T = TypeVar("T") T_co = TypeVar("T_co", covariant=True) - -registry = orm.ModelRegistry(Database("sqlite:///main.db")) +_pth = "/main.db" +if Path("/data").exists(): + _pth = "/data/main.db" +registry = orm.ModelRegistry(Database("sqlite://" + _pth)) async def get_or_none(model: T, **kw) -> Optional[T_co]: