def display_downloaded_content(self):
"""Open downloaded non-html content in a separate application.
Called when an unsupported content type is finished downloading.
"""
file_path = (
QDir.toNativeSeparators(
QDir.tempPath() + "/XXXXXX_" + self.content_filename
)
)
myfile = QTemporaryFile(file_path)
myfile.setAutoRemove(False)
if myfile.open():
myfile.write(self.reply.readAll())
myfile.close()
subprocess.Popen([
(self.config.get("content_handlers")
.get(str(self.content_type))),
myfile.fileName()
])
# Sometimes downloading files opens an empty window.
# So if the current window has no URL, close it.
if(str(self.url().toString()) in ('', 'about:blank')):
self.close()
def display_downloaded_content(self):
"""Open downloaded non-html content in a separate application.
Called when an unsupported content type is finished downloading.
"""
debug("displaying downloaded content from {}".format(self.reply.url()))
file_path = (
QDir.toNativeSeparators(
QDir.tempPath() + "/XXXXXX_" + self.content_filename
)
)
myfile = QTemporaryFile(file_path)
myfile.setAutoRemove(False)
if myfile.open():
myfile.write(self.reply.readAll())
myfile.close()
subprocess.Popen([
(self.config.get("content_handlers")
.get(str(self.content_type))),
myfile.fileName()
])
# Sometimes downloading files opens an empty window.
# So if the current window has no URL, close it.
if(str(self.url().toString()) in ('', 'about:blank')):
self.close()
def run(self):
"""
代码检测工作函数
"""
while 1:
if not self._running:
logger.info('code checker quit')
break
if not self.background_checking:
QThread.msleep(500)
continue
if self._queue.qsize() == 0:
QThread.msleep(500)
continue
try:
widget, code = self._queue.get(False, 0.5)
# 创建临时文件
file = QTemporaryFile(self)
file.setAutoRemove(True)
if file.open():
with open(file.fileName(), 'wb') as fp:
fp.write(code.encode())
file.close()
# 使用flake8检测代码
results = []
with StringIO() as out, redirect_stdout(out):
app = Application()
app.initialize(
['flake8', '--exit-zero', '--config',
os.path.join(os.path.dirname(__file__), 'config', '.flake8')])
app.run_checks([file.fileName()])
app.report()
results = out.getvalue().split('\n')
results = [ret for ret in results if re.search(r'\d+:\d+:[EFW]\d+:.*?', ret)]
# if results:
self.checked.emit(widget, results) # 如果为空，也应该这样做。将空列表传入可以清空所有的标记。
file.deleteLater()
del file
except Exception as e:
logger.warning(str(e))
def save(self):
"""
Public slot to save the history entries to disk.
"""
historyFile = QFile(self.getFileName())
if not historyFile.exists():
self.__lastSavedUrl = ""
saveAll = self.__lastSavedUrl == ""
first = len(self.__history) - 1
if not saveAll:
# find the first one to save
for index in range(len(self.__history)):
if self.__history[index].url == self.__lastSavedUrl:
first = index - 1
break
if first == len(self.__history) - 1:
saveAll = True
if saveAll:
# use a temporary file when saving everything
f = QTemporaryFile()
f.setAutoRemove(False)
opened = f.open()
else:
f = historyFile
opened = f.open(QIODevice.Append)
if not opened:
E5MessageBox.warning(
None,
self.tr("Saving History"),
self.tr("""<p>Unable to open history file <b>{0}</b>.<br/>""" """Reason: {1}</p>""").format(
f.fileName(), f.errorString()
),
)
return
for index in range(first, -1, -1):
data = QByteArray()
stream = QDataStream(data, QIODevice.WriteOnly)
stream.setVersion(QDataStream.Qt_4_6)
itm = self.__history[index]
stream.writeUInt32(HISTORY_VERSION)
stream.writeString(itm.url.encode("utf-8"))
stream << itm.dateTime
stream.writeString(itm.title.encode("utf-8"))
f.write(data)
f.close()
if saveAll:
if historyFile.exists() and not historyFile.remove():
E5MessageBox.warning(
None,
self.tr("Saving History"),
self.tr("""<p>Error removing old history file <b>{0}</b>.""" """<br/>Reason: {1}</p>""").format(
historyFile.fileName(), historyFile.errorString()
),
)
if not f.copy(historyFile.fileName()):
E5MessageBox.warning(
None,
self.tr("Saving History"),
self.tr(
"""<p>Error moving new history file over old one """ """(<b>{0}</b>).<br/>Reason: {1}</p>"""
).format(historyFile.fileName(), f.errorString()),
)
self.historySaved.emit()
try:
self.__lastSavedUrl = self.__history[0].url
except IndexError:
self.__lastSavedUrl = ""
def save(self):
"""
Public slot to save the history entries to disk.
"""
historyFile = QFile(self.getFileName())
if not historyFile.exists():
self.__lastSavedUrl = ""
saveAll = self.__lastSavedUrl == ""
first = len(self.__history) - 1
if not saveAll:
# find the first one to save
for index in range(len(self.__history)):
if self.__history[index].url == self.__lastSavedUrl:
first = index - 1
break
if first == len(self.__history) - 1:
saveAll = True
if saveAll:
# use a temporary file when saving everything
f = QTemporaryFile()
f.setAutoRemove(False)
opened = f.open()
else:
f = historyFile
opened = f.open(QIODevice.Append)
if not opened:
E5MessageBox.warning(
None, self.tr("Saving History"),
self.tr("""<p>Unable to open history file <b>{0}</b>.<br/>"""
"""Reason: {1}</p>""").format(f.fileName(),
f.errorString()))
return
for index in range(first, -1, -1):
data = QByteArray()
stream = QDataStream(data, QIODevice.WriteOnly)
stream.setVersion(QDataStream.Qt_4_6)
itm = self.__history[index]
stream.writeUInt32(HISTORY_VERSION)
stream.writeString(itm.url.encode("utf-8"))
stream << itm.dateTime
stream.writeString(itm.title.encode('utf-8'))
f.write(data)
f.close()
if saveAll:
if historyFile.exists() and not historyFile.remove():
E5MessageBox.warning(
None, self.tr("Saving History"),
self.tr(
"""<p>Error removing old history file <b>{0}</b>."""
"""<br/>Reason: {1}</p>""").format(
historyFile.fileName(), historyFile.errorString()))
if not f.copy(historyFile.fileName()):
E5MessageBox.warning(
None, self.tr("Saving History"),
self.tr(
"""<p>Error moving new history file over old one """
"""(<b>{0}</b>).<br/>Reason: {1}</p>""").format(
historyFile.fileName(), f.errorString()))
self.historySaved.emit()
try:
self.__lastSavedUrl = self.__history[0].url
except IndexError:
self.__lastSavedUrl = ""
class TranslatorWidget(QWidget, Ui_TranslatorWidget):
"""
Class implementing the translator widget.
"""
def __init__(self, plugin, translator, parent=None):
"""
Constructor
@param plugin reference to the plugin object (TranslatorPlugin)
@param translator reference to the translator object (Translator)
@param parent reference to the parent widget (QWidget)
"""
super(TranslatorWidget, self).__init__(parent)
self.setupUi(self)
self.__plugin = plugin
self.__translator = translator
self.__languages = TranslatorLanguagesDb(self)
self.__translatorRequest = None
self.__translationEngine = None
self.__mediaPlayer = None
self.__mediaFile = None
audioAvailable = (MULTIMEDIA_AVAILABLE
and bool(QMediaPlayer.hasSupport("audio/mpeg")))
self.pronounceOrigButton.setVisible(audioAvailable)
self.pronounceTransButton.setVisible(audioAvailable)
self.pronounceOrigButton.setIcon(
self.__translator.getAppIcon("pronounce.png"))
self.pronounceTransButton.setIcon(
self.__translator.getAppIcon("pronounce.png"))
self.swapButton.setIcon(self.__translator.getAppIcon("swap.png"))
self.translateButton.setIcon(
self.__translator.getAppIcon("translate.png"))
self.clearButton.setIcon(UI.PixmapCache.getIcon("editDelete.png"))
self.preferencesButton.setIcon(UI.PixmapCache.getIcon("configure.png"))
self.translateButton.setEnabled(False)
self.clearButton.setEnabled(False)
self.pronounceOrigButton.setEnabled(False)
self.pronounceTransButton.setEnabled(False)
selectedEngine = self.__plugin.getPreferences("SelectedEngine")
self.__updateEngines()
engineIndex = self.engineComboBox.findData(selectedEngine)
self.engineComboBox.setCurrentIndex(engineIndex)
self.__engineComboBoxCurrentIndexChanged(engineIndex)
self.engineComboBox.currentIndexChanged.connect(
self.__engineComboBoxCurrentIndexChanged)
self.__plugin.updateLanguages.connect(self.__updateLanguages)
def __updateLanguages(self):
"""
Private slot to update the language combo boxes.
"""
self.__ensureTranslationEngineReady()
if self.__translationEngine is not None:
supportedCodes = self.__translationEngine.supportedLanguages()
enabledCodes = self.__plugin.getPreferences("EnabledLanguages")
# 1. save current selections
origLanguage = self.origLanguageComboBox.itemData(
self.origLanguageComboBox.currentIndex())
# 2. reload the original language combo box
self.origLanguageComboBox.blockSignals(True)
self.origLanguageComboBox.clear()
for code in enabledCodes:
if code in supportedCodes:
language = self.__languages.getLanguage(code)
if language:
icon = self.__languages.getLanguageIcon(code)
self.origLanguageComboBox.addItem(icon, language, code)
self.origLanguageComboBox.model().sort(0)
origIndex = self.origLanguageComboBox.findData(origLanguage)
if origIndex == -1:
origIndex = 0
self.origLanguageComboBox.blockSignals(False)
self.origLanguageComboBox.setCurrentIndex(origIndex)
def __updateEngines(self):
"""
Private slot to update the engines combo box.
"""
currentEngine = self.engineComboBox.itemData(
self.engineComboBox.currentIndex())
self.engineComboBox.clear()
for engineName in TranslatorEngines.supportedEngineNames():
icon = TranslatorEngines.getEngineIcon(engineName)
self.engineComboBox.addItem(
icon, TranslatorEngines.engineDisplayName(engineName),
engineName)
self.engineComboBox.model().sort(0)
self.engineComboBox.setCurrentIndex(
self.engineComboBox.findData(currentEngine))
def __originalLanguage(self):
"""
Private method to return the code of the selected original language.
@return code of the original language (string)
"""
return self.origLanguageComboBox.itemData(
self.origLanguageComboBox.currentIndex())
def __translationLanguage(self):
"""
Private method to return the code of the selected translation language.
@return code of the translation language (string)
"""
return self.transLanguageComboBox.itemData(
self.transLanguageComboBox.currentIndex())
@pyqtSlot()
def on_translateButton_clicked(self):
"""
Private slot to translate the entered text.
"""
self.transEdit.clear()
result, ok = self.__translate(self.origEdit.toPlainText(),
self.__originalLanguage(),
self.__translationLanguage())
if ok:
self.transEdit.setHtml(result)
else:
E5MessageBox.critical(self, self.tr("Translation Error"), result)
@pyqtSlot()
def on_pronounceOrigButton_clicked(self):
"""
Private slot to pronounce the original text.
"""
self.__pronounce(self.origEdit.toPlainText(),
self.__originalLanguage())
@pyqtSlot()
def on_pronounceTransButton_clicked(self):
"""
Private slot to pronounce the translated text.
"""
self.__pronounce(self.transEdit.toPlainText(),
self.__translationLanguage())
@pyqtSlot()
def on_swapButton_clicked(self):
"""
Private slot to swap the languages.
"""
# save selected language codes
oLanguage = self.origLanguageComboBox.itemData(
self.origLanguageComboBox.currentIndex())
tLanguage = self.transLanguageComboBox.itemData(
self.transLanguageComboBox.currentIndex())
oIdx = self.origLanguageComboBox.findData(tLanguage)
if oIdx < 0:
oIdx = 0
self.origLanguageComboBox.setCurrentIndex(oIdx)
tIdx = self.transLanguageComboBox.findData(oLanguage)
if tIdx < 0:
tIdx = 0
self.transLanguageComboBox.setCurrentIndex(tIdx)
origText = self.origEdit.toPlainText()
self.origEdit.setPlainText(self.transEdit.toPlainText())
self.transEdit.setPlainText(origText)
@pyqtSlot()
def on_clearButton_clicked(self):
"""
Private slot to clear the text fields.
"""
self.origEdit.clear()
self.transEdit.clear()
@pyqtSlot()
def on_origEdit_textChanged(self):
"""
Private slot to handle changes of the original text.
"""
self.__updatePronounceButtons()
self.__updateClearButton()
self.__updateTranslateButton()
@pyqtSlot()
def on_transEdit_textChanged(self):
"""
Private slot to handle changes of the translation text.
"""
self.__updatePronounceButtons()
self.__updateClearButton()
@pyqtSlot(int)
def on_origLanguageComboBox_currentIndexChanged(self, index):
"""
Private slot to handle the selection of the original language.
@param index current index (integer)
"""
self.__plugin.setPreferences("OriginalLanguage",
self.origLanguageComboBox.itemData(index))
supportedTargetCodes = (
self.__translationEngine.supportedTargetLanguages(
self.origLanguageComboBox.itemData(index)))
if supportedTargetCodes is not None:
enabledCodes = self.__plugin.getPreferences("EnabledLanguages")
transLanguage = self.transLanguageComboBox.itemData(
self.transLanguageComboBox.currentIndex())
self.transLanguageComboBox.clear()
if len(supportedTargetCodes) > 0:
for code in enabledCodes:
if code in supportedTargetCodes:
language = self.__languages.getLanguage(code)
if language:
icon = self.__languages.getLanguageIcon(code)
self.transLanguageComboBox.addItem(
icon, language, code)
self.transLanguageComboBox.model().sort(0)
index = self.transLanguageComboBox.findData(transLanguage)
if index == -1:
index = 0
self.transLanguageComboBox.setCurrentIndex(index)
self.__updateTranslateButton()
@pyqtSlot(int)
def on_transLanguageComboBox_currentIndexChanged(self, index):
"""
Private slot to handle the selection of the translation language.
@param index current index (integer)
"""
self.__plugin.setPreferences(
"TranslationLanguage", self.transLanguageComboBox.itemData(index))
@pyqtSlot()
def __availableTranslationsLoaded(self):
"""
Private slot to handle the availability of translations.
"""
origLanguage = self.__plugin.getPreferences("OriginalLanguage")
transLanguage = self.__plugin.getPreferences("TranslationLanguage")
self.__updateLanguages()
origIndex = self.origLanguageComboBox.findData(origLanguage)
self.origLanguageComboBox.setCurrentIndex(origIndex)
self.on_origLanguageComboBox_currentIndexChanged(origIndex)
self.transLanguageComboBox.setCurrentIndex(
self.transLanguageComboBox.findData(transLanguage))
def __ensureTranslationEngineReady(self):
"""
Private slot to ensure, that the currently selected translation engine
is ready.
"""
engineName = self.engineComboBox.itemData(
self.engineComboBox.currentIndex())
if (self.__translationEngine is not None
and self.__translationEngine.engineName() != engineName):
self.__translationEngine.availableTranslationsLoaded.disconnect(
self.__availableTranslationsLoaded)
self.__translationEngine.deleteLater()
self.__translationEngine = None
if self.__translationEngine is None:
self.__translationEngine = TranslatorEngines.getTranslationEngine(
engineName, self.__plugin, self)
if self.__translationEngine is not None:
self.__translationEngine.availableTranslationsLoaded.connect(
self.__availableTranslationsLoaded)
@pyqtSlot(int)
def __engineComboBoxCurrentIndexChanged(self, index):
"""
Private slot to handle the selection of a translation service.
@param index current index
@type int
"""
self.__ensureTranslationEngineReady()
if self.__translationEngine is not None:
self.__updateTranslateButton()
self.__updatePronounceButtons()
self.__plugin.setPreferences("SelectedEngine",
self.engineComboBox.itemData(index))
def __updatePronounceButtons(self):
"""
Private slot to set the state of the pronounce buttons.
"""
if self.__translationEngine is not None:
hasTTS = self.__translationEngine.hasTTS()
else:
hasTTS = False
self.pronounceOrigButton.setEnabled(
hasTTS and bool(self.origEdit.toPlainText()))
self.pronounceTransButton.setEnabled(
hasTTS and bool(self.transEdit.toPlainText()))
def __updateClearButton(self):
"""
Private slot to set the state of the clear button.
"""
enable = (bool(self.origEdit.toPlainText())
or bool(self.transEdit.toPlainText()))
self.clearButton.setEnabled(enable)
def __updateTranslateButton(self):
"""
Private slot to set the state of the translate button.
"""
enable = bool(self.origEdit.toPlainText())
enable &= bool(self.__translationLanguage())
enable &= bool(self.__originalLanguage())
self.translateButton.setEnabled(enable)
def __translate(self, text, originalLanguage, translationLanguage):
"""
Private method to translate the given text.
@param text text to be translated (string)
@param originalLanguage language code of the original (string)
@param translationLanguage language code of the translation (string)
@return tuple of translated text (string) and flag indicating
success (boolean)
"""
if self.__translatorRequest is None:
from .TranslatorRequest import TranslatorRequest
self.__translatorRequest = TranslatorRequest(self)
self.__ensureTranslationEngineReady()
if self.__translationEngine is None:
return "", False
else:
result, ok = self.__translationEngine.getTranslation(
self.__translatorRequest, text, originalLanguage,
translationLanguage)
return result, ok
def __pronounce(self, text, language):
"""
Private method to pronounce the given text.
@param text text to be pronounced (string)
@param language language code of the text (string)
"""
if not text or not language:
return
if self.__translatorRequest is None:
from .TranslatorRequest import TranslatorRequest
self.__translatorRequest = TranslatorRequest(self)
if self.__mediaPlayer is None:
self.__mediaPlayer = QMediaPlayer(self)
self.__mediaPlayer.stateChanged.connect(
self.__mediaPlayerStateChanged)
if self.__mediaPlayer.state() == QMediaPlayer.PlayingState:
return
self.__ensureTranslationEngineReady()
if self.__translationEngine is not None:
if not self.__translationEngine.hasTTS():
E5MessageBox.critical(
self, self.tr("Translation Error"),
self.tr("The selected translation service does not"
" support the Text-to-Speech function."))
return
data, ok = self.__translationEngine.getTextToSpeechData(
self.__translatorRequest, text, language)
if ok:
self.__mediaFile = QTemporaryFile(self)
self.__mediaFile.open()
self.__mediaFile.setAutoRemove(False)
self.__mediaFile.write(data)
self.__mediaPlayer.setMedia(QMediaContent(), self.__mediaFile)
self.__mediaPlayer.play()
else:
E5MessageBox.critical(self, self.tr("Translation Error"), data)
def __mediaPlayerStateChanged(self, state):
"""
Private slot handling changes of the media player state.
@param state media player state (QAudio.State)
"""
if state == QMediaPlayer.StoppedState:
self.__mediaFile.close()
self.__mediaFile.remove()
self.__mediaFile = None
@pyqtSlot()
def on_preferencesButton_clicked(self):
"""
Private slot to open the Translator configuration page.
"""
e5App().getObject("UserInterface").showPreferences("translatorPage")
def save(self, parent, filename=None):
"""Saves the current file and sets is_modified to False.
If the file has never been saved (i.e., doesn't have a filename) or is not writable, this method calls save_as.
:type parent: QWidget - The parent window for dialogs, alerts, etc.
:type filename: str - Full path of file or None (the default)
:rtype: bool - True = Successfully saved file, False = error (user was notified) or user cancelled
"""
if filename is None:
filename = self.filename
# If the current file has never been saved...
if filename is None:
# ...call save_as instead
self.on_file_saveas()
wavfile = QFile(filename)
# If the original file exists, but can't be opened for writing...
if wavfile.exists() and not wavfile.open(QIODevice.ReadWrite):
# ...then treat this as a Save-As command
return self.save_as(parent)
tmpfile = QTemporaryFile(filename)
# If we can't open our temporary file for writing...
if not QTemporaryFile.open(QIODevice.WriteOnly):
# ...something's wrong (disk full?), so report it to the user and exit
QMessageBox.critical(
parent, make_caption('Warning'),
"Unable to create temporary file:\n\n\t<b>{}</b>\n\n\n{}".
format(tmpfile.fileName(), tmpfile.errorString()))
return False
# If the original file exists...
if wavfile.exists():
# ...set the temporary file permissions to match the original
tmpfile.setPermissions(wavfile.permissions()) # Ignore errors
elif sys.platform == 'win32':
# ...otherwise (on Windows) use standard permissions
tmpfile.setPermissions(QFile.WriteGroup | QFile.WriteOther)
elif sys.platform.startswith('linux') or sys.platform.startswith(
'darwin'):
# ...otherwise (on Linux) use the file mode creation mask for the current process
tmpfile.setPermissions(self.umask_as_permission()
& Document.PERMISSION_MASK)
else:
raise RuntimeError(
'Unsupported platform: ' +
sys.platform) # TODO - check this at startup, not here
tmpfile.close() # TODO - is this necessary?
# Write out the WAV file
try:
with wave.open(tmpfile.fileName(), mode='wb') as wave_write:
# Set the audio file properties
wave_write.setnchannels(self.channels)
wave_write.setsampwidth(self.bytes_per_sample)
wave_write.setframerate(self.sampling_rate)
wave_write.setnframes(self.frames)
wave_write.setcomptype(None)
# Write the contents of memory out to the file
wave_write.writeframes(self.frames)
except wave.Error as e:
# An error occurred, notify user and return False
QMessageBox.critical(
parent, make_caption('Warning'),
"Unable to write file:\n\n\t<b>{}</b>\n\n\n{}".format(
tmpfile.fileName(), e.args[0]))
return False
# Data was written successfully to temp file, so inform QTemporaryFile not to remove it automatically; we're
# going to rename it to the original file
tmpfile.setAutoRemove(False)
backup_filename = filename + '~'
# Remove any previous backup
QFile.remove(backup_filename)
# Rename the original file to the backup name
QFile.rename(filename, backup_filename)
# Rename the temporary file to the original name
if not tmpfile.rename(filename):
# If anything goes wrong, rename the backup file back to the original name
QFile.rename(backup_filename, filename)
QMessageBox.critical(
parent, make_caption('Warning'),
"Unable to rename file:\n\n\tFrom: <b>{}</b>\n\tTo: <b>{}</b>\n\n\n{}"
.format(backup_filename, filename, tmpfile.errorString()))
return False
settings = QSettings()
if not settings.value(
Settings.CREATE_BACKUP_ON_SAVE,
defaultValue=Settings.CREATE_BACKUP_ON_SAVE_DFLT):
QFile.remove(backup_filename)
self.is_modified = False
self._set_filename(filename)
return True
