97 lines
3.2 KiB
Python
97 lines
3.2 KiB
Python
|
"""Add distance matrix.
|
||
|
|
|
||
|
|
Revision: #b4dd0b8903a5 at 2021-03-01 16:14:06
|
||
|
|
Revises: #8bfb928a31f8
|
||
|
|
"""
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
import sqlalchemy as sa
|
||
|
|
from alembic import op
|
||
|
|
from sqlalchemy.dialects import postgresql
|
||
|
|
|
||
|
|
from urban_meal_delivery import configuration
|
||
|
|
|
||
|
|
|
||
|
|
revision = 'b4dd0b8903a5'
|
||
|
|
down_revision = '8bfb928a31f8'
|
||
|
|
branch_labels = None
|
||
|
|
depends_on = None
|
||
|
|
|
||
|
|
|
||
|
|
config = configuration.make_config('testing' if os.getenv('TESTING') else 'production')
|
||
|
|
|
||
|
|
|
||
|
|
def upgrade():
|
||
|
|
"""Upgrade to revision b4dd0b8903a5."""
|
||
|
|
op.create_table(
|
||
|
|
'addresses_addresses',
|
||
|
|
sa.Column('first_address_id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('second_address_id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('city_id', sa.SmallInteger(), nullable=False),
|
||
|
|
sa.Column('air_distance', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('bicycle_distance', sa.Integer(), nullable=True),
|
||
|
|
sa.Column('bicycle_duration', sa.Integer(), nullable=True),
|
||
|
|
sa.Column('directions', postgresql.JSON(), nullable=True),
|
||
|
|
sa.PrimaryKeyConstraint(
|
||
|
|
'first_address_id',
|
||
|
|
'second_address_id',
|
||
|
|
name=op.f('pk_addresses_addresses'),
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
['first_address_id', 'city_id'],
|
||
|
|
[
|
||
|
|
f'{config.CLEAN_SCHEMA}.addresses.id',
|
||
|
|
f'{config.CLEAN_SCHEMA}.addresses.city_id',
|
||
|
|
],
|
||
|
|
name=op.f(
|
||
|
|
'fk_addresses_addresses_to_addresses_via_first_address_id_city_id',
|
||
|
|
),
|
||
|
|
onupdate='RESTRICT',
|
||
|
|
ondelete='RESTRICT',
|
||
|
|
),
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
['second_address_id', 'city_id'],
|
||
|
|
[
|
||
|
|
f'{config.CLEAN_SCHEMA}.addresses.id',
|
||
|
|
f'{config.CLEAN_SCHEMA}.addresses.city_id',
|
||
|
|
],
|
||
|
|
name=op.f(
|
||
|
|
'fk_addresses_addresses_to_addresses_via_second_address_id_city_id',
|
||
|
|
),
|
||
|
|
onupdate='RESTRICT',
|
||
|
|
ondelete='RESTRICT',
|
||
|
|
),
|
||
|
|
sa.UniqueConstraint(
|
||
|
|
'first_address_id',
|
||
|
|
'second_address_id',
|
||
|
|
name=op.f('uq_addresses_addresses_on_first_address_id_second_address_id'),
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
'first_address_id < second_address_id',
|
||
|
|
name=op.f('ck_addresses_addresses_on_distances_are_symmetric_for_bicycles'),
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
'0 <= air_distance AND air_distance < 20000',
|
||
|
|
name=op.f('ck_addresses_addresses_on_realistic_air_distance'),
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
'bicycle_distance < 25000',
|
||
|
|
name=op.f('ck_addresses_addresses_on_realistic_bicycle_distance'),
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
'air_distance <= bicycle_distance',
|
||
|
|
name=op.f('ck_addresses_addresses_on_air_distance_is_shortest'),
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
'0 <= bicycle_duration AND bicycle_duration <= 3600',
|
||
|
|
name=op.f('ck_addresses_addresses_on_realistic_bicycle_travel_time'),
|
||
|
|
),
|
||
|
|
schema=config.CLEAN_SCHEMA,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def downgrade():
|
||
|
|
"""Downgrade to revision 8bfb928a31f8."""
|
||
|
|
op.drop_table('addresses_addresses', schema=config.CLEAN_SCHEMA)
|