first

.gitignore

@@ -0,0 +1,3 @@
+*.pyc
+*.pyo
+*.mo

README.md

@@ -0,0 +1,7 @@
+# Дополнение Яндекс.Переводчик для NVDA
+
+После установки зайдите в NVDA меню, Параметры, Жесты ввода, назначте удобные вам сочетания клавиш для YandexTranslate.
+
+Переводит выделенный фрагмент текста или текст из буфера обмена.
+
+Настроить можно в NVDA меню, Сервис, Настройки Yandex Translate.

globalPlugins/yandexTranslate/__init__.py

@@ -0,0 +1,366 @@
+import os
+import json
+import time
+import wx
+import threading
+import http.client
+import sys
+import pickle
+from urllib.parse import urlencode
+
+import globalPluginHandler
+import gui
+import globalVars
+import speech
+import api
+import textInfos
+import tones
+import ui
+import addonHandler
+import languageHandler
+from logHandler import log
+
+from .languages import languages
+
+from .yandexfreetranslate import YandexFreeTranslate
+
+addonHandler.initTranslation()
+_cache = {}
+FILE_CONFIG_PATH = os.path.join(globalVars.appArgs.configPath, "YandexTranslateSettings.pickle")
+
+conf = {
+	"key": "",
+	"sourceLang": "auto",
+	"primaryTargetLang": "en",
+	"secondaryTargetLang": "en",
+	"switchLang": "en",
+	"copyToClipBoard": True,
+	"signals": False,
+}
+
+ERRORS = {
+	401: _("Invalid API key"),
+	402: _("This API key has been blocked"),
+	403: _("You have reached the daily limit for requests"),
+	404: _("You have reached the daily limit for the volume of translated text"),
+	413: _("The text size exceeds the maximum"),
+	422: _("The text could not be translated"),
+	501: _("The specified translation direction is not supported"),
+}
+
+# Decorator to lock the scripts on the secure desktop
+def secureScript(script):
+	def wrapper(self, gesture):
+		if globalVars.appArgs.secure:
+			ui.message(_("Action cannot be performed because NVDA running on secure desktop"))
+		else:
+			script(self, gesture)
+	return wrapper
+
+class YandexTranslateSettingsDialog(gui.SettingsDialog):
+	title = _("Yandex Translate Settings")
+
+	def makeSettings(self, sizer):
+		self.langList = [", ".join((lang, code)) for code, lang in languages.items()]
+		self.langList.sort()
+		settingsSizerHelper = gui.guiHelper.BoxSizerHelper(self, sizer=sizer)
+
+		self.sourceLang = settingsSizerHelper.addLabeledControl(_("Source language:"), wx.Choice, choices=[_("Detect language automatically")+", auto"]+self.langList)
+		if conf["sourceLang"] == "auto":
+			self.sourceLang.SetSelection(0)
+		else:
+			self.sourceLang.SetStringSelection(", ".join((languages[conf["sourceLang"]], conf["sourceLang"])))
+
+		self.primaryTargetLang = settingsSizerHelper.addLabeledControl(_("Primary target language:"), wx.Choice, choices=self.langList)
+		self.primaryTargetLang.SetStringSelection(", ".join((languages[conf["primaryTargetLang"]], conf["primaryTargetLang"])))
+
+		self.secondaryTargetLang = settingsSizerHelper.addLabeledControl(_("Secondary target language:"), wx.Choice, choices=self.langList)
+		self.secondaryTargetLang.SetStringSelection(", ".join((languages[conf["secondaryTargetLang"]], conf["secondaryTargetLang"])))
+
+		self.switchLang = settingsSizerHelper.addLabeledControl(_("Language translation, if the language of the text coincides with the target:"), wx.Choice, choices=self.langList)
+		self.switchLang.SetStringSelection(", ".join((languages[conf["switchLang"]], conf["switchLang"])))
+
+		self.copyToClipBoard = wx.CheckBox(self, label=_("Copy translation to clipboard"))
+		self.copyToClipBoard.SetValue(conf["copyToClipBoard"])
+		settingsSizerHelper.addItem(self.copyToClipBoard)
+
+		self.signals = wx.CheckBox(self, label=_("Play tones when translation waiting"))
+		self.signals.SetValue(conf["signals"])
+		settingsSizerHelper.addItem(self.signals)
+
+		# self.key = settingsSizerHelper.addLabeledControl(_("API key:"), wx.TextCtrl, value=conf["key"])
+
+		self.generate_new_key = wx.CheckBox(self, label=_("Generate new API key"))
+		settingsSizerHelper.addItem(self.generate_new_key)
+
+	def postInit(self):
+		self.sourceLang.SetFocus()
+
+	def onOk(self, event):
+		conf["sourceLang"] = self.sourceLang.GetStringSelection().split()[-1]
+		conf["primaryTargetLang"] = self.primaryTargetLang.GetStringSelection().split()[-1]
+		conf["secondaryTargetLang"] = self.secondaryTargetLang.GetStringSelection().split()[-1]
+		conf["switchLang"] = self.switchLang.GetStringSelection().split()[-1]
+		conf["copyToClipBoard"] = self.copyToClipBoard.Value
+		conf["signals"] = self.signals.Value
+		try:
+			if self.generate_new_key.Value or conf["key"] == "":
+				yt = YandexFreeTranslate()
+				conf["key"] = yt.regenerate_key()
+		except Exception as identifier:
+			text = _("Failed to get a new key. Check your internet connection, wait a bit or go to Yandex, enter the captcha and try again.")
+			ui.message(text)
+			log.error(sys.exc_info()[1])
+			gui.messageBox(text, _("Error saving settings"), style=wx.OK | wx.ICON_ERROR)
+			import webbrowser
+			webbrowser.open_new("https://translate.yandex.ru/")
+
+
+		try:
+			with open(FILE_CONFIG_PATH, "wb") as fileConfig:
+				pickle.dump(conf, fileConfig, pickle.HIGHEST_PROTOCOL)
+		except (IOError, OSError) as e:
+			gui.messageBox(e.strerror, _("Error saving settings"), style=wx.OK | wx.ICON_ERROR)
+
+		super(YandexTranslateSettingsDialog, self).onOk(event)
+
+class Beeper(threading.Thread):
+
+	def __init__(self):
+		super(Beeper, self).__init__()
+		self.daemon = True
+		self._mutex = threading.Lock()
+		self.start()
+
+	def run(self):
+		while True:
+			time.sleep(1)
+			if self._mutex.locked():
+				return
+			tones.beep(500, 100)
+			time.sleep(0.1)
+
+	def stop(self):
+		self._mutex.acquire()
+
+class YandexTranslate(threading.Thread):
+
+	def __init__(self, callback, useLangSwitch=True, **kwargs):
+		super(YandexTranslate, self).__init__()
+		self._callback = callback
+		self._kwargs = kwargs
+		self._host = "translate.yandex.net"
+		self._connection = http.client.HTTPSConnection(self._host)
+		# self._connection = http.client.HTTPSConnection("localhost", 8080)
+		# self._connection.set_tunnel(self._host)
+		self._beeper = None
+		self._useLangSwitch = useLangSwitch
+		self.daemon = True
+		self.start()
+
+	def run(self):
+		if conf["signals"]:
+			self._beeper = Beeper()
+
+		self._kwargs["options"] = "4"
+		self._kwargs["reason"] = "auto"
+		self._kwargs["srv"] = "tr-text"
+		self._kwargs["id"] = conf["key"].encode("utf-8")
+		if isinstance(self._kwargs["text"], str):
+			self._kwargs["text"] = [self._kwargs["text"]]
+
+		self._kwargs["text"] = [s.encode("utf-8") for s in self._kwargs["text"]]
+		self._kwargs["text"] = tuple(self._kwargs["text"])
+
+		status, request = self._HTTPRequest()
+
+		if status:
+			sourceLang, targetLang = request["lang"].split("-")
+			if self._useLangSwitch and sourceLang == targetLang != conf["switchLang"]:
+				self._kwargs["lang"] = "-".join((sourceLang, conf["switchLang"]))
+				status, request = self._HTTPRequest()
+
+		self._connection.close()
+		if self._beeper:
+			self._beeper.stop()
+		wx.CallAfter(self._callback, status, request)
+
+	def _HTTPRequest(self):
+		cacheKey = (self._kwargs["lang"], self._kwargs["text"])
+		if cacheKey in _cache:
+			return True, _cache[cacheKey]
+
+		headers = {
+			"Host": self._host,
+			"Content-Type": "application/x-www-form-urlencoded",
+		}
+
+
+		try:
+			self._connection.request("POST", "/api/v1/tr.json/translate", urlencode(self._kwargs, True), headers)
+			responseData = json.load(self._connection.getresponse())
+		except Exception as e:
+			return False, e
+
+		responseCode = responseData["code"]
+		if responseCode != 200:
+			return False, responseCode
+
+		_cache[cacheKey] = responseData
+		return True, responseData
+
+class GlobalPlugin(globalPluginHandler.GlobalPlugin):
+	scriptCategory = _("Yandex Translate")
+
+	def __init__(self):
+		super(GlobalPlugin, self).__init__()
+		if globalVars.appArgs.secure: return
+
+		self.llastTranslatedText = None
+		self.targetLang = "primaryTargetLang"
+		self.autoTranslate = False
+		speech.speak = self.speakDecorator(speech.speak)
+
+		if languageHandler.getLanguage() in languages:
+			conf["primaryTargetLang"] = languageHandler.getLanguage()
+
+		try:
+			with open(FILE_CONFIG_PATH, "rb") as fileConfig:
+				conf.update(pickle.load(fileConfig))
+		except Exception:
+			pass
+
+		try:
+			yt = YandexFreeTranslate()
+			if conf["key"] == "": conf["key"] = yt.get_key()
+		except Exception as identifier:
+			pass
+
+		# Creates submenu of addon
+		gui.mainFrame.sysTrayIcon.Bind(wx.EVT_MENU,
+			lambda e: gui.mainFrame._popupSettingsDialog(YandexTranslateSettingsDialog),
+			gui.mainFrame.sysTrayIcon.toolsMenu.Append(wx.ID_ANY, _("Yandex Translate Settings...")))
+
+	def speakDecorator(self, speak):
+		def wrapper(speechSequence, *args, **kwargs):
+			self.speechSequence = speechSequence
+			if not self.autoTranslate:
+				return speak(speechSequence, *args, **kwargs)
+			def autoTranslateHandler(status, request):
+				if not status:
+					return speak(speechSequence, *args, **kwargs)
+				translatedSpeechSequence = []
+				t = [s for s in request["text"]] # need copy of the list
+				for item in speechSequence:
+					if isinstance(item, str):
+						translatedSpeechSequence.append(t.pop(0))
+					else:
+						translatedSpeechSequence.append(item)
+				return speak(translatedSpeechSequence, *args, **kwargs)
+			YandexTranslate(autoTranslateHandler, useLangSwitch=False, text=[s for s in speechSequence if isinstance(s, str)], lang=self.getLang())
+		return wrapper
+
+	def errorHandler(self, msg):
+		if isinstance(msg, Exception):
+			text = _("Unfortunately the translation is not available. Please check your Internet connection")
+		else:
+			text = ERRORS.get(msg)
+
+		if text is None:
+			text = _("Error: %s") % msg
+
+		ui.message(text)
+		log.error(msg)
+
+	def translateHandler(self, status, request):
+		if not status:
+			self.errorHandler(request)
+			return
+
+		self.llastTranslatedText = "\n".join(request["text"])
+		ui.message(self.llastTranslatedText)
+
+		if conf["copyToClipBoard"]:
+			api.copyToClip(self.llastTranslatedText)
+
+	def getLang(self):
+		if conf["sourceLang"] == "auto":
+			return conf[self.targetLang]
+		return conf["sourceLang"] + "-" + conf[self.targetLang]
+
+	def getSelectedText(self):
+		obj = api.getCaretObject()
+		try:
+			info = obj.makeTextInfo(textInfos.POSITION_SELECTION)
+			if info or not info.isCollapsed:
+				return info.text
+		except (RuntimeError, NotImplementedError):
+			return None
+
+	@secureScript
+	def script_translate(self, gesture):
+		text = self.getSelectedText()
+		if not text:
+			try:
+				text = api.getClipData()
+			except Exception:
+				ui.message(_("No text to translate"))
+				return
+
+		YandexTranslate(self.translateHandler, text=text, lang=self.getLang())
+	script_translate.__doc__ = _("Translates the selected text. If there is no selection, it translates text from the clipboard")
+
+	@secureScript
+	def script_translateSpokenPhrase(self, gesture):
+		text = " ".join([i for i in self.speechSequence if isinstance(i, str)])
+
+		YandexTranslate(self.translateHandler, text=text, lang=self.getLang())
+	script_translateSpokenPhrase.__doc__ = _("Translates the last spoken phrase")
+
+	@secureScript
+	def script_translateNavigatorObject(self, gesture):
+		obj = api.getNavigatorObject()
+		text = obj.name
+
+		if not text:
+			try:
+				text = obj.makeTextInfo(textInfos.POSITION_ALL).clipboardText
+				if not text: raise RuntimeError()
+			except (RuntimeError, NotImplementedError):
+				ui.message(_("No text to translate"))
+				return
+
+		YandexTranslate(self.translateHandler, text=text, lang=self.getLang())
+	script_translateNavigatorObject.__doc__ = _("Translates text from navigator object")
+
+	@secureScript
+	def script_switchTargetLang(self, gesture):
+		if self.targetLang == "primaryTargetLang":
+			self.targetLang = "secondaryTargetLang"
+		else:
+			self.targetLang = "primaryTargetLang"
+		ui.message(languages[conf[self.targetLang]])
+	script_switchTargetLang.__doc__ = _("Switching between the primary and secondary target language")
+
+	@secureScript
+	def script_copyLlastTranslatedText(self, gesture):
+		if self.llastTranslatedText:
+			api.copyToClip(self.llastTranslatedText)
+			ui.message(_("Copy to clipboard"))
+		else:
+			ui.message(_("No translation to copy"))
+	script_copyLlastTranslatedText.__doc__ = _("Copy last translation to clipboard")
+
+	@secureScript
+	def script_showSettingsDialog(self, gesture):
+		gui.mainFrame._popupSettingsDialog(YandexTranslateSettingsDialog)
+	script_showSettingsDialog.__doc__ = _("Shows the settings dialog")
+
+	def script_switchAutoTranslate(self, gesture):
+		if not self.autoTranslate:
+			ui.message(_("Automatic translation enabled"))
+			self.autoTranslate = True
+		else:
+			self.autoTranslate = False
+			ui.message(_("Automatic translation disabled"))
+	script_switchAutoTranslate.__doc__ = _("Switches the function of automatic translation")