Просмотр исходного кода

Suppress migration output during testing

jeremystretch 4 лет назад
Родитель
Сommit
ca44cda112
1 измененных файлов с 3 добавлено и 1 удалено
  1. 3 1
      netbox/ipam/migrations/0048_prefix_populate_depth_children.py

+ 3 - 1
netbox/ipam/migrations/0048_prefix_populate_depth_children.py

@@ -1,3 +1,4 @@
+import sys
 from django.db import migrations
 
 from ipam.utils import rebuild_prefixes
@@ -22,7 +23,8 @@ def populate_prefix_hierarchy(apps, schema_editor):
     VRF = apps.get_model('ipam', 'VRF')
 
     total_count = Prefix.objects.count()
-    print(f'\nUpdating {total_count} prefixes...')
+    if 'test' not in sys.argv:
+        print(f'\nUpdating {total_count} prefixes...')
 
     # Rebuild the global table
     rebuild_prefixes(None)