Is it possible to store the alembic connect string outside of alembic.ini?

后端 未结 10 1604
[愿得一人]
[愿得一人] 2021-01-30 15:41

I\'m using Alembic with SQL Alchemy. With SQL Alchemy, I tend to follow a pattern where I don\'t store the connect string with the versioned code. Instead I have file secr

10条回答
  •  别那么骄傲
    2021-01-30 16:21

    I was looking for a while how to manage this for mutli-databases

    Here is what I did. I have two databases : logs and ohlc

    According to the doc, I have setup the alembic like that

    alembic init --template multidb
    

    alembic.ini

    databases = logs, ohlc
    [logs]
    sqlalchemy.url = postgresql://botcrypto:botcrypto@localhost/logs
    [ohlc]
    sqlalchemy.url = postgresql://botcrypto:botcrypto@localhost/ohlc
    

    env.py

    [...]
    # this is the Alembic Config object, which provides
    # access to the values within the .ini file in use.
    config = context.config
    
    # Interpret the config file for Python logging.
    # This line sets up loggers basically.
    fileConfig(config.config_file_name)
    logger = logging.getLogger('alembic.env')
    
    # overwrite alembic.ini db urls from the config file
    settings_path = os.environ.get('SETTINGS')
    if settings_path:
        with open(settings_path) as fd:
            settings = conf.load(fd, context=os.environ) # loads the config.yml
        config.set_section_option("ohlc", "sqlalchemy.url", settings["databases"]["ohlc"])
        config.set_section_option("logs", "sqlalchemy.url", settings["databases"]["logs"])
    else:
        logger.warning('Environment variable SETTINGS missing - use default alembic.ini configuration')
    [...]
    

    config.yml

    databases:
        logs: postgresql://botcrypto:botcrypto@127.0.0.1:5432/logs
        ohlc: postgresql://botcrypto:botcrypto@127.0.0.1:5432/ohlc
    

    usage

    SETTINGS=config.yml alembic upgrade head
    

    Hope it can helps !

提交回复
热议问题