diff --git a/ippisite/ippidb/management/commands/import_v1_data.py b/ippisite/ippidb/management/commands/import_v1_data.py index df53a3ea9933d445d35960694b12d6a97919af3b..108c344dcdb61d71d1a40110c5baee33bc04d5c1 100644 --- a/ippisite/ippidb/management/commands/import_v1_data.py +++ b/ippisite/ippidb/management/commands/import_v1_data.py @@ -118,11 +118,17 @@ class Command(BaseCommand): b.id_source = row[2] b.save(autofill=True) except Exception as e: - self.stdout.write( - self.style.ERROR('Failed inserting {}'.format(row[2]))) + if options['stoponfail']: + import traceback + self.stderr.write(traceback.format_exc()) + raise CommandError( + 'Failed inserting {} {}'.format(row[1], row[2])) + else: + self.stdout.write( + self.style.ERROR('Failed inserting {} {}'.format(row[1], row[2]))) else: self.stdout.write( - self.style.SUCCESS('Successfully inserted {}'.format(row[2]))) + self.style.SUCCESS('Successfully inserted {} {}'.format(row[1], row[2]))) if options['proteins']: cursor.execute("""SELECT * FROM protein""") rows = cursor.fetchall()