diff --git a/todosrht-initdb b/todosrht-initdb index 3dea9aa..dad35c1 100755 --- a/todosrht-initdb +++ b/todosrht-initdb @@ -10,6 +10,8 @@ import todosrht.alembic import todosrht.types import todosrht.webhooks +import srht.alembic + from alembic import command from alembic.config import Config from srht.config import cfg @@ -25,3 +27,8 @@ config = Config() config.set_main_option("sqlalchemy.url", connection_string) config.set_main_option("script_location", alembic_path) command.stamp(config, "head") + +alembic_path = list(srht.alembic.__path__)[0] + +config.set_main_option("script_location", alembic_path) +command.stamp(config, "head")