diff --git a/hubsrht-initdb b/hubsrht-initdb index 2e7bb65..d290a8a 100755 --- a/hubsrht-initdb +++ b/hubsrht-initdb @@ -9,6 +9,8 @@ Idempotent. If the tables already exist, they will not be re-created. import hubsrht.alembic import hubsrht.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")