Hierarchy removed from level values

We don't need to have duplicate of levels hierarchy in the
level values. Without duplication we can easy change hierarchy
levels in the environment.
Cascade deletion added on level deletion to level values.

Change-Id: I9aded18c93b8c3f0f08e59817d33e7d21ff12d54
This commit is contained in:
Alexander Kislitsky 2016-08-18 17:48:56 +03:00
parent b7369eaa26
commit a21d1a1212
6 changed files with 95 additions and 27 deletions

View File

@ -185,20 +185,15 @@ class EnvironmentHierarchyLevel(ModelMixin, db.Model):
class EnvironmentHierarchyLevelValue(ModelMixin, db.Model):
level_id = fk(EnvironmentHierarchyLevel)
level_id = fk(EnvironmentHierarchyLevel, ondelete='CASCADE')
level = db.relationship(EnvironmentHierarchyLevel)
value = db.Column(db.String(128))
@sa_decl.declared_attr
def parent_id(cls):
return db.Column(pk_type, db.ForeignKey(cls.id))
__table_args__ = (
db.UniqueConstraint('level_id', 'value'),
)
@sa_decl.declared_attr
def parent(cls):
return db.relationship(cls, remote_side=cls.id)
# TODO(yorik-sar): add UniqueConstraint for all fields
__repr_attrs__ = ('id', 'level', 'parent', 'value')
__repr_attrs__ = ('id', 'level', 'value')
class ResourceValues(ModelMixin, db.Model):

View File

@ -18,7 +18,6 @@ from tuning_box import db
def iter_environment_level_values(environment, levels):
env_levels = db.EnvironmentHierarchyLevel.get_for_environment(environment)
level_pairs = zip(env_levels, levels)
parent_level_value = None
for env_level, (level_name, level_value) in level_pairs:
if env_level.name != level_name:
raise werkzeug.exceptions.BadRequest(
@ -27,11 +26,9 @@ def iter_environment_level_values(environment, levels):
level_value_db = db.get_or_create(
db.EnvironmentHierarchyLevelValue,
level=env_level,
parent=parent_level_value,
value=level_value,
)
yield level_value_db
parent_level_value = level_value_db
def get_environment_level_value(environment, levels):

View File

@ -0,0 +1,78 @@
# 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.
"""Remove hierarchy for level values
Revision ID: a86472389a70
Revises: 0c586adad733
Create Date: 2016-08-18 14:00:03.197693
"""
# revision identifiers, used by Alembic.
revision = 'a86472389a70'
down_revision = '0c586adad733'
branch_labels = None
depends_on = None
from alembic import context
from alembic import op
import sqlalchemy as sa
def upgrade():
table_prefix = context.config.get_main_option('table_prefix')
table_name = table_prefix + 'environment_hierarchy_level_value'
with op.batch_alter_table(table_name) as batch:
batch.drop_column('parent_id')
batch.drop_constraint(
table_name + '_level_id_fkey',
type_='foreignkey'
)
batch.create_foreign_key(
table_name + '_level_id_fkey',
table_prefix + 'environment_hierarchy_level',
['level_id'], ['id'], ondelete='CASCADE'
)
batch.create_unique_constraint(
table_name + '_level_id_value_unique',
['level_id', 'value']
)
def downgrade():
table_prefix = context.config.get_main_option('table_prefix')
table_name = table_prefix + 'environment_hierarchy_level_value'
with op.batch_alter_table(table_name) as batch:
batch.drop_constraint(
table_name + '_level_id_value_unique',
type_='unique'
)
batch.drop_constraint(
table_name + '_level_id_fkey',
type_='foreignkey'
)
batch.create_foreign_key(
table_name + '_level_id_fkey',
table_prefix + 'environment_hierarchy_level',
['level_id'], ['id']
)
batch.add_column(sa.Column('parent_id', sa.Integer(), nullable=True))
batch.create_foreign_key(
table_name + '_parent_id_fkey',
table_name,
['parent_id'], ['id']
)

View File

@ -38,11 +38,10 @@ class TestLevelsHierarchy(BaseTest):
self.assertIsNotNone(level_value)
self.assertEqual(level_value.level.name, 'lvl2')
self.assertEqual(level_value.value, 'val2')
level_value = level_value.parent
self.assertIsNotNone(level_value)
self.assertEqual(level_value.level.name, 'lvl1')
self.assertEqual(level_value.value, 'val1')
self.assertIsNone(level_value.parent)
level = level_value.level.parent
self.assertIsNotNone(level)
self.assertEqual(level.name, 'lvl1')
self.assertIsNone(level.parent)
def test_get_environment_level_value_bad_level(self):
self._fixture()

View File

@ -51,11 +51,10 @@ class TestResourceOverrides(BaseTest):
self.assertIsNotNone(level_value)
self.assertEqual(level_value.level.name, 'lvl2')
self.assertEqual(level_value.value, 'val2')
level_value = level_value.parent
self.assertIsNotNone(level_value)
self.assertEqual(level_value.level.name, 'lvl1')
self.assertEqual(level_value.value, 'val1')
self.assertIsNone(level_value.parent)
level = level_value.level.parent
self.assertIsNotNone(level)
self.assertEqual(level.name, 'lvl1')
self.assertIsNone(level.parent)
def test_get_resource_values_local_override(self):
self._fixture()

View File

@ -50,10 +50,10 @@ class TestResourceValues(BaseTest):
level_value = resource_values.level_value
self.assertEqual(level_value.level.name, 'lvl2')
self.assertEqual(level_value.value, 'val2')
level_value = level_value.parent
self.assertEqual(level_value.level.name, 'lvl1')
self.assertEqual(level_value.value, 'val1')
self.assertIsNone(level_value.parent)
level = level_value.level.parent
self.assertIsNotNone(level)
self.assertEqual(level.name, 'lvl1')
self.assertIsNone(level.parent)
def test_put_resource_values_bad_level(self):
self._fixture()