# -*- 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 """Database migration.""" from django.db import migrations from aiida.backends.djsite.db.migrations import upgrade_schema_version REVISION = '1.0.8' DOWN_REVISION = '1.0.7' [docs]class Migration(migrations.Migration): """Database migration.""" dependencies = [ ('db', '0006_delete_dbpath'), ] operations = [ # I am first migrating the wrongly declared returnlinks out of # the InlineCalculations. # This bug is reported #628 https://github.com/aiidateam/aiida-core/issues/628 # There is an explicit check in the code of the inline calculation # ensuring that the calculation returns UNSTORED nodes. # Therefore, no cycle can be created with that migration! # # this command: # 1) selects all links that # - joins an InlineCalculation (or subclass) as input # - joins a Data (or subclass) as output # - is marked as a returnlink. # 2) set for these links the type to 'createlink' migrations.RunSQL( """ UPDATE db_dblink set type='createlink' WHERE db_dblink.id IN ( SELECT db_dblink_1.id FROM db_dbnode AS db_dbnode_1 JOIN db_dblink AS db_dblink_1 ON db_dblink_1.input_id = db_dbnode_1.id JOIN db_dbnode AS db_dbnode_2 ON db_dblink_1.output_id = db_dbnode_2.id WHERE db_dbnode_1.type LIKE 'calculation.inline.%' AND db_dbnode_2.type LIKE 'data.%' AND db_dblink_1.type = 'returnlink' ); """ ), # Now I am updating the link-types that are null because of either an export and subsequent import # https://github.com/aiidateam/aiida-core/issues/685 # or because the link types don't exist because the links were added before the introduction of link types. # This is reported here: https://github.com/aiidateam/aiida-core/issues/687 # # The following sql statement: # 1) selects all links that # - joins Data (or subclass) or Code as input # - joins Calculation (or subclass) as output: includes WorkCalculation, InlineCalcuation, JobCalculations... # - has no type (null) # 2) set for these links the type to 'inputlink' migrations.RunSQL( """ UPDATE db_dblink set type='inputlink' where id in ( SELECT db_dblink_1.id FROM db_dbnode AS db_dbnode_1 JOIN db_dblink AS db_dblink_1 ON db_dblink_1.input_id = db_dbnode_1.id JOIN db_dbnode AS db_dbnode_2 ON db_dblink_1.output_id = db_dbnode_2.id WHERE ( db_dbnode_1.type LIKE 'data.%' or db_dbnode_1.type = 'code.Code.' ) AND db_dbnode_2.type LIKE 'calculation.%' AND ( db_dblink_1.type = null OR db_dblink_1.type = '') ); """ ), # # The following sql statement: # 1) selects all links that # - join JobCalculation (or subclass) or InlineCalculation as input # - joins Data (or subclass) as output. # - has no type (null) # 2) set for these links the type to 'createlink' migrations.RunSQL( """ UPDATE db_dblink set type='createlink' where id in ( SELECT db_dblink_1.id FROM db_dbnode AS db_dbnode_1 JOIN db_dblink AS db_dblink_1 ON db_dblink_1.input_id = db_dbnode_1.id JOIN db_dbnode AS db_dbnode_2 ON db_dblink_1.output_id = db_dbnode_2.id WHERE db_dbnode_2.type LIKE 'data.%' AND ( db_dbnode_1.type LIKE 'calculation.job.%' OR db_dbnode_1.type = 'calculation.inline.InlineCalculation.' ) AND ( db_dblink_1.type = null OR db_dblink_1.type = '') ); """ ), # The following sql statement: # 1) selects all links that # - join WorkCalculation as input. No subclassing was introduced so far, so only one type string is checked # - join Data (or subclass) as output. # - has no type (null) # 2) set for these links the type to 'returnlink' migrations.RunSQL( """ UPDATE db_dblink set type='returnlink' where id in ( SELECT db_dblink_1.id FROM db_dbnode AS db_dbnode_1 JOIN db_dblink AS db_dblink_1 ON db_dblink_1.input_id = db_dbnode_1.id JOIN db_dbnode AS db_dbnode_2 ON db_dblink_1.output_id = db_dbnode_2.id WHERE db_dbnode_2.type LIKE 'data.%' AND db_dbnode_1.type = 'calculation.work.WorkCalculation.' AND ( db_dblink_1.type = null OR db_dblink_1.type = '') ); """ ), # Now I update links that are CALLS: # The following sql statement: # 1) selects all links that # - join WorkCalculation as input. No subclassing was introduced so far, so only one type string is checked # - join Calculation (or subclass) as output. Includes JobCalculation and WorkCalculations and all subclasses. # - has no type (null) # 2) set for these links the type to 'calllink' migrations.RunSQL( """ UPDATE db_dblink set type='calllink' where id in ( SELECT db_dblink_1.id FROM db_dbnode AS db_dbnode_1 JOIN db_dblink AS db_dblink_1 ON db_dblink_1.input_id = db_dbnode_1.id JOIN db_dbnode AS db_dbnode_2 ON db_dblink_1.output_id = db_dbnode_2.id WHERE db_dbnode_1.type = 'calculation.work.WorkCalculation.' AND db_dbnode_2.type LIKE 'calculation.%' AND ( db_dblink_1.type = null OR db_dblink_1.type = '') ); """ ), upgrade_schema_version(REVISION, DOWN_REVISION) ]