X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/f27fd052b9f3419cc4048711565db9984ca1a818..9531d8a55106384d86cda24825b5f9a9d769d33d:/apps/catalogue/management/commands/importbooks.py diff --git a/apps/catalogue/management/commands/importbooks.py b/apps/catalogue/management/commands/importbooks.py index c0a5df1ae..0aa808137 100644 --- a/apps/catalogue/management/commands/importbooks.py +++ b/apps/catalogue/management/commands/importbooks.py @@ -1,9 +1,10 @@ import os import sys +from optparse import make_option from django.core.management.base import BaseCommand from django.core.management.color import color_style -from optparse import make_option +from django.core.files import File from catalogue.models import Book @@ -38,7 +39,6 @@ class Command(BaseCommand): for dir_name in directories: if not os.path.isdir(dir_name): print self.style.ERROR("%s: Not a directory. Skipping." % dir_name) - files_skipped += 1 else: for file_name in os.listdir(dir_name): file_path = os.path.join(dir_name, file_name) @@ -46,8 +46,6 @@ class Command(BaseCommand): # Skip files that are not XML files if not ext == '.xml': - print self.style.NOTICE("%s: Not an XML file. Skipping." % file_path) - files_skipped += 1 continue if verbose > 0: @@ -73,7 +71,15 @@ class Command(BaseCommand): book.txt_file.save('%s.txt' % book.slug, File(file(file_base + '.txt'))) if verbose: print "Importing %s.txt" % file_base - + if os.path.isfile(os.path.join(dir_name, book.slug + '.mp3')): + book.mp3_file.save('%s.mp3' % book.slug, File(file(os.path.join(dir_name, book.slug + '.mp3')))) + if verbose: + print "Importing %s.mp3" % book.slug + if os.path.isfile(os.path.join(dir_name, book.slug + '.ogg')): + book.ogg_file.save('%s.ogg' % book.slug, File(file(os.path.join(dir_name, book.slug + '.ogg')))) + if verbose: + print "Importing %s.ogg" % book.slug + book.save() except Book.AlreadyExists, msg: