From 349a146d666f846567f4abd18bc44c9ed56ad438 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nicolas=20=C3=89vrard?= <nicoe@b2ck.com>
Date: Mon, 16 Sep 2013 12:26:10 +0200
Subject: [PATCH] Allow customization of translation in po files

review968002
---
 CHANGELOG                            |  1 +
 trytond/ir/translation.py            | 42 ++++++++++++++++++++++++++++
 trytond/ir/view/translation_form.xml |  2 ++
 3 files changed, 45 insertions(+)

diff --git a/CHANGELOG b/CHANGELOG
index 941734b040..5071eca995 100644
--- a/CHANGELOG
+++ b/CHANGELOG
@@ -1,3 +1,4 @@
+* Allow customization of translation in po files
 * Use python-sql
 * Add convert_domain method on Fields
 * Add sql_format and sql_type methods on Fields
diff --git a/trytond/ir/translation.py b/trytond/ir/translation.py
index fb88e10a3d..dbd00c7e0b 100644
--- a/trytond/ir/translation.py
+++ b/trytond/ir/translation.py
@@ -75,6 +75,7 @@ class Translation(ModelSQL, ModelView):
     fuzzy = fields.Boolean('Fuzzy')
     model = fields.Function(fields.Char('Model'), 'get_model',
             searcher='search_model')
+    overriding_module = fields.Char('Overriding Module', readonly=True)
     _translation_cache = Cache('ir.translation', size_limit=10240,
         context=False)
     _get_language_cache = Cache('ir.translation')
@@ -87,6 +88,11 @@ class Translation(ModelSQL, ModelView):
                 'UNIQUE (name, res_id, lang, type, src_md5, module)',
                 'Translation must be unique'),
         ]
+        cls._error_messages.update({
+                'translation_overridden': \
+                    "You can not export translation %(name)s because it is an "
+                    "overridden translation by module %(overriding_module)s",
+                })
 
     @classmethod
     def __register__(cls, module_name):
@@ -770,6 +776,31 @@ class Translation(ModelSQL, ModelView):
             key2ids.setdefault(key, []).append(translation.id)
             id2translation[translation.id] = translation
 
+        def override_translation(ressource_id, new_translation, fuzzy):
+            res_id_module, res_id = ressource_id.split('.')
+            if res_id:
+                model_data, = ModelData.search([
+                        ('module', '=', res_id_module),
+                        ('fs_id', '=', res_id),
+                        ])
+                res_id = model_data.db_id
+            else:
+                res_id = None
+            with contextlib.nested(Transaction().set_user(0),
+                    Transaction().set_context(module=res_id_module)):
+                translation, = cls.search([
+                        ('name', '=', name),
+                        ('res_id', '=', res_id),
+                        ('lang', '=', lang),
+                        ('type', '=', ttype),
+                        ('module', '=', res_id_module),
+                        ])
+                if translation.value != new_translation:
+                    translation.value = new_translation
+                    translation.overriding_module = module
+                    translation.fuzzy = fuzzy
+                    translation.save()
+
         for entry in pofile:
             ttype, name, res_id = entry.msgctxt.split(':')
             src = entry.msgid
@@ -777,6 +808,10 @@ class Translation(ModelSQL, ModelView):
             fuzzy = 'fuzzy' in entry.flags
             noupdate = False
 
+            if '.' in res_id:
+                override_translation(res_id, value, fuzzy)
+                continue
+
             model = name.split(',')[0]
             if model in fs_id2model_data and res_id in fs_id2model_data[model]:
                 model_data = fs_id2model_data[model][res_id]
@@ -823,6 +858,7 @@ class Translation(ModelSQL, ModelView):
                         cls.write(translations2, {
                             'value': value,
                             'fuzzy': fuzzy,
+                            'overriding_module': None,
                             })
                     translations |= set(cls.browse(ids))
 
@@ -866,6 +902,12 @@ class Translation(ModelSQL, ModelView):
                 ('module', '=', module),
                 ], order=[])
         for translation in translations:
+            if (translation.overriding_module
+                    and translation.overriding_module != module):
+                cls.raise_user_error('translation_overridden', {
+                        'name': translation.name,
+                        'name': translation.overriding_module,
+                        })
             flags = [] if not translation.fuzzy else ['fuzzy']
             trans_ctxt = '%(type)s:%(name)s:' % {
                 'type': translation.type,
diff --git a/trytond/ir/view/translation_form.xml b/trytond/ir/view/translation_form.xml
index c891691362..66039bbbf3 100644
--- a/trytond/ir/view/translation_form.xml
+++ b/trytond/ir/view/translation_form.xml
@@ -14,6 +14,8 @@ this repository contains the full copyright notices and license terms. -->
     <field name="lang"/>
     <label name="module"/>
     <field name="module"/>
+    <label name="overriden_by"/>
+    <field name="overriden_by"/>
     <group col="2" colspan="6" yexpand="1" yfill="1" id="src_value">
         <separator name="src"/>
         <separator name="value"/>