From 16dd661d48598aac8c12c989696d1a9fee270c45 Mon Sep 17 00:00:00 2001
From: nex <me@nexy7574.co.uk>
Date: Mon, 20 Mar 2023 15:41:20 +0000
Subject: [PATCH] Change db to use /data if it exists

---
 utils/db.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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]: