Add revues

media/management/commands/import_marvel.py

@@ -0,0 +1,50 @@
+from argparse import FileType
+from sys import stdin
+
+from django.core.management import BaseCommand
+from media.models import Auteur, BD
+
+
+class Command(BaseCommand):
+    def add_arguments(self, parser):
+        parser.add_argument('input', nargs='?',
+                            type=FileType('r'),
+                            default=stdin,
+                            help="Marvel comic to be imported.")
+
+    def handle(self, *args, **options):
+        file = options["input"]
+        revues = []
+        for line in file:
+            revues.append(line[:-1].split('|', 2))
+
+        print("Registering", len(revues), "Marvel comics")
+
+        imported = 0
+
+        for revue in revues:
+            if len(revue) != 3:
+                continue
+
+            title = revue[0]
+            number = revue[1]
+            authors = [Auteur.objects.get_or_create(name=n)[0]
+                       for n in revue[2].split('|')]
+            bd = BD.objects.create(
+                title=title,
+                subtitle=number,
+                side_identifier="{:.3} {:.3} {:0>2}"
+                                .format(authors[0].name.upper(),
+                                        title.upper(),
+                                        number),
+            )
+
+            bd.authors.set(authors)
+            bd.save()
+
+            self.stdout.write(self.style.SUCCESS(
+                "Comic imported"))
+            imported += 1
+
+        self.stdout.write(self.style.SUCCESS(
+            "{count} comics imported".format(count=imported)))