GPQBot

1
import pathlib
2
import sys
3
from logging.config import fileConfig
4

5
from alembic import context
6
from sqlalchemy import engine_from_config, pool
7

8
# Import models to make them visible by alembic
9

10
# init config
11
sys.path.append(str(pathlib.Path(__file__).resolve().parents[3]))
12

13
# Imports from `app` should go after `path` patch
14
from app.settings import settings  # isort:skip
15
from app.db.sqlalchemy import Base, make_url_sync  # isort:skip
16

17

18
postgres_dsn = make_url_sync(settings.POSTGRES_DSN)
19
context_config = context.config
20
fileConfig(context_config.config_file_name)
21
target_metadata = Base.metadata
22
context_config.set_main_option("sqlalchemy.url", postgres_dsn)
23

24

25
def run_migrations_online() -> None:
26
    connectable = engine_from_config(
27
        context_config.get_section(context_config.config_ini_section),
28
        prefix="sqlalchemy.",
29
        poolclass=pool.NullPool,
30
    )
31

32
    with connectable.connect() as connection:
33
        context.configure(connection=connection, target_metadata=target_metadata)
34

35
        with context.begin_transaction():
36
            context.run_migrations()
37

38

39
run_migrations_online()