Alexander Hess
f996376b13
- add Grid, Pixel, and AddressPixelAssociation ORM models - each Grid belongs to a City an is characterized by the side_length of all the square Pixels contained in it - Pixels aggregate Addresses => many-to-many relationship (that is modeled with SQLAlchemy's Association Pattern to implement a couple of constraints)
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
"""Configure Alembic's migration environment."""
|
|
|
|
import os
|
|
from logging import config as log_config
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import context
|
|
|
|
from urban_meal_delivery import config as umd_config
|
|
from urban_meal_delivery import db
|
|
|
|
|
|
# Disable the --sql option, a.k.a, the "offline mode".
|
|
if context.is_offline_mode():
|
|
raise NotImplementedError('The --sql option is not implemented in this project')
|
|
|
|
|
|
# Set up the default Python logger from the alembic.ini file.
|
|
log_config.fileConfig(context.config.config_file_name)
|
|
|
|
|
|
def include_object(obj, _name, type_, _reflected, _compare_to):
|
|
"""Only include the clean schema into --autogenerate migrations."""
|
|
if type_ in {'table', 'column'} and obj.schema != umd_config.CLEAN_SCHEMA:
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
engine = sa.create_engine(umd_config.DATABASE_URI)
|
|
|
|
with engine.connect() as connection:
|
|
context.configure(
|
|
connection=connection,
|
|
include_object=include_object,
|
|
target_metadata=db.Base.metadata,
|
|
version_table='{alembic_table}{test_schema}'.format(
|
|
alembic_table=umd_config.ALEMBIC_TABLE,
|
|
test_schema=(f'_{umd_config.CLEAN_SCHEMA}' if os.getenv('TESTING') else ''),
|
|
),
|
|
version_table_schema=umd_config.ALEMBIC_TABLE_SCHEMA,
|
|
)
|
|
|
|
with context.begin_transaction():
|
|
context.run_migrations()
|