diff --git a/mansrht-initdb b/mansrht-initdb index d187c09..b890c9a 100755 --- a/mansrht-initdb +++ b/mansrht-initdb @@ -9,6 +9,8 @@ Idempotent. If the tables already exist, they will not be re-created. import mansrht.alembic import mansrht.types +import srht.alembic + from alembic import command from alembic.config import Config from srht.config import cfg @@ -24,3 +26,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")