X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/3199cbfa76c763b4082b5a8d8f971d74f67e27c0..0cae17bec6d31806615fae59a5b3945016285fbe:/catalogue/management/commands/importbooks.py diff --git a/catalogue/management/commands/importbooks.py b/catalogue/management/commands/importbooks.py deleted file mode 100644 index 22dd237f7..000000000 --- a/catalogue/management/commands/importbooks.py +++ /dev/null @@ -1,48 +0,0 @@ -import os - -from django.core.management.base import BaseCommand -from django.core.management.color import color_style -from optparse import make_option - -from catalogue.models import Book - - -class Command(BaseCommand): - option_list = BaseCommand.option_list + ( - make_option('--verbosity', action='store', dest='verbosity', default='1', - type='choice', choices=['0', '1', '2'], - help='Verbosity level; 0=minimal output, 1=normal output, 2=all output'), - ) - help = 'Imports books from the specified directories.' - args = 'directory [directory ...]' - - def handle(self, *directories, **options): - from django.db import transaction - - self.style = color_style() - - verbosity = int(options.get('verbosity', 1)) - show_traceback = options.get('traceback', False) - - # Start transaction management. - transaction.commit_unless_managed() - transaction.enter_transaction_management() - transaction.managed(True) - - for dir_name in directories: - if not os.path.isdir(dir_name): - print self.style.ERROR("Skipping '%s': not a directory." % dir_name) - else: - for file_name in os.listdir(dir_name): - file_path = os.path.join(dir_name, file_name) - if not os.path.splitext(file_name)[1] == '.xml': - print self.style.NOTICE("Skipping '%s': not an XML file." % file_path) - continue - if verbosity > 0: - print "Parsing '%s'" % file_path - - Book.from_xml_file(file_path) - - transaction.commit() - transaction.leave_transaction_management() -