-# -*- coding: utf-8 -*-
-#
# This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
#
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
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')
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
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:
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)