src/librarian/elements/poetry/strofa.py

   1 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
   3 #
   4 from copy import copy
   5 import re
   6 from ..base import WLElement
   7 from .wers import Wers
   8
   9
  10 class Strofa(WLElement):
  11     SHOULD_HAVE_ID = True
  12
  13     TXT_TOP_MARGIN = 2
  14     TXT_BOTTOM_MARGIN = 2
  15     TXT_LEGACY_TOP_MARGIN = 1
  16     TXT_LEGACY_BOTTOM_MARGIN = 0
  17
  18     EPUB_TAG = HTML_TAG = 'div'
  19     EPUB_CLASS = HTML_CLASS = 'stanza'
  20
  21     def epub_build(self, builder):
  22         super().epub_build(builder)
  23         builder.start_element(
  24             'div',
  25             {
  26                 'class': 'stanza-spacer'
  27             }
  28         )
  29         builder.push_text('\u00a0');
  30         builder.end_element()
  31
  32     def get_verses(self):
  33         from librarian.parser import parser
  34
  35         verses = [
  36             parser.makeelement('wers')
  37         ]
  38         if self.text:
  39             # Before any tags. These are text-only verses.
  40             pieces = re.split(r"/\s+", self.text)
  41             for piece in pieces[:-1]:
  42                 verses[-1].text = piece
  43                 verses.append(parser.makeelement('wers'))
  44             verses[-1].text = pieces[-1]
  45
  46         for child in self:
  47             if child.tail:
  48                 pieces = re.split(r"/\s+", child.tail)
  49                 child_copy = copy(child)
  50                 child_copy.tail = pieces[0]
  51                 verses[-1].append(child_copy)
  52
  53                 for piece in pieces[1:]:
  54                     verses.append(parser.makeelement('wers'))
  55                     verses[-1].text = piece
  56
  57             else:
  58                 verses[-1].append(child)
  59
  60         for verse in verses:
  61             verse.stanza = self
  62             if len(verse) == 1 and isinstance(verse[0], Wers):
  63                 assert not (verse.text or '').strip()
  64                 assert not (verse[0].tail or '').strip()
  65                 yield verse[0]
  66             else:
  67                 yield verse
  68
  69     def _build_inner(self, builder, build_method):
  70         for child in self.get_verses():
  71             getattr(child, build_method)(builder)