|
1 # -*- coding: utf-8 -*- |
|
2 |
|
3 # Copyright (c) 2014 - 2019 Detlev Offenbach <detlev@die-offenbachs.de> |
|
4 # |
|
5 |
|
6 """ |
|
7 Module implementing the Yandex translation engine. |
|
8 """ |
|
9 |
|
10 from __future__ import unicode_literals |
|
11 try: |
|
12 str = unicode |
|
13 except NameError: |
|
14 pass |
|
15 |
|
16 import json |
|
17 |
|
18 from PyQt5.QtCore import QUrl, QByteArray, QTimer |
|
19 |
|
20 import Utilities |
|
21 |
|
22 from .TranslationEngine import TranslationEngine |
|
23 |
|
24 |
|
25 class YandexEngine(TranslationEngine): |
|
26 """ |
|
27 Class implementing the translation engine for the Yandex |
|
28 translation service. |
|
29 """ |
|
30 TranslatorUrl = "https://translate.yandex.net/api/v1.5/tr.json/translate" |
|
31 TranslatorLimit = 10000 |
|
32 |
|
33 def __init__(self, plugin, parent=None): |
|
34 """ |
|
35 Constructor |
|
36 |
|
37 @param plugin reference to the plugin object (TranslatorPlugin) |
|
38 @param parent reference to the parent object (QObject) |
|
39 """ |
|
40 super(YandexEngine, self).__init__(plugin, parent) |
|
41 |
|
42 self.__errors = { |
|
43 401: self.tr("Invalid API key."), |
|
44 402: self.tr("API key has been blocked."), |
|
45 403: self.tr("Daily limit for requests has been reached."), |
|
46 404: self.tr("Daily limit for the volume of translated text" |
|
47 " reached."), |
|
48 413: self.tr("Text size exceeds the maximum."), |
|
49 422: self.tr("Text could not be translated."), |
|
50 501: self.tr("The specified translation direction is not" |
|
51 " supported."), |
|
52 } |
|
53 |
|
54 QTimer.singleShot(0, self.availableTranslationsLoaded.emit) |
|
55 |
|
56 def engineName(self): |
|
57 """ |
|
58 Public method to return the name of the engine. |
|
59 |
|
60 @return engine name (string) |
|
61 """ |
|
62 return "yandex" |
|
63 |
|
64 def supportedLanguages(self): |
|
65 """ |
|
66 Public method to get the supported languages. |
|
67 |
|
68 @return list of supported language codes (list of string) |
|
69 """ |
|
70 return ["ar", "be", "bg", "bs", "ca", "cs", "da", "de", "el", "en", |
|
71 "es", "et", "fi", "fr", "ga", "gl", "hi", "hr", "hu", "id", |
|
72 "is", "it", "iw", "ja", "ka", "ko", "lt", "lv", "mk", "mt", |
|
73 "nl", "no", "pl", "pt", "ro", "ru", "sk", "sl", "sq", "sr", |
|
74 "sv", "th", "tl", "tr", "uk", "vi", "zh-CN", "zh-TW", |
|
75 ] |
|
76 |
|
77 def getTranslation(self, requestObject, text, originalLanguage, |
|
78 translationLanguage): |
|
79 """ |
|
80 Public method to translate the given text. |
|
81 |
|
82 @param requestObject reference to the request object |
|
83 (TranslatorRequest) |
|
84 @param text text to be translated (string) |
|
85 @param originalLanguage language code of the original (string) |
|
86 @param translationLanguage language code of the translation (string) |
|
87 @return tuple of translated text (string) and flag indicating |
|
88 success (boolean) |
|
89 """ |
|
90 if len(text) > self.TranslatorLimit: |
|
91 return (self.tr("Only texts up to {0} characters are allowed.") |
|
92 .format(self.TranslatorLimit), False) |
|
93 |
|
94 apiKey = self.plugin.getPreferences("YandexKey") |
|
95 if not apiKey: |
|
96 return self.tr("A valid Yandex key is required."), False |
|
97 |
|
98 params = QByteArray( |
|
99 "key={0}&lang={1}-{2}&text=".format( |
|
100 apiKey, originalLanguage, translationLanguage).encode("utf-8")) |
|
101 encodedText = QByteArray(Utilities.html_encode(text).encode("utf-8"))\ |
|
102 .toPercentEncoding() |
|
103 request = params + encodedText |
|
104 response, ok = requestObject.post(QUrl(self.TranslatorUrl), request) |
|
105 if ok: |
|
106 try: |
|
107 responseDict = json.loads(response) |
|
108 except ValueError: |
|
109 return self.tr("Invalid response received"), False |
|
110 |
|
111 if responseDict["code"] != 200: |
|
112 try: |
|
113 error = self.__errors[responseDict["code"]] |
|
114 except KeyError: |
|
115 error = self.tr("Unknown error code ({0}) received.")\ |
|
116 .format(responseDict["code"]) |
|
117 return error, False |
|
118 |
|
119 sentences = responseDict["text"] |
|
120 result = "" |
|
121 for sentence in sentences: |
|
122 result += sentence.replace("\n", "<br/>") |
|
123 else: |
|
124 result = response |
|
125 return result, ok |