src/isbn/utils.py

   1 # -*- coding: utf-8 -*-
   2 from librarian import RDFNS, DCNS
   3
   4 FORMATS = ('PDF', 'HTML', 'TXT', 'EPUB', 'MOBI')
   5
   6 FORMATS_WITH_CHILDREN = ('PDF', 'EPUB', 'MOBI')
   7
   8 PRODUCT_FORMS = {
   9     'HTML': 'EC',
  10     'PDF': 'EB',
  11     'TXT': 'EB',
  12     'EPUB': 'ED',
  13     'MOBI': 'ED',
  14     'SOFT': 'BC',
  15 }
  16
  17 PRODUCT_FORM_DETAILS = {
  18     'HTML': 'E105',
  19     'PDF': 'E107',
  20     'TXT': 'E112',
  21     'EPUB': 'E101',
  22     'MOBI': 'E127',
  23 }
  24
  25 PRODUCT_FORMATS = {
  26     'E105': ('html', 'text/html'),
  27     'E107': ('pdf', 'application/pdf'),
  28     'E112': ('txt', 'text/plain'),
  29     'E101': ('epub', 'application/epub+zip'),
  30     'E127': ('mobi', 'application/x-mobipocket-ebook'),
  31 }
  32
  33 VOLUME_SEPARATORS = (u'. część ', u', część ', u', tom ', u'. der tragödie ')
  34
  35
  36 def is_institution(name):
  37     return name.startswith(u'Zgromadzenie Ogólne')
  38
  39
  40 def get_volume(title):
  41     for volume_separator in VOLUME_SEPARATORS:
  42         if volume_separator in title.lower():
  43             vol_idx = title.lower().index(volume_separator)
  44             stripped = title[:vol_idx]
  45             vol_name = title[vol_idx + 2:]
  46             return stripped, vol_name
  47     return title, ''
  48
  49
  50 def dc_values(desc, tag):
  51     return [e.text.strip() for e in desc.findall('.//' + DCNS(tag))]
  52
  53
  54 def isbn_data(wldoc, file_format=None):
  55     desc = wldoc.edoc.find('.//' + RDFNS('Description'))
  56     title, volume = get_volume(dc_values(desc, 'title')[0])
  57     author = '; '.join(author.strip() for author in dc_values(desc, 'creator'))
  58     data = {
  59         'imprint': '; '.join(dc_values(desc, 'publisher')),
  60         'title': title,
  61         'subtitle': '',
  62         'year': '',
  63         'part_number': volume,
  64         'name': author if not is_institution(author) else '',
  65         'corporate_name': author if is_institution(author) else '',
  66         'edition_type': 'DGO',
  67         'edition_number': '1',
  68         'language': dc_values(desc, 'language')[0],
  69         'dc_slug': wldoc.book_info.url.slug,
  70     }
  71     if file_format:
  72         data['product_form'] = PRODUCT_FORMS[file_format]
  73         data['product_form_detail'] = PRODUCT_FORM_DETAILS[file_format]
  74         return data
  75     else:
  76         has_children = len(dc_values(desc, 'relation.hasPart')) > 0
  77         data['formats'] = FORMATS_WITH_CHILDREN if has_children else FORMATS
  78         return data