[teisenbe] Use the imp module to import modules.

Also add a simple translator module

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(),))