tuning-box/tuning_box/migrations/versions/f16eb4eff7c_initial_revision.py
Yuriy Taraday e35dd5a26c Add foreign key names to all FKs in initial migration
I'm adjusting initial_revision to match what happened on PostgreSQL
databases. Models should be adjusted as well later, and test should be
fixed to actually check foreign keys' names.

Change-Id: I0eb459ac539925569999bdbe1c406c2508e5d3f7
2016-03-18 22:39:37 +03:00

170 lines
6.5 KiB
Python

# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
"""Initial revision
Revision ID: f16eb4eff7c
Revises:
Create Date: 2016-03-02 17:10:04.750584
"""
# revision identifiers, used by Alembic.
revision = 'f16eb4eff7c'
down_revision = None
branch_labels = None
depends_on = None
from alembic import context
from alembic import op
import sqlalchemy as sa
import tuning_box.db
def upgrade():
table_prefix = context.config.get_main_option('table_prefix')
op.create_table(
table_prefix + 'component',
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('name', sa.String(length=128), nullable=True),
)
op.create_table(
table_prefix + 'environment',
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
)
op.create_table(
table_prefix + 'namespace',
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('name', sa.String(length=128), nullable=True),
)
table_name = table_prefix + 'environment_components'
op.create_table(
table_name,
sa.Column('environment_id', sa.Integer(), nullable=True),
sa.Column('component_id', sa.Integer(), nullable=True),
sa.ForeignKeyConstraint(
['component_id'], [table_prefix + 'component.id'],
name=table_name + '_component_id_fkey',
),
sa.ForeignKeyConstraint(
['environment_id'], [table_prefix + 'environment.id'],
name=table_name + '_environment_id_fkey',
),
)
table_name = table_prefix + 'environment_hierarchy_level'
op.create_table(
table_name,
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('environment_id', sa.Integer(), nullable=True),
sa.Column('name', sa.String(length=128), nullable=True),
sa.Column('parent_id', sa.Integer(), nullable=True),
sa.ForeignKeyConstraint(
['environment_id'], [table_prefix + 'environment.id'],
name=table_name + '_environment_id_fkey',
),
sa.ForeignKeyConstraint(
['parent_id'], [table_prefix + 'environment_hierarchy_level.id'],
name=table_name + '_parent_id_fkey',
),
sa.UniqueConstraint(
'environment_id', 'name',
name=table_name + '_environment_id_name_key',
),
sa.UniqueConstraint(
'environment_id', 'parent_id',
name=table_name[:-4] + '_environment_id_parent_id_key',
),
)
table_name = table_prefix + 'schema'
op.create_table(
table_name,
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('name', sa.String(length=128), nullable=True),
sa.Column('component_id', sa.Integer(), nullable=True),
sa.Column('namespace_id', sa.Integer(), nullable=True),
sa.Column('content', tuning_box.db.Json(), nullable=True),
sa.ForeignKeyConstraint(
['component_id'], [table_prefix + 'component.id'],
name=table_name + '_component_id_fkey'),
sa.ForeignKeyConstraint(
['namespace_id'], [table_prefix + 'namespace.id'],
name=table_name + '_namespace_id_fkey'),
)
table_name = table_prefix + 'template'
op.create_table(
table_name,
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('name', sa.String(length=128), nullable=True),
sa.Column('component_id', sa.Integer(), nullable=True),
sa.Column('content', tuning_box.db.Json(), nullable=True),
sa.ForeignKeyConstraint(
['component_id'], [table_prefix + 'component.id'],
name=table_name + '_component_id_fkey',
),
)
table_name = table_prefix + 'environment_hierarchy_level_value'
op.create_table(
table_name,
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('level_id', sa.Integer(), nullable=True),
sa.Column('parent_id', sa.Integer(), nullable=True),
sa.Column('value', sa.String(length=128), nullable=True),
sa.ForeignKeyConstraint(
['level_id'], [table_prefix + 'environment_hierarchy_level.id'],
name=table_name + '_level_id_fkey',
),
sa.ForeignKeyConstraint(
['parent_id'], [table_name + '.id'],
name=table_name + '_parent_id_fkey',
),
)
table_name = table_prefix + 'environment_schema_values'
op.create_table(
table_name,
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('environment_id', sa.Integer(), nullable=True),
sa.Column('schema_id', sa.Integer(), nullable=True),
sa.Column('level_value_id', sa.Integer(), nullable=True),
sa.Column('values', tuning_box.db.Json(), nullable=True),
sa.ForeignKeyConstraint(
['environment_id'], [table_prefix + 'environment.id'],
name=table_name + '_environment_id_fkey',
),
sa.ForeignKeyConstraint(
['level_value_id'],
[table_prefix + 'environment_hierarchy_level_value.id'],
name=table_name + '_level_value_id_fkey',
),
sa.ForeignKeyConstraint(
['schema_id'], [table_prefix + 'schema.id'],
name=table_name + '_schema_id_fkey',
),
sa.UniqueConstraint(
'environment_id', 'schema_id', 'level_value_id',
name=table_name[:-6] + 'environment_id_schema_id_leve_key',
),
)
def downgrade():
table_prefix = context.config.get_main_option('table_prefix')
op.drop_table(table_prefix + 'environment_schema_values')
op.drop_table(table_prefix + 'environment_hierarchy_level_value')
op.drop_table(table_prefix + 'template')
op.drop_table(table_prefix + 'schema')
op.drop_table(table_prefix + 'environment_hierarchy_level')
op.drop_table(table_prefix + 'environment_components')
op.drop_table(table_prefix + 'namespace')
op.drop_table(table_prefix + 'environment')
op.drop_table(table_prefix + 'component')