X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/0d32008f4ab55669ec5eb13ed1af3c2741092ef5..3204e4303148302d278036eebcfc8cb105cc97d7:/src/catalogue/management/commands/insert_isbn.py?ds=sidebyside diff --git a/src/catalogue/management/commands/insert_isbn.py b/src/catalogue/management/commands/insert_isbn.py index 7548cb1f..e342c90d 100644 --- a/src/catalogue/management/commands/insert_isbn.py +++ b/src/catalogue/management/commands/insert_isbn.py @@ -8,8 +8,6 @@ import csv import sys from django.contrib.auth.models import User from lxml import etree -from optparse import make_option - from collections import defaultdict from django.core.management import BaseCommand @@ -42,24 +40,20 @@ def url_for_format(slug, format): class Command(BaseCommand): - option_list = BaseCommand.option_list + ( - # make_option('-q', '--quiet', action='store_false', dest='verbose', - # default=True, help='Less output'), - # make_option('-d', '--dry-run', action='store_true', dest='dry_run', - # default=False, help="Don't actually touch anything"), - make_option( - '-u', '--username', dest='username', metavar='USER', - help='Assign commits to this user (required, preferably yourself).'), - ) args = 'csv_file' + def add_arguments(self, parser): + self.add_argument( + '-u', '--username', dest='username', metavar='USER', + help='Assign commits to this user (required, preferably yourself).') + def handle(self, csv_file, **options): username = options.get('username') if username: user = User.objects.get(username=username) else: - print 'Please provide a username.' + print('Please provide a username.') sys.exit(1) csvfile = open(csv_file, 'rb') @@ -69,7 +63,7 @@ class Command(BaseCommand): csvfile.close() for slug, isbn_list in isbn_lists.iteritems(): - print 'processing %s' % slug + print('processing %s' % slug) book = Book.objects.get(dc_slug=slug) chunk = book.chunk_set.first() old_head = chunk.head @@ -77,7 +71,7 @@ class Command(BaseCommand): tree = etree.fromstring(src) isbn_node = tree.find('.//' + DCNS("relation.hasFormat")) if isbn_node is not None: - print '%s already contains ISBN metadata, skipping' % slug + print('%s already contains ISBN metadata, skipping' % slug) continue desc = tree.find(".//" + RDFNS("Description")) for format, isbn in isbn_list: @@ -92,12 +86,12 @@ class Command(BaseCommand): element.tail = '\n' desc.append(element) new_head = chunk.commit( - etree.tostring(tree, encoding=unicode), + etree.tostring(tree, encoding='unicode'), author=user, description='automatyczne dodanie isbn' ) - print 'committed %s' % slug + print('committed %s' % slug) if old_head.publishable: new_head.set_publishable(True) else: - print 'Warning: %s not publishable' % slug + print('Warning: %s not publishable' % slug)