# -*- coding: utf-8 -*- ########################################################################### # Copyright (c), The AiiDA team. All rights reserved. # # This file is part of the AiiDA code. # # # # The code is hosted on GitHub at https://github.com/aiidateam/aiida-core # # For further information on the license, see the LICENSE.txt file # # For further information please visit http://www.aiida.net # ########################################################################### # pylint: disable=invalid-name """Data migration for `TrajectoryData` nodes where symbol lists are moved from repository array to attribute. This process has to be done in two separate consecutive migrations to prevent data loss in between. """ # Remove when https://github.com/PyCQA/pylint/issues/1931 is fixed # pylint: disable=no-name-in-module,import-error from django.db import migrations from aiida.backends.djsite.db.migrations import upgrade_schema_version from aiida.backends.general.migrations import utils from . import ModelModifierV0025 REVISION = '1.0.27' DOWN_REVISION = '1.0.26' [docs]def delete_trajectory_symbols_array(apps, _): """Delete the symbols array from all `TrajectoryData` nodes.""" DbNode = apps.get_model('db', 'DbNode') DbAttribute = apps.get_model('db', 'DbAttribute') modifier = ModelModifierV0025(apps, DbAttribute) nodes = DbNode.objects.filter(type='node.data.array.trajectory.TrajectoryData.').values_list('id', 'uuid') for pk, uuid in nodes: modifier.del_value_for_node(DbNode.objects.get(pk=pk), 'array|symbols') utils.delete_numpy_array_from_repository(uuid, 'symbols') [docs]def create_trajectory_symbols_array(apps, _): """Create the symbols array for all `TrajectoryData` nodes.""" import numpy DbNode = apps.get_model('db', 'DbNode') DbAttribute = apps.get_model('db', 'DbAttribute') modifier = ModelModifierV0025(apps, DbAttribute) nodes = DbNode.objects.filter(type='node.data.array.trajectory.TrajectoryData.').values_list('id', 'uuid') for pk, uuid in nodes: symbols = numpy.array(modifier.get_value_for_node(pk, 'symbols')) utils.store_numpy_array_in_repository(uuid, 'symbols', symbols) modifier.set_value_for_node(DbNode.objects.get(pk=pk), 'array|symbols', list(symbols.shape)) [docs]class Migration(migrations.Migration): """Deleting duplicated information stored in TrajectoryData symbols numpy arrays""" dependencies = [ ('db', '0026_trajectory_symbols_to_attribute'), ] operations = [ migrations.RunPython(delete_trajectory_symbols_array, reverse_code=create_trajectory_symbols_array), upgrade_schema_version(REVISION, DOWN_REVISION) ]