From 7c876348d7b9500922091d857f11e1494c9a7f1e Mon Sep 17 00:00:00 2001 From: Matthew Smith Date: Tue, 10 Dec 2019 22:50:28 +0000 Subject: [PATCH] Asset fixes (#383) --- ...5_1937_squashed_0009_auto_20191205_2041.py | 60 ------------------- assets/migrations/0008_auto_20191206_2124.py | 51 ++++++++++++++++ 2 files changed, 51 insertions(+), 60 deletions(-) delete mode 100644 assets/migrations/0008_auto_20191205_1937_squashed_0009_auto_20191205_2041.py create mode 100644 assets/migrations/0008_auto_20191206_2124.py diff --git a/assets/migrations/0008_auto_20191205_1937_squashed_0009_auto_20191205_2041.py b/assets/migrations/0008_auto_20191205_1937_squashed_0009_auto_20191205_2041.py deleted file mode 100644 index edddb178..00000000 --- a/assets/migrations/0008_auto_20191205_1937_squashed_0009_auto_20191205_2041.py +++ /dev/null @@ -1,60 +0,0 @@ -# Generated by Django 2.0.13 on 2019-12-05 20:42 - -import re -from django.db import migrations, models -import django.db.migrations.operations.special - -def forwards(apps, schema_editor): - AssetModel = apps.get_model('assets', 'Asset') - - for row in AssetModel.objects.all(): - - row.asset_id = row.asset_id.upper() - asset_search = re.search("^([A-Z0-9]*?[A-Z]?)([0-9]+)$", row.asset_id) - if asset_search is None: # If the asset_id doesn't have a number at the end - row.asset_id += "1" - - asset_search = re.search("^([A-Z0-9]*?[A-Z]?)([0-9]+)$", row.asset_id) - row.asset_id_prefix = asset_search.group(1) - row.asset_id_number = int(asset_search.group(2)) - - row.save(update_fields=['asset_id', 'asset_id_prefix', 'asset_id_number']) - -# Functions from the following migrations need manual copying. -# Move them and any dependencies into this file, then update the -# RunPython operations to refer to the local versions: -# assets.migrations.0008_auto_20191205_1937 - -class Migration(migrations.Migration): - - replaces = [('assets', '0008_auto_20191205_1937'), ('assets', '0009_auto_20191205_2041')] - - dependencies = [ - ('assets', '0007_auto_20190108_0202_squashed_0014_auto_20191017_2052'), - ] - - operations = [ - migrations.AddField( - model_name='asset', - name='asset_id_number', - field=models.IntegerField(default=1), - ), - migrations.AddField( - model_name='asset', - name='asset_id_prefix', - field=models.CharField(default='', max_length=8), - ), - migrations.RunPython( - code=forwards, - reverse_code=django.db.migrations.operations.special.RunPython.noop, - ), - migrations.AlterModelOptions( - name='asset', - options={'ordering': ['asset_id_prefix', 'asset_id_number'], 'permissions': (('asset_finance', 'Can see financial data for assets'), ('view_asset', 'Can view an asset'))}, - ), - migrations.AlterField( - model_name='asset', - name='asset_id', - field=models.CharField(max_length=15, unique=True), - ) - ] diff --git a/assets/migrations/0008_auto_20191206_2124.py b/assets/migrations/0008_auto_20191206_2124.py new file mode 100644 index 00000000..f9002182 --- /dev/null +++ b/assets/migrations/0008_auto_20191206_2124.py @@ -0,0 +1,51 @@ +# Generated by Django 2.0.13 on 2019-12-06 21:24 + +from django.db import migrations, models, transaction +import re + +def forwards(apps, schema_editor): + AssetModel = apps.get_model('assets', 'Asset') + with transaction.atomic(): + for row in AssetModel.objects.all(): + + row.asset_id = row.asset_id.upper() + asset_search = re.search("^([A-Z0-9]*?[A-Z]?)([0-9]+)$", row.asset_id) + if asset_search is None: # If the asset_id doesn't have a number at the end + row.asset_id += "1" + + asset_search = re.search("^([A-Z0-9]*?[A-Z]?)([0-9]+)$", row.asset_id) + row.asset_id_prefix = asset_search.group(1) + row.asset_id_number = int(asset_search.group(2)) + + row.save(update_fields=['asset_id', 'asset_id_prefix', 'asset_id_number']) +class Migration(migrations.Migration): + + dependencies = [ + ('assets', '0007_auto_20190108_0202_squashed_0014_auto_20191017_2052'), + ] + + operations = [ + migrations.AlterModelOptions( + name='asset', + options={'ordering': ['asset_id_prefix', 'asset_id_number'], 'permissions': (('asset_finance', 'Can see financial data for assets'), ('view_asset', 'Can view an asset'))}, + ), + migrations.AddField( + model_name='asset', + name='asset_id_number', + field=models.IntegerField(default=1), + ), + migrations.AddField( + model_name='asset', + name='asset_id_prefix', + field=models.CharField(default='', max_length=8), + ), + migrations.AlterField( + model_name='asset', + name='asset_id', + field=models.CharField(max_length=15, unique=True), + ), + migrations.RunPython( + code=forwards, + reverse_code=migrations.operations.special.RunPython.noop, + ), + ]