#!/usr/bin/env python # coding=utf-8 """ translate.py - Willie Translation Module Copyright 2008, Sean B. Palmer, inamidst.com Licensed under the Eiffel Forum License 2. http://willie.dftba.net """ import re import willie.web as web from time import sleep import urllib2, json import random import os mangle_lines = {} def setup(willie): random.seed() def translate(text, input='auto', output='en'): raw = False if output.endswith('-raw'): output = output[:-4] raw = True opener = urllib2.build_opener() opener.addheaders = [( 'User-Agent', 'Mozilla/5.0' + '(X11; U; Linux i686)' + 'Gecko/20071127 Firefox/2.0.0.11' )] input, output = urllib2.quote(input), urllib2.quote(output) try: if text is not text.encode("utf-8"): text = text.encode("utf-8") except: pass text = urllib2.quote(text) result = opener.open('http://translate.google.com/translate_a/t?' + ('client=t&hl=en&sl=%s&tl=%s&multires=1' % (input, output)) + ('&otf=1&ssel=0&tsel=0&uptl=en&sc=1&text=%s' % text)).read() while ',,' in result: result = result.replace(',,', ',null,') data = json.loads(result) if raw: return str(data), 'en-raw' try: language = data[2] # -2][0][0] except: language = '?' return ''.join(x[0] for x in data[0]), language def tr(willie, trigger): """Translates a phrase, with an optional language hint.""" input, output, phrase = trigger.groups() phrase = phrase.encode('utf-8') if (len(phrase) > 350) and (not trigger.admin): return willie.reply('Phrase must be under 350 characters.') input = input or 'auto' input = input.encode('utf-8') output = (output or 'en').encode('utf-8') if input != output: msg, input = translate(phrase, input, output) if isinstance(msg, str): msg = msg.decode('utf-8') if msg: msg = web.decode(msg) # msg.replace(''', "'") msg = '"%s" (%s to %s, translate.google.com)' % (msg, input, output) else: msg = 'The %s to %s translation failed, sorry!' % (input, output) willie.reply(msg) else: willie.reply('Language guessing failed, so try suggesting one!') tr.rule = ('$nick', ur'(?:([a-z]{2}) +)?(?:([a-z]{2}|en-raw) +)?["“](.+?)["”]\? *$') tr.example = '$nickname: "mon chien"? or $nickname: fr "mon chien"?' tr.priority = 'low' def tr2(willie, trigger): """Translates a phrase, with an optional language hint.""" command = trigger.group(2).encode('utf-8') def langcode(p): return p.startswith(':') and (2 < len(p) < 10) and p[1:].isalpha() args = ['auto', 'en'] for i in xrange(2): if not ' ' in command: break prefix, cmd = command.split(' ', 1) if langcode(prefix): args[i] = prefix[1:] command = cmd phrase = command if (len(phrase) > 350) and (not trigger.admin): return willie.reply('Phrase must be under 350 characters.') src, dest = args if src != dest: msg, src = translate(phrase, src, dest) if isinstance(msg, str): msg = msg.decode('utf-8') if msg: msg = web.decode(msg) # msg.replace(''', "'") msg = '"%s" (%s to %s, translate.google.com)' % (msg, src, dest) else: msg = 'The %s to %s translation failed, sorry!' % (src, dest) willie.reply(msg) else: willie.reply('Language guessing failed, so try suggesting one!') tr2.commands = ['tr'] tr2.priority = 'low' def mangle(willie, trigger): """Repeatedly translate the input until it makes no sense.""" global mangle_lines if trigger.group(2) is None: try: phrase = (mangle_lines[trigger.sender.lower()], '') except: willie.reply("What do you want me to mangle?") return else: phrase = (trigger.group(2).encode('utf-8').strip(), '') if phrase[0] == '': willie.reply("What do you want me to mangle?") return for lang in ['fr', 'de', 'es', 'it', 'no', 'he', 'la', 'ja' ]: backup = phrase phrase = translate(phrase[0], 'en', lang) if not phrase: phrase = backup break backup = phrase phrase = translate(phrase[0], lang, 'en') if not phrase: phrase = backup break willie.reply(phrase[0]) mangle.commands = ['mangle'] def get_random_lang(long_list, short_list): random_index = random.randint(0, len(long_list)-1) random_lang = long_list[random_index] if not random_lang in short_list: short_list.append(random_lang) else: return get_random_lang(long_list, short_list) return short_list def more_mangle(willie, trigger): ''' Research version of mangle ''' global mangle_lines long_lang_list = ['fr', 'de', 'es', 'it', 'no', 'he', 'la', 'ja', 'cy', 'ar', 'yi', 'zh', 'nl', 'ru', 'fi', 'hi', 'af'] lang_list = [] for index in range(0, 8): lang_list = get_random_lang(long_lang_list, lang_list) random.shuffle(lang_list) if trigger.group(2) is None: try: phrase = (mangle_lines[trigger.sender.lower()], '') except: willie.reply("What do you want me to mangle?") return else: phrase = (trigger.group(2).encode('utf-8').strip(), '') if phrase[0] == '': willie.reply("What do you want me to mangle?") return research_logfile = open(os.path.join(willie.config.logdir, 'mangle.log'), 'a') research_logfile.write('Phrase: %s\n' % str(phrase)) research_logfile.write('Lang_list: %s\n' % lang_list) for lang in lang_list: backup = phrase phrase = translate(phrase[0], 'en', lang) if not phrase: phrase = backup break backup = phrase phrase = translate(phrase[0], lang, 'en') research_logfile.write('-> %s\n' % str(phrase)) if not phrase: phrase = backup break research_logfile.write('->[FINAL] %s\n' % str(phrase)) research_logfile.write('----------------------------\n\n\n') research_logfile.close() willie.reply(phrase[0]) more_mangle.commands = ['mangle2'] def collect_mangle_lines(willie, trigger): global mangle_lines mangle_lines[trigger.sender.lower()] = "%s said '%s'" % (trigger.nick, trigger.group(0).encode('utf-8').strip()) collect_mangle_lines.rule = ('(.*)') collect_mangle_lines.priority = 'low' if __name__ == '__main__': print __doc__.strip()