Fix error during report generation

[wolnelektury.git] / wolnelektury / translation.py

diff --git a/wolnelektury/translation.py b/wolnelektury/translation.py
index dad3638..95b9dd7 100644 (file)
--- a/wolnelektury/translation.py
+++ b/wolnelektury/translation.py
@@ -4,15 +4,23 @@
 #
 
 from modeltranslation.translator import translator, TranslationOptions
 #
 
 from modeltranslation.translator import translator, TranslationOptions

-from catalogue.models import Tag
+from catalogue.models import Collection, Tag

 from infopages.models import InfoPage
 from infopages.models import InfoPage

+from chunks.models import Chunk

 
 class InfoPageTranslationOptions(TranslationOptions):
     fields = ('title', 'left_column', 'right_column')
 
 class TagTranslationOptions(TranslationOptions):
 
 class InfoPageTranslationOptions(TranslationOptions):
     fields = ('title', 'left_column', 'right_column')
 
 class TagTranslationOptions(TranslationOptions):

-    fields = ('name', 'description')
+    fields = ('name', 'description', 'wiki_link')
+
+class CollectionTranslationOptions(TranslationOptions):
+    fields = ('title', 'description')
+
+class ChunkTranslationOptions(TranslationOptions):
+    fields = ('content',)

 
 translator.register(InfoPage, InfoPageTranslationOptions)
 translator.register(Tag, TagTranslationOptions)
 
 translator.register(InfoPage, InfoPageTranslationOptions)
 translator.register(Tag, TagTranslationOptions)

-
+translator.register(Collection, CollectionTranslationOptions)
+translator.register(Chunk, ChunkTranslationOptions)