from random import random from time import sleep from django.core.management import BaseCommand from media.forms import MediaAdminForm from media.models import BD, FutureMedia, Manga, Roman class Command(BaseCommand): def handle(self, *args, **options): for future_medium in FutureMedia.objects.all(): isbn = future_medium.isbn type_str = future_medium.type if type_str == 'bd': cl = BD elif type_str == 'manga': cl = Manga elif type_str == 'roman': cl = Roman else: self.stderr.write(self.style.WARNING( "Unknown medium type: {type}. Ignoring..." .format(type=type_str))) continue if cl.objects.filter(isbn=isbn).exists(): self.stderr.write(self.style.WARNING( f"ISBN {isbn} for type {type_str} already exists, " f"remove it")) future_medium.delete() continue form = MediaAdminForm(instance=cl(), data={"isbn": isbn, "_isbn": True, }) # Don't DDOS any website sleep(5 + (4 * random() - 1)) try: form.full_clean() if hasattr(form.instance, "subtitle") and \ not form.instance.subtitle: form.instance.subtitle = "" form.save() future_medium.delete() self.stdout.write(self.style.SUCCESS( "Medium with ISBN {isbn} successfully imported" .format(isbn=isbn))) except Exception as e: self.stderr.write(self.style.WARNING( "An error occured while importing ISBN {isbn}: {error}" .format(isbn=isbn, error=str(e.__class__) + "(" + str(e) + ")")))