more presets and timing functionality to data generator

This commit is contained in:
AlanCoding
2017-01-25 09:16:07 -05:00
parent f265d2fe0f
commit c22cf2f7e6
2 changed files with 20 additions and 15 deletions

View File

@@ -7,6 +7,7 @@ import sys
# Python
from collections import defaultdict
from optparse import make_option, OptionParser
from datetime import datetime
import logging
@@ -184,6 +185,9 @@ def mock_save(self, *args, **kwargs):
PrimordialModel.save = mock_save
startTime = datetime.now()
try:
with transaction.atomic():
@@ -694,3 +698,4 @@ except Rollback:
pass
print('')
print('script execution time: {}'.format(datetime.now() - startTime))