add database stamp for core.sr.ht

2 years ago · 2b98206e43
parent 981622c92f
commit 2b98206e43
1 changed files with 7 additions and 0 deletions
--- a/7
+++ b/7
@ -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")