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