diff options
Diffstat (limited to 'rrs/migrations/0013_reup_layerbranch_populate.py')
-rw-r--r-- | rrs/migrations/0013_reup_layerbranch_populate.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/rrs/migrations/0013_reup_layerbranch_populate.py b/rrs/migrations/0013_reup_layerbranch_populate.py new file mode 100644 index 0000000000..f5e88a76c4 --- /dev/null +++ b/rrs/migrations/0013_reup_layerbranch_populate.py @@ -0,0 +1,29 @@ +# -*- coding: utf-8 -*- +from __future__ import unicode_literals + +from django.db import migrations, models +import settings + + +def populate_layerbranch(apps, schema_editor): + RecipeUpstreamHistory = apps.get_model('rrs', 'RecipeUpstreamHistory') + LayerBranch = apps.get_model('layerindex', 'LayerBranch') + if not settings.CORE_LAYER_NAME: + raise Exception('Please set CORE_LAYER_NAME in settings.py') + core_layerbranch = LayerBranch.objects.filter(layer__name=settings.CORE_LAYER_NAME).first() + if not core_layerbranch: + raise Exception('Unable to find core layer "%s" specified in CORE_LAYER_NAME in settings.py' % settings.CORE_LAYER_NAME) + for row in RecipeUpstreamHistory.objects.all(): + row.layerbranch = core_layerbranch + row.save() + + +class Migration(migrations.Migration): + + dependencies = [ + ('rrs', '0012_reup_layerbranch_field'), + ] + + operations = [ + migrations.RunPython(populate_layerbranch, reverse_code=migrations.RunPython.noop), + ] |