1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495 |
- from __future__ import with_statement
- import os
- from alembic import context
- from sqlalchemy import engine_from_config, pool
- from logging.config import fileConfig
- # 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)
- # add your model's MetaData object here
- # for 'autogenerate' support
- # from myapp import mymodel
- # target_metadata = mymodel.Base.metadata
- # target_metadata = None
- from app.db.base import Base # noqa
- target_metadata = Base.metadata
- # other values from the config, defined by the needs of env.py,
- # can be acquired:
- # my_important_option = config.get_main_option("my_important_option")
- # ... etc.
- '''
- def get_url():
- user = os.getenv("POSTGRES_USER", "postgres")
- password = os.getenv("POSTGRES_PASSWORD", "")
- server = os.getenv("POSTGRES_SERVER", "db")
- db = os.getenv("POSTGRES_DB", "app")
- return f"postgresql://{user}:{password}@{server}/{db}"
- '''
- def get_url():
- user = "choozmo"
- password = "pAssw0rd"
- host = "db.ptt.cx:3306"
- db_name = "video_maker"
- return f'mysql://{user}:{password}@{host}/{db_name}'
- def run_migrations_offline():
- """Run migrations in 'offline' mode.
- This configures the context with just a URL
- and not an Engine, though an Engine is acceptable
- here as well. By skipping the Engine creation
- we don't even need a DBAPI to be available.
- Calls to context.execute() here emit the given string to the
- script output.
- """
- url = get_url()
- context.configure(
- url=url, target_metadata=target_metadata, literal_binds=True, compare_type=True
- )
- with context.begin_transaction():
- context.run_migrations()
- def run_migrations_online():
- """Run migrations in 'online' mode.
- In this scenario we need to create an Engine
- and associate a connection with the context.
- """
- configuration = config.get_section(config.config_ini_section)
- configuration["sqlalchemy.url"] = get_url()
- connectable = engine_from_config(
- configuration, prefix="sqlalchemy.", poolclass=pool.NullPool,
- )
- with connectable.connect() as connection:
- context.configure(
- connection=connection, target_metadata=target_metadata, compare_type=True
- )
- with context.begin_transaction():
- context.run_migrations()
- if context.is_offline_mode():
- run_migrations_offline()
- else:
- run_migrations_online()
|