fixes to vierwer/folding catalogue in menu

[wolnelektury.git] / wolnelektury / translation.py

diff --git a/wolnelektury/translation.py b/wolnelektury/translation.py
index 227896e..c72f654 100644 (file)
--- a/wolnelektury/translation.py
+++ b/wolnelektury/translation.py
@@ -4,16 +4,19 @@
 #
 
 from modeltranslation.translator import translator, TranslationOptions
+from catalogue.models import Collection, Tag
 from infopages.models import InfoPage
-from catalogue.models import Book
 
 class InfoPageTranslationOptions(TranslationOptions):
-    fields = ('page_title', 'title', 'left_column', 'right_column')
+    fields = ('title', 'left_column', 'right_column')
 
-translator.register(InfoPage, InfoPageTranslationOptions)
+class TagTranslationOptions(TranslationOptions):
+    fields = ('name', 'description', 'wiki_link')
 
-class BookTranslationOptions(TranslationOptions):
-    fields = ('_short_html', )
+class CollectionTranslationOptions(TranslationOptions):
+    fields = ('title', 'description')
 
-translator.register(Book, BookTranslationOptions)
+translator.register(InfoPage, InfoPageTranslationOptions)
+translator.register(Tag, TagTranslationOptions)
+translator.register(Collection, CollectionTranslationOptions)