sopel/modules/translate.py

# coding=utf-8
"""
translate.py - Sopel Translation Module
Copyright 2008, Sean B. Palmer, inamidst.com
Copyright © 2013-2014, Elad Alfassa <elad@fedoraproject.org>
Licensed under the Eiffel Forum License 2.

http://sopel.chat
"""
import json
import random
import sys
import requests

from module import rule, commands, priority, example

mangle_lines = {}
if sys.version_info.major >= 3:
	unicode = str


def translate(text, in_lang='auto', out_lang='en', verify_ssl=True):
	raw = False
	if unicode(out_lang).endswith('-raw'):
		out_lang = out_lang[:-4]
		raw = True

	headers = {
		'User-Agent': 'Mozilla/5.0' +
		'(X11; U; Linux i686)' +
		'Gecko/20071127 Firefox/2.0.0.11'
	}

	query = {
		"client": "gtx",
		"sl": in_lang,
		"tl": out_lang,
		"dt": "t",
		"q": text,
	}
	url = "http://translate.googleapis.com/translate_a/single"
	result = requests.get(url, params=query, timeout=40, headers=headers,
						  verify=verify_ssl).text

	if result == '[,,""]':
		return None, in_lang

	while ',,' in result:
		result = result.replace(',,', ',null,')
		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


@commands('translate', 'tr')
@example('.tr :en :fr my dog', '"mon chien" (en to fr, translate.google.com)')
@example('.tr היי', '"Hey" (iw to en, translate.google.com)')
@example('.tr mon chien', '"my dog" (fr to en, translate.google.com)')
def tr2(bot, trigger):
	"""Translates a phrase, with an optional language hint."""
	command = trigger.group(2)

	if not command:
		return bot.reply('You did not give me anything to translate')

	def langcode(p):
		return p.startswith(':') and (2 < len(p) < 10) and p[1:].isalpha()

	args = ['auto', 'en']

	for i in range(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 bot.reply('Phrase must be under 350 characters.')

	if phrase.strip() == '':
		return bot.reply('You need to specify a string for me to translate!')

	src, dest = args
	if src != dest:
		msg, src = translate(phrase, src, dest,
							 verify_ssl=bot.config.core.verify_ssl)
		if sys.version_info.major < 3 and isinstance(msg, str):
			msg = msg.decode('utf-8')
		if msg:
			#msg = web.decode(msg)  # msg.replace('&#39;', "'")
			msg = '"%s" (%s to %s, translate.google.com)' % (msg, src, dest)
		else:
			msg = 'The %s to %s translation failed, are you sure you specified valid language abbreviations?' % (src, dest)

		bot.reply(msg)
	else:
		bot.reply('Language guessing failed, so try suggesting one!')


def get_random_lang(long_list, short_list):
	random_index = random.randint(0, len(long_list) - 1)
	random_lang = long_list[random_index]
	if random_lang not in short_list:
		short_list.append(random_lang)
	else:
		return get_random_lang(long_list, short_list)
	return short_list


@commands('mangle', 'mangle2')
def mangle(bot, trigger):
	"""Repeatedly translate the input until it makes absolutely no sense."""
	verify_ssl = bot.config.core.verify_ssl
	global mangle_lines
	long_lang_list = ['fr', 'de', 'es', 'it', 'no', 'he', 'la', 'ja', 'cy', 'ar', 'yi', 'zh', 'nl', 'ru', 'fi', 'hi', 'af', 'jw', 'mr', 'ceb', 'cs', 'ga', 'sv', 'eo', 'el', 'ms', 'lv']
	lang_list = []
	for __ 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:
			bot.reply("What do you want me to mangle?")
			return
	else:
		phrase = (trigger.group(2).strip(), '')
	if phrase[0] == '':
		bot.reply("What do you want me to mangle?")
		return
	for lang in lang_list:
		backup = phrase
		try:
			phrase = translate(phrase[0], 'en', lang,
							   verify_ssl=verify_ssl)
		except:
			phrase = False
		if not phrase:
			phrase = backup
			break

		try:
			phrase = translate(phrase[0], lang, 'en', verify_ssl=verify_ssl)
		except:
			phrase = backup
			continue

		if not phrase:
			phrase = backup
			break
	bot.reply(phrase[0])


@rule('(.*)')
@priority('low')
def collect_mangle_lines(bot, trigger):
	global mangle_lines
	mangle_lines[trigger.sender.lower()] = "%s said '%s'" % (trigger.nick, (trigger.group(0).strip()))


if __name__ == "__main__":
	from test_tools import run_example_tests
	run_example_tests(__file__)