Move more exporter logic to the exporter class

Signed-off-by: Michal Čihař <michal@cihar.com>

weblate/trans/exporters.py

@@ -52,10 +52,16 @@ class BaseExporter(object):
     name = ''
     has_lang = False
 
-    def __init__(self, project, language, url):
-        self.project = project
-        self.language = language
-        self.url = url
+    def __init__(self, project=None, language=None, url=None,
+                 translation=None):
+        if translation is not None:
+            self.project = translation.subproject.project
+            self.langauge = obj.language
+            self.url = obj.get_absolute_url()
+        else:
+            self.project = project
+            self.language = language
+            self.url = url
         self.storage = self.get_storage()
         self.storage.setsourcelanguage(project.source_language.code)
         self.storage.settargetlanguage(language.code)
@@ -72,6 +78,10 @@ class BaseExporter(object):
             unit.target = word.target
         self.storage.addunit(unit)
 
+    def add_units(self, translation):
+        for unit intranslation .unit_set.iterator():
+            self.add_unit(unit)
+
     def add_unit(self, unit):
         if unit.is_plural():
             output = self.storage.UnitClass(

weblate/trans/views/files.py

@@ -41,16 +41,11 @@ def download_translation_format(request, project, subproject, lang, fmt):
     obj = get_translation(request, project, subproject, lang)
 
     try:
-        exporter = get_exporter(fmt)(
-            obj.subproject.project,
-            obj.language,
-            obj.get_absolute_url()
-        )
+        exporter = get_exporter(fmt)(translation=obj)
     except KeyError:
         raise Http404('File format not supported')
 
-    for unit in obj.unit_set.iterator():
-        exporter.add_unit(unit)
+    exporter.add_units(obj)
 
     # Save to response
     return exporter.get_response(