2ab0510c4b55043e36e252ab9df42d3f0d696e81
[redakcja.git] / apps / catalogue / management / commands / add_parent.py
1 # -*- coding: utf-8 -*-
2 #
3 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
5 #
6 import sys
7
8 from datetime import date
9 from lxml import etree
10
11 from django.core.management import BaseCommand
12
13 from catalogue.models import Book
14 from librarian import RDFNS, DCNS
15
16 TEMPLATE = '''<utwor>
17 <rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#">
18 <rdf:Description rdf:about="http://redakcja.wolnelektury.pl/documents/book/%(slug)s/">
19 %(dc)s
20 </rdf:Description>
21 </rdf:RDF>
22
23 </utwor>
24 '''
25
26 DC_TEMPLATE = '<dc:%(tag)s xml:lang="pl" xmlns:dc="http://purl.org/dc/elements/1.1/">%(value)s</dc:%(tag)s>'
27
28 DC_TAGS = (
29     'creator',
30     'title',
31     'relation.hasPart',
32     'contributor.translator',
33     'contributor.editor',
34     'contributor.technical_editor',
35     'contributor.funding',
36     'contributor.thanks',
37     'publisher',
38     'subject.period',
39     'subject.type',
40     'subject.genre',
41     'description',
42     'identifier.url',
43     'source',
44     'source.URL',
45     'rights.license',
46     'rights',
47     'date.pd',
48     'format',
49     'type',
50     'date',
51     'audience',
52     'language',
53 )
54
55 IDENTIFIER_PREFIX = 'http://wolnelektury.pl/katalog/lektura/'
56
57
58 def dc_desc_element(book):
59     xml = book.materialize()
60     tree = etree.fromstring(xml)
61     return tree.find(".//" + RDFNS("Description"))
62
63
64 def distinct_dc_values(tag, desc_elements):
65     values = set()
66     for desc in desc_elements:
67         values.update(elem.text for elem in desc.findall(DCNS(tag)))
68     return values
69
70
71 class Command(BaseCommand):
72     args = 'slug'
73
74     def handle(self, slug, **options):
75         children_slugs = [line.strip() for line in sys.stdin]
76         children = Book.objects.filter(dc_slug__in=children_slugs)
77         desc_elements = [dc_desc_element(child) for child in children]
78         title = u'Utwory wybrane'
79         own_attributes = {
80             'title': title,
81             'relation.hasPart': [IDENTIFIER_PREFIX + child_slug for child_slug in children_slugs],
82             'identifier.url': IDENTIFIER_PREFIX + slug,
83             'date': date.today().isoformat(),
84         }
85         dc_tags = []
86         for tag in DC_TAGS:
87             if tag in own_attributes:
88                 values = own_attributes[tag]
89                 if not isinstance(values, list):
90                     values = [values]
91             else:
92                 values = distinct_dc_values(tag, desc_elements)
93             for value in values:
94                 dc_tags.append(DC_TEMPLATE % {'tag': tag, 'value': value})
95         xml = TEMPLATE % {'slug': slug, 'dc': '\n'.join(dc_tags)}
96         Book.create(
97             text=xml,
98             creator=None,
99             slug=slug,
100             title=title,
101             gallery=slug)