diff --git a/plugins/google_translation/plugin.py b/plugins/google_translation/plugin.py
index 70de53553436af44b999607af7626e298eb361e4..d00d1b993ae4773911fd7d01ad15724ea1d88ded 100644
--- a/plugins/google_translation/plugin.py
+++ b/plugins/google_translation/plugin.py
@@ -30,9 +30,13 @@ import re
import urllib2
import HTMLParser
import new
+import gtk
from pprint import pformat
from sys import getfilesystemencoding
+import chat_control
+import groupchat_control
+
from common import helpers
from common import gajim
@@ -41,6 +45,67 @@ from plugins.helpers import log_calls, log
from common import ged
from common import nec
+languages = {
+ _('Afrikaans'): 'af',
+ _('Albanian'): 'sq',
+ _('Armenian'): 'hy',
+ _('Azerbaijani'): 'az',
+ _('Arabic'): 'ar',
+ _('Basque'): 'eu',
+ _('Belarusian'): 'be',
+ _('Bulgarian'): 'bg',
+ _('Catalan'): 'ca',
+ _('Chinese (Simplified)'): 'zh-cn',
+ _('Chinese (Traditional)'): 'zh-tw',
+ _('Croatian'): 'hr',
+ _('Czech'): 'cs',
+ _('Danish'): 'da',
+ _('Dutch'): 'nl',
+ _('English'): 'en',
+ _('Estonian'): 'et',
+ _('Filipino'): 'tl',
+ _('Finnish'): 'fi',
+ _('French'): 'fr',
+ _('Galician'): 'gl',
+ _('Georgian'): 'ka',
+ _('German'): 'de',
+ _('Greek'): 'el',
+ _('Haitian Creole'): 'ht',
+ _('Hebrew'): 'iw',
+ _('Hindi'): 'hi',
+ _('Hungarian'): 'hu',
+ _('Icelandic'): 'is',
+ _('Indonesian'): 'id',
+ _('Italian'): 'it',
+ _('Irish'): 'da',
+ _('Japanese'): 'ja',
+ _('Korean'): 'ko',
+ _('Latvian'): 'lv',
+ _('Lithuanian'): 'lt',
+ _('Macedonian'): 'mk',
+ _('Malay'): 'ml',
+ _('Maltese'): 'mt',
+ _('Norwegian'): 'no',
+ _('Persian'): 'fa',
+ _('Polish'): 'pl',
+ _('Portuguese'): 'pt-BR',
+ _('Romanian'): 'ro',
+ _('Russian'): 'ru',
+ _('Serbian'): 'sr',
+ _('Slovak'): 'sk',
+ _('Slovenian'): 'sl',
+ _('Spanish'): 'es',
+ _('Swahili'): 'sw',
+ _('Swedish'): 'sv',
+ _('Thai'): 'th',
+ _('Turkish'): 'tr',
+ _('Ukrainian'): 'uk',
+ _('Urdu'): 'ur',
+ _('Vietnamese'): 'vi',
+ _('Welsh'): 'cy',
+ _('Yiddish'): 'yi',
+}
+
class GoogleTranslationPlugin(GajimPlugin):
@log_calls('GoogleTranslationPlugin')
@@ -50,10 +115,7 @@ class GoogleTranslationPlugin(GajimPlugin):
self.config_dialog = None
self.config_default_values = {
- 'from_lang' :
- (u'en', u'Language of text to be translated'),
- 'to_lang' :
- (u'fr', u'Language to which translation will be made'),
+ 'per_jid_config': ({}, ''),
'user_agent' :
(u'Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.8.1.12) '
'Gecko/20080213 Firefox/2.0.0.11',
@@ -63,6 +125,12 @@ class GoogleTranslationPlugin(GajimPlugin):
self.events_handlers = {'decrypted-message-received': (ged.PREGUI,
self._nec_decrypted_message_received)}
+ self.gui_extension_points = {
+ 'chat_control_base' : (self.connect_with_control,
+ self.disconnect_from_control),
+ }
+
+ self.controls = []
self.translated_text_re = re.compile(
r'google.language.callbacks.id100\(\'22\', '
'{"translatedText":"(?P<text>[^"]*)"}, 200, null, 200\)')
@@ -103,11 +171,16 @@ class GoogleTranslationPlugin(GajimPlugin):
def _nec_decrypted_message_received(self, obj):
if not obj.msgtxt:
return
- from_lang = self.config['from_lang']
- to_lang = self.config['to_lang']
+ if obj.jid not in self.config['per_jid_config']:
+ return
+ if not self.config['per_jid_config'][obj.jid]['enabled']:
+ return
+ from_lang = self.config['per_jid_config'][obj.jid]['from']
+ to_lang = self.config['per_jid_config'][obj.jid]['to']
translated_text = self.translate_text(obj.msgtxt, from_lang, to_lang)
if translated_text:
- obj.msgtxt = translated_text
+ obj.msgtxt = translated_text + '\n/' + _('Original text:') + '/ ' +\
+ obj.msgtxt
@log_calls('GoogleTranslationPlugin')
def activate(self):
@@ -116,3 +189,107 @@ class GoogleTranslationPlugin(GajimPlugin):
@log_calls('GoogleTranslationPlugin')
def deactivate(self):
pass
+
+ @log_calls('GoogleTranslationPlugin')
+ def connect_with_control(self, control):
+ base = Base(self, control)
+ self.controls.append(base)
+
+ @log_calls('GoogleTranslationPlugin')
+ def disconnect_from_control(self, chat_control):
+ for base in self.controls:
+ base.disconnect_from_control()
+ self.controls = []
+
+class Base(object):
+ def __init__(self, plugin, control):
+ self.plugin = plugin
+ self.control = control
+ self.contact = control.contact
+ self.account = control.account
+ self.jid = self.contact.jid
+ if self.jid in self.plugin.config['per_jid_config']:
+ self.config = self.plugin.config['per_jid_config'][self.jid]
+ else:
+ self.config = {'from': '', 'to': 'en', 'enabled': False}
+ self.create_buttons()
+
+ def create_buttons(self):
+ if isinstance(self.control, chat_control.ChatControl):
+ vbox = self.control.xml.get_object('vbox106')
+ elif isinstance(self.control, groupchat_control.GroupchatControl):
+ vbox = self.control.xml.get_object('gc_textviews_vbox')
+ else:
+ return
+
+ self.expander = gtk.Expander(_('Google translation'))
+ hbox = gtk.HBox(spacing=6)
+ self.expander.add(hbox)
+ label = gtk.Label(_('Translate from'))
+ hbox.pack_start(label, False, False)
+ liststore1 = gtk.ListStore(str, str)
+ liststore2 = gtk.ListStore(str, str)
+ cb1 = gtk.ComboBox(liststore1)
+ cb2 = gtk.ComboBox(liststore2)
+ cell = gtk.CellRendererText()
+ cb1.pack_start(cell, True)
+ cb1.add_attribute(cell, 'text', 0)
+ cell = gtk.CellRendererText()
+ cb2.pack_start(cell, True)
+ cb2.add_attribute(cell, 'text', 0)
+ #Language to translate from
+ liststore1.append([_('Auto'), ''])
+ if self.config['from'] == '':
+ cb1.set_active(0)
+ if self.config['from'] == '':
+ cb1.set_active(0)
+ i = 0
+ ls = languages.items()
+ ls.sort()
+ for l in ls:
+ liststore1.append(l)
+ if l[1] == self.config['from']:
+ cb1.set_active(i+1)
+ liststore2.append(l)
+ if l[1] == self.config['to']:
+ cb2.set_active(i)
+ i += 1
+
+ hbox.pack_start(cb1, False, False)
+ label = gtk.Label(_('to'))
+ hbox.pack_start(label, False, False)
+ hbox.pack_start(cb2, False, False)
+
+ cb = gtk.CheckButton(_('enable'))
+ if self.config['enabled']:
+ cb.set_active(True)
+ hbox.pack_start(cb, False, False)
+ vbox.pack_start(self.expander, False, False)
+ vbox.reorder_child(self.expander, 1)
+
+ cb1.connect('changed', self.on_cb_changed, 'from')
+ cb2.connect('changed', self.on_cb_changed, 'to')
+ cb.connect('toggled', self.on_cb_toggled)
+ self.expander.show_all()
+
+ def on_cb_changed(self, widget, option):
+ model = widget.get_model()
+ it = widget.get_active_iter()
+ self.config[option] = model[it][1]
+ self.plugin.config['per_jid_config'][self.jid] = self.config
+ self.plugin.config.save()
+
+ def on_cb_toggled(self, widget):
+ self.config['enabled'] = widget.get_active()
+ self.plugin.config['per_jid_config'][self.jid] = self.config
+ self.plugin.config.save()
+
+ def disconnect_from_control(self):
+ if isinstance(self.control, chat_control.ChatControl):
+ vbox = self.control.xml.get_object('vbox106')
+ elif isinstance(self.control, groupchat_control.GroupchatControl):
+ vbox = self.control.xml.get_object('gc_textviews_vbox')
+ else:
+ return
+
+ vbox.remove(self.expander)