fix: db client should be in the clients directory (#17)

spack · Mar 7, 2024 · 936c88e · 936c88e
1 parent f0db576
commit 936c88e
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 3 deletions.
diff --git a/gantry/db/__init__.py → gantry/clients/db/__init__.py b/gantry/db/__init__.py → gantry/clients/db/__init__.py
diff --git a/gantry/db/get.py → gantry/clients/db/get.py b/gantry/db/get.py → gantry/clients/db/get.py
diff --git a/gantry/db/insert.py → gantry/clients/db/insert.py b/gantry/db/insert.py → gantry/clients/db/insert.py
@@ -2,7 +2,7 @@
 
 import aiosqlite
 
-from gantry.db.get import get_node
+from gantry.clients.db.get import get_node
 
 logger = logging.getLogger(__name__)
 

diff --git a/gantry/routes/collection.py b/gantry/routes/collection.py
@@ -2,7 +2,7 @@
 
 import aiosqlite
 
-from gantry import db
+from gantry.clients import db
 from gantry.clients.gitlab import GitlabClient
 from gantry.clients.prometheus import PrometheusClient
 from gantry.clients.prometheus.util import IncompleteData

diff --git a/gantry/tests/test_db.py b/gantry/tests/test_db.py
@@ -1,4 +1,4 @@
-from gantry.db.insert import insert_job, insert_node
+from gantry.clients.db.insert import insert_job, insert_node
 from gantry.tests.defs import db as defs