65 lines
2.2 KiB
Python
Executable File
65 lines
2.2 KiB
Python
Executable File
#
|
|
# Copyright (c) 2016 Wind River Systems, Inc.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
import json
|
|
|
|
from nfv_vim.database import model
|
|
|
|
from nfv_common import debug
|
|
|
|
DLOG = debug.debug_get_logger('nfv_vim.database')
|
|
|
|
|
|
def _migrate_hosts_v5_to_v6(session, hosts_v5, hosts_v6):
|
|
"""
|
|
Migrate host_v5 table to host_v6 table
|
|
"""
|
|
if 0 == len(hosts_v6):
|
|
for host_v5 in hosts_v5:
|
|
host_v6 = model.Host_v6()
|
|
host_v6.data = host_v5.data
|
|
nfvi_host_data = json.loads(host_v5.nfvi_host_data)
|
|
nfvi_host_data['openstack_compute'] = False
|
|
nfvi_host_data['openstack_control'] = False
|
|
nfvi_host_data['remote_storage'] = False
|
|
host_v6.nfvi_host_data = json.dumps(nfvi_host_data)
|
|
session.add(host_v6)
|
|
|
|
|
|
def _migrate_instance_types_v4_to_v5(session, instance_types_v4,
|
|
instance_types_v5):
|
|
"""
|
|
Migrate instance_types_v4 table to instance_types_v5 table
|
|
"""
|
|
if 0 == len(instance_types_v5):
|
|
for instance_type_v4 in instance_types_v4:
|
|
instance_type_v5 = model.InstanceType_v5()
|
|
del instance_type_v4.data['storage_type']
|
|
instance_type_v5.data = instance_type_v4.data
|
|
session.add(instance_type_v5)
|
|
|
|
|
|
def migrate_tables(session, table_names):
|
|
"""
|
|
Migrate database tables
|
|
"""
|
|
if 'hosts_v5' in table_names and 'hosts_v6' in table_names:
|
|
hosts_v5_query = session.query(model.Host_v5)
|
|
hosts_v5 = hosts_v5_query.all()
|
|
hosts_v6_query = session.query(model.Host_v6)
|
|
hosts_v6 = hosts_v6_query.all()
|
|
_migrate_hosts_v5_to_v6(session, hosts_v5, hosts_v6)
|
|
hosts_v5_query.delete()
|
|
|
|
if 'instance_types_v4' in table_names and \
|
|
'instance_types_v5' in table_names:
|
|
instance_types_v4_query = session.query(model.InstanceType)
|
|
instance_types_v4 = instance_types_v4_query.all()
|
|
instance_types_v5_query = session.query(model.InstanceType_v5)
|
|
instance_types_v5 = instance_types_v5_query.all()
|
|
_migrate_instance_types_v4_to_v5(session, instance_types_v4,
|
|
instance_types_v5)
|
|
instance_types_v4_query.delete()
|