added basic create, drop & reset db functions

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
 

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()