Coverage for src/backend/InvenTree/build/migrations/0036_auto_20220707_1101.py: 66%

1# Generated by Django 3.2.14 on 2022-07-07 11:01 

2 

3from django.db import migrations 

4 

5 

6def update_build_reference(apps, schema_editor): 

7 """Update the build order reference. 

8 

9 Ref: https://github.com/inventree/InvenTree/pull/3267 

10 

11 Performs the following steps: 

12 - Extract existing 'prefix' value 

13 - Generate a build order pattern based on the prefix value 

14 - Update any existing build order references with the specified prefix 

15 """ 

16 

17 InvenTreeSetting = apps.get_model('common', 'inventreesetting') 

18 

19 try: 

20 prefix = InvenTreeSetting.objects.get(key='BUILDORDER_REFERENCE_PREFIX').value 

21 except Exception: 

22 prefix = 'BO-' 

23 

24 # Construct a reference pattern 

25 pattern = prefix + '{ref:04d}' 

26 

27 # Create or update the BuildOrder.reference pattern 

28 try: 

29 setting = InvenTreeSetting.objects.get(key='BUILDORDER_REFERENCE_PATTERN') 

30 setting.value = pattern 

31 setting.save() 

32 except InvenTreeSetting.DoesNotExist: 

33 setting = InvenTreeSetting.objects.create( 

34 key='BUILDORDER_REFERENCE_PATTERN', 

35 value=pattern, 

36 ) 

37 

38 # Update any existing build order references with the prefix 

39 Build = apps.get_model('build', 'build') 

40 

41 n = 0 

42 

43 for build in Build.objects.all(): 43 ↛ 44line 43 didn't jump to line 44 because the loop on line 43 never started

44 if not build.reference.startswith(prefix): 

45 build.reference = prefix + build.reference 

46 build.save() 

47 n += 1 

48 

49 if n > 0: 49 ↛ 50line 49 didn't jump to line 50 because the condition on line 49 was never true

50 print(f"Updated reference field for {n} BuildOrder objects") 

51 

52 

53class Migration(migrations.Migration): 

54 

55 dependencies = [ 

56 ('build', '0035_alter_build_notes'), 

57 ] 

58 

59 operations = [ 

60 migrations.RunPython( 

61 update_build_reference, 

62 reverse_code=migrations.RunPython.noop, 

63 ) 

64 ]