add database stamp for core.sr.ht

This commit is contained in:
gildarts 2020-08-10 13:03:43 -04:00 committed by Drew DeVault
parent 2125de418f
commit 8c9312c3cd
1 changed files with 7 additions and 0 deletions

View File

@ -9,6 +9,8 @@ Idempotent. If the tables already exist, they will not be re-created.
import mansrht.alembic import mansrht.alembic
import mansrht.types import mansrht.types
import srht.alembic
from alembic import command from alembic import command
from alembic.config import Config from alembic.config import Config
from srht.config import cfg from srht.config import cfg
@ -24,3 +26,8 @@ config = Config()
config.set_main_option("sqlalchemy.url", connection_string) config.set_main_option("sqlalchemy.url", connection_string)
config.set_main_option("script_location", alembic_path) config.set_main_option("script_location", alembic_path)
command.stamp(config, "head") command.stamp(config, "head")
alembic_path = list(srht.alembic.__path__)[0]
config.set_main_option("script_location", alembic_path)
command.stamp(config, "head")