diff --git a/ippisite/ippidb/management/__init__.py b/ippisite/ippidb/management/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/ippisite/ippidb/management/commands/__init__.py b/ippisite/ippidb/management/commands/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/ippisite/ippidb/management/commands/import_v1_data.py b/ippisite/ippidb/management/commands/import_v1_data.py new file mode 100644 index 0000000000000000000000000000000000000000..f6b60f8e6008d4b686bb28c54c3841cea9a6b4f2 --- /dev/null +++ b/ippisite/ippidb/management/commands/import_v1_data.py @@ -0,0 +1,24 @@ +from django.core.management import BaseCommand +import mysql.connector +from ippidb.models import Bibliography + +class Command(BaseCommand): + help = "Import data from the local v1 database" + + def handle(self, *args, **options): + conn = mysql.connector.connect(host="localhost",user="root",password="ippidb", database="ippidb") + cursor = conn.cursor() + cursor.execute("""SELECT * FROM biblio""") + rows = cursor.fetchall() + Bibliography.objects.all().delete() + self.stdout.write(self.style.SUCCESS('Successfully flushed bibliography table')) + for row in rows: + try: + b = Bibliography() + b.source = 'PM' + b.id_source = row[2] + b.save() + except Exception as e: + self.stdout.write(self.style.ERROR('Failed inserting {}'.format(row[2]))) + else: + self.stdout.write(self.style.SUCCESS('Successfully inserted {}'.format(row[2])))