From 87f1f94399e55d95e9fa9c12ca17dda87a34ce7e Mon Sep 17 00:00:00 2001 From: Matthew Dillon Date: Sat, 3 Mar 2018 14:30:50 -0700 Subject: [PATCH] MAINT: Add new collection types (#56) Fixes #50 --- .../0008_DATA_new_collection_type_values.py | 23 +++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 ccdb/collections_ccdb/migrations/0008_DATA_new_collection_type_values.py diff --git a/ccdb/collections_ccdb/migrations/0008_DATA_new_collection_type_values.py b/ccdb/collections_ccdb/migrations/0008_DATA_new_collection_type_values.py new file mode 100644 index 0000000..badacb4 --- /dev/null +++ b/ccdb/collections_ccdb/migrations/0008_DATA_new_collection_type_values.py @@ -0,0 +1,23 @@ +from django.db import migrations +from django.forms import modelform_factory + + +class Migration(migrations.Migration): + def migrate(apps, schema_editor): + CollectionType = apps.get_model('collections_ccdb', 'CollectionType') + CollectionTypeForm = modelform_factory(CollectionType, + fields=('name',)) + for ct in ['Juvenile', 'Mixed Ages']: + form = CollectionTypeForm(dict(name=ct)) + if form.is_valid(): + CollectionType.objects.create(**form.cleaned_data) + else: + print('collection type', form.errors.as_data()) + + dependencies = [ + ('collections_ccdb', '0007_collection_measurements'), + ] + + operations = [ + migrations.RunPython(migrate), + ]