Suppress migration output during testing

This commit is contained in:
jeremystretch 2021-06-02 16:02:38 -04:00
parent f34fb28d59
commit 15d26dfe1f

View File

@ -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)