diff --git a/ippisite/ippidb/management/commands/import_v1_data.py b/ippisite/ippidb/management/commands/import_v1_data.py index df53a3ea9933d445d35960694b12d6a97919af3b..259d17564fc5a3625e9d12c220cfaa82513fd62d 100644 --- a/ippisite/ippidb/management/commands/import_v1_data.py +++ b/ippisite/ippidb/management/commands/import_v1_data.py @@ -116,13 +116,26 @@ class Command(BaseCommand): else: b.source = 'PT' b.id_source = row[2] + b.cytotox = row[6]=='Y' + b.in_silico = row[7]=='Y' + b.in_vitro = row[8]=='Y' + b.in_vivo = row[9]=='Y' + b.in_cellulo = row[10]=='Y' + b.pharmacokinetic = row[11]=='Y' + b.xray = row[12]=='Y' 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()