From 910e7539e2a5911c5b5b4f8645ae53b1d5abc7f3 Mon Sep 17 00:00:00 2001
From: Calum Andrew Morrell <calum@drulum.com>
Date: Fri, 3 Jan 2025 08:29:02 +0000
Subject: [PATCH] added basic create, drop & reset db functions

---
 dg_osm/db/models.py |  6 ++++--
 dg_osm/db/utils.py  | 13 +++++++++++++
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/dg_osm/db/models.py b/dg_osm/db/models.py
index b6a4796..a32fabf 100644
--- a/dg_osm/db/models.py
+++ b/dg_osm/db/models.py
@@ -1,7 +1,9 @@
-from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, relationship
-from sqlalchemy import Boolean, ForeignKey, String
 from typing import Optional, List
 
+from sqlalchemy import Boolean, ForeignKey, String
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, relationship
+
+
 class Base(DeclarativeBase):
     pass
 
diff --git a/dg_osm/db/utils.py b/dg_osm/db/utils.py
index e69de29..1a60c7f 100644
--- a/dg_osm/db/utils.py
+++ b/dg_osm/db/utils.py
@@ -0,0 +1,13 @@
+from dg_osm.db import engine
+from dg_osm.db import models
+
+
+def create_database() -> None:
+    models.Base.metadata.create_all(engine)
+
+def drop_database() -> None:
+    models.Base.metadata.drop_all(engine)
+
+def reset_database() -> None:
+    drop_database()
+    create_database()