Fix for comments.

[librarian.git] / src / librarian / elements / __init__.py

diff --git a/src/librarian/elements/__init__.py b/src/librarian/elements/__init__.py
index 512b4ef..b08c3e1 100644 (file)
--- a/src/librarian/elements/__init__.py
+++ b/src/librarian/elements/__init__.py
@@ -1,16 +1,23 @@
+# This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
+#
 from lxml import etree
 from . import (blocks, comments, drama, figures, footnotes, front, headers,
-               masters, paragraphs, poetry, root, separators, styles, themes)
+               masters, paragraphs, poetry, ref, root, separators, styles, themes,
+               tools, base)
 
 
 WL_ELEMENTS = {
+    'snippet': base.Snippet,
     'meta': etree.ElementBase,
     'coverClass': etree.ElementBase,
     "developmentStage": etree.ElementBase,
     "coverBarColor": etree.ElementBase,
     "coverBoxPosition": etree.ElementBase, 
     "coverLogoUrl": etree.ElementBase,
-   
+    "contentWarning": etree.ElementBase,
+    "endnotes": etree.ElementBase,
+
     "utwor": root.Utwor,
     "dramat_wierszowany_l": masters.Master,
     "dramat_wierszowany_lp": masters.Master,
@@ -20,6 +27,8 @@ WL_ELEMENTS = {
     "opowiadanie": masters.Master,
     "powiesc": masters.Master,
 
+    "blok": tools.WLElement,
+    
     "autor_utworu": front.AutorUtworu,
     "dzielo_nadrzedne": front.DzieloNadrzedne,
     "nazwa_utworu": front.NazwaUtworu,
@@ -33,6 +42,7 @@ WL_ELEMENTS = {
     "dlugi_cytat": blocks.DlugiCytat,
     "poezja_cyt": blocks.PoezjaCyt,
     "dlugi_cyt": blocks.DlugiCytat,  ### ???
+    "ramka": blocks.Ramka,
     
     "slowo_obce": styles.SlowoObce,
     "tytul_dziela": styles.TytulDziela,
@@ -52,22 +62,25 @@ WL_ELEMENTS = {
     "kwestia": drama.Kwestia,
     "didask_tekst": drama.DidaskTekst,
 
-    "dedykacja": paragraphs.Akap,
-    "miejsce_czas": paragraphs.Akap,
+    "dedykacja": blocks.Dedykacja,
+    "miejsce_czas": drama.MiejsceCzas,
 
     "uwaga": comments.Uwaga,
 
     "wers": poetry.Wers,
     "wers_wciety": poetry.WersWciety,
     "wers_cd": poetry.WersCd,
-    "wers_akap": poetry.Wers,
+    "wers_akap": poetry.WersAkap,
     "zastepnik_wersu": poetry.ZastepnikWersu,
-    "wers_do_prawej": poetry.Wers,
+    "wers_do_prawej": poetry.WersDoPrawej,
+    "wers_srodek": poetry.WersSrodek,
     
-    "pa": footnotes.Footnote,
-    "pe": footnotes.Footnote,
-    "pr": footnotes.Footnote,
-    "pt": footnotes.Footnote,
+    "pa": footnotes.PA,
+    "pe": footnotes.PE,
+    "pr": footnotes.PR,
+    "pt": footnotes.PT,
+
+    "ref": ref.Ref,
 
     "begin": themes.Begin,
     "end": themes.End,
@@ -75,16 +88,21 @@ WL_ELEMENTS = {
 
     "nota": blocks.Nota,
 
-    "nota_red": comments.Abstrakt,
-    "extra": comments.Abstrakt,
+    "nota_red": comments.NotaRed,
+    "extra": comments.Uwaga,
     "abstrakt": comments.Abstrakt,
 
     "naglowek_czesc": headers.NaglowekCzesc,
     "naglowek_akt": headers.NaglowekCzesc,
-    "naglowek_scena": headers.NaglowekRozdzial,
+    "naglowek_scena": headers.NaglowekScena,
     "naglowek_rozdzial": headers.NaglowekRozdzial,
     "naglowek_podrozdzial": headers.NaglowekPodrozdzial,
     "srodtytul": headers.NaglowekCzesc,
+    "podtytul_czesc": headers.PodtytulCzesc,
+    "podtytul_akt": headers.PodtytulCzesc,
+    "podtytul_scena": headers.PodtytulRozdzial,
+    "podtytul_rozdzial": headers.PodtytulRozdzial,
+    "podtytul_podrozdzial": headers.PodtytulPodrozdzial,
 
     "naglowek_listy": drama.NaglowekListy,
 
@@ -92,24 +110,33 @@ WL_ELEMENTS = {
     "sekcja_swiatlo": separators.SekcjaSwiatlo,
     "separator_linia": separators.SeparatorLinia,
 
-    "wieksze_odstepy": styles.Wyroznienie,
-    "mat": styles.Wyroznienie,
-    "www": styles.Wyroznienie,
-    "indeks_dolny": styles.Wyroznienie,
+    "wieksze_odstepy": styles.WiekszeOdstepy,
+    "mat": styles.Mat,
+    "www": styles.WWW,
+    "indeks_dolny": styles.IndeksDolny,
 
-    "tabela": paragraphs.Akap,
-    "tabelka": paragraphs.Akap,
-    "wiersz": paragraphs.Akap,
-    "kol": paragraphs.Akap,
+    "tabela": figures.Tabela,
+    "tabelka": figures.Tabela,
+    "wiersz": figures.Wiersz,
+    "kol": figures.Kol,
 
+    "animacja": figures.Animacja,
     "ilustr": figures.Ilustr,
 
-#    sklodowska-badanie-cial-radioaktywnych.xml
-    "mrow": paragraphs.Akap,
-    "mi": paragraphs.Akap,
-    "mo": paragraphs.Akap,
-    "msup": paragraphs.Akap,
-    "mn": paragraphs.Akap,
-    "mfrac": paragraphs.Akap,
-    "mfenced": paragraphs.Akap,
+    "numeracja": tools.Numeracja,
+    "rownolegle": tools.Rownolegle,
+    "tab": tools.Tab,
+
+    # Section
+    "wywiad_pyt": blocks.WywiadPyt,
+    "wywiad_odp": blocks.WywiadOdp,
+
+    # Inline MathML, should really be namespaced.
+    "mrow": etree.ElementBase,
+    "mi": etree.ElementBase,
+    "mo": etree.ElementBase,
+    "msup": etree.ElementBase,
+    "mn": etree.ElementBase,
+    "mfrac": etree.ElementBase,
+    "mfenced": etree.ElementBase,
 }