chenx221/Textractor_test
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

small improvements

Browse Source 
Added check for the empty answer from the translator
Small refactors
This commit is contained in:
zeheyler 2020-12-09 14:00:57 +03:00
parent 6958b22e68
commit ea6ddc7a7c
2 changed files with 27 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
extensions/devtoolsdeepltranslate.cpp
View File

@ -15,6 +15,7 @@ const wchar_t* ERROR_GOT_TIMEOUT = L"Error: timeout (s)";
const wchar_t* ERROR_COMMAND_FAIL = L"Error: command failed";
const wchar_t* ERROR_LANGUAGE = L"Error: target languages do not match";
const wchar_t* ERROR_NOTE = L"Error: notification";
const wchar_t* ERROR_EMPTY_ANSWER = L"Error: empty translation";
QString URL = "https://www.deepl.com/en/translator";
QStringList languagesTo
@ -56,6 +57,7 @@ std::pair<bool, std::wstring> Translate(const std::wstring& text, DevTools* devT
{
QString qtext = S(text);
qtext.remove(QString(12288)); // japanese space (no need for translator)
qtext.replace(QString(12289), ","); // replace the japanese comma with the latin comma for correct translation
// Remove quotes
bool checkQuote = false;
@ -148,7 +150,7 @@ std::pair<bool, std::wstring> Translate(const std::wstring& text, DevTools* devT
std::this_thread::sleep_for(std::chrono::milliseconds(100));
timer += 0.1;
}
if (timer >= timerStop)
if (timer >= 2 * timerStop)
errorCode = 5;
// Set methods to receive
@ -243,7 +245,9 @@ std::pair<bool, std::wstring> Translate(const std::wstring& text, DevTools* devT
// Catch the translation
QString OuterHTML;
if (errorCode == 0 && !devTools->SendRequest("DOM.getOuterHTML", { {"nodeId", targetNodeId + 1} }, root))
if (errorCode == 0)
{
if (!devTools->SendRequest("DOM.getOuterHTML", { {"nodeId", targetNodeId + 1} }, root))
{
targetNodeId = -1;
errorCode = 1;
@ -252,14 +256,18 @@ std::pair<bool, std::wstring> Translate(const std::wstring& text, DevTools* devT
{
OuterHTML = root.value("result").toObject().value("outerHTML").toString();
}
if (OuterHTML == "<div></div>")
}
if (errorCode == 0 && OuterHTML == "<div></div>")
{
// Try to catch the notification
int noteNodeId = -1;
if (errorCode == 0 && (!devTools->SendRequest("DOM.querySelector", { {"nodeId", docFound}, {"selector", "div.lmt__system_notification"} }, root)
|| root.value("result").toObject().value("nodeId").toInt() == 0) && timer >= timerStop)
if (!devTools->SendRequest("DOM.querySelector", { {"nodeId", docFound}, {"selector", "div.lmt__system_notification"} }, root)
|| root.value("result").toObject().value("nodeId").toInt() == 0)
{
if (timer >= timerStop)
errorCode = 2;
else
errorCode = 6;
}
else
{
@ -340,6 +348,8 @@ std::pair<bool, std::wstring> Translate(const std::wstring& text, DevTools* devT
return { false, FormatString(L"%s (%s): %s", ERROR_LANGUAGE, S(targetLang), S(OuterHTML)) };
else if (errorCode == 5)
return { false, FormatString(L"%s: %d", ERROR_GOT_TIMEOUT, 2*timerStop) };
else if (errorCode == 6)
return { false, FormatString(L"%s", ERROR_EMPTY_ANSWER) };
else
return { false, FormatString(L"%s", TRANSLATION_ERROR) };
}

5
extensions/devtoolswrapper.cpp
View File

@ -62,6 +62,11 @@ void EraseControlCharacters(std::wstring& text)
{
text.erase(it--);
}
if (*it == '\\' && *(it + 1) == 'n')
{
text.erase((it + 1)--);
text.erase(it--);
}
}
}