diff options
Diffstat (limited to 'plugins')
-rw-r--r-- | plugins/screen_detach.py | 1 | ||||
-rw-r--r-- | plugins/test.py | 4 | ||||
-rw-r--r-- | plugins/translate.py | 33 |
3 files changed, 38 insertions, 0 deletions
diff --git a/plugins/screen_detach.py b/plugins/screen_detach.py index 6ebc77f2..6ee96896 100644 --- a/plugins/screen_detach.py +++ b/plugins/screen_detach.py @@ -1,3 +1,4 @@ +from plugin import BasePlugin import os import stat import pyinotify diff --git a/plugins/test.py b/plugins/test.py index fc7aedc6..13ba1e9c 100644 --- a/plugins/test.py +++ b/plugins/test.py @@ -1,5 +1,9 @@ +from plugin import BasePlugin + class Plugin(BasePlugin): def init(self): + self.add_command('plugintest', self.command_plugintest, 'Test command') + self.add_event_handler('message', self.on_message) self.core.information("Plugin loaded") def cleanup(self): diff --git a/plugins/translate.py b/plugins/translate.py new file mode 100644 index 00000000..625d78e7 --- /dev/null +++ b/plugins/translate.py @@ -0,0 +1,33 @@ +from plugin import BasePlugin +import urllib.request +from urllib.parse import urlencode +import xhtml +import json + +TARGET_LANG = 'en' + +def translate(s, target=TARGET_LANG, source=''): + f = urllib.request.urlopen('http://ajax.googleapis.com/ajax/services/language/translate', urlencode({ 'v': '1.0', 'q': s, 'langpair': '%s|%s' % (source, target) })) + response = json.loads(str(f.read(), 'utf-8'))['responseData'] + return (response['translatedText'], response['detectedSourceLanguage']) + +class Plugin(BasePlugin): + def init(self): + self.add_event_handler('groupchat_message', self.on_groupchat_message) + + def on_groupchat_message(self, message): + try: + room_from = message.getMucroom() + if message['type'] == 'error': + return + + if room_from == 'poezio@kikoo.louiz.org': + nick_from = message['mucnick'] + body = xhtml.get_body_from_message_stanza(message) + room = self.core.get_room_by_name(room_from) + text, lang = translate(body) + if lang != TARGET_LANG: + room.add_message(text, nickname=nick_from) + except Exception as e: + import traceback + self.core.information("Exception in translator! %s" % (traceback.format_exc(),)) |