config/sysinv/sysinv/sysinv/sysinv/db/sqlalchemy/migrate_repo/versions/054_system_security_profile.py

36 lines
1.2 KiB
Python

# vim: tabstop=4 shiftwidth=4 softtabstop=4
#
# Copyright (c) 2017 Wind River Systems, Inc.
#
# SPDX-License-Identifier: Apache-2.0
#
from sqlalchemy import Column, MetaData, Table
from sqlalchemy import String
from sysinv.common import constants
def _populate_security_profile(system_table):
sys = list(system_table.select().where(system_table.c.uuid is not None).execute())
if len(sys) > 0:
if sys[0].security_profile is None:
# the Extended Security Profile has to explicitly selected on boot,
# if this is missing then assume a Standard Security Profile
system_table.update().where(system_table.c.uuid == sys[0].uuid).\
values({'security_profile': constants.SYSTEM_SECURITY_PROFILE_STANDARD}).execute()
def upgrade(migrate_engine):
meta = MetaData()
meta.bind = migrate_engine
i_system = Table('i_system', meta, autoload=True)
i_system.create_column(Column('security_profile', String(255)))
_populate_security_profile(i_system)
def downgrade(migrate_engine):
# As per other openstack components, downgrade is
# unsupported in this release.
raise NotImplementedError('SysInv database downgrade is unsupported.')