save mp3 in context menu

Author: 646b

ldoce5viewer/qtgui/main.py

@@ -540,16 +540,28 @@ def _onTextEdited(self, text):
     #----------
 
     def _playbackAudio(self, path):
+        self._getAudioData(path, lambda data: self._soundplayer.play(data))
+
+    def _getAudioData(self,  path,  callback):
         (archive, name) = path.lstrip('/').split('/', 1)
         if archive in ('us_hwd_pron', 'gb_hwd_pron', 'exa_pron', 'sfx', 'sound'):
             def finished():
                 if reply.error() == QNetworkReply.NoError:
-                    self._soundplayer.play(reply.readAll())
+                    callback(reply.readAll())
 
             url = QUrl('dict:///{0}/{1}'.format(archive, name))
             reply = self._networkAccessManager.get(QNetworkRequest(url))
             reply.finished.connect(finished)
 
+    def downloadSelectedAudio(self):
+        path = self._ui.webView.audioUrlToDownload.path()
+        def showSaveDialog(data):
+            filename = QFileDialog.getSaveFileName(self,  u'Save mp3',  '',  u'MP3 Files (*.mp3)')
+            if filename != '':
+                file = open(filename, "w")
+                file.write(data)
+                file.close()
+        self._getAudioData(path, showSaveDialog)
 
     def _onWebViewLinkClicked(self, url):
         scheme = url.scheme()
@@ -1126,6 +1138,7 @@ def act_conn(action, slot):
         act_conn(ui.actionSearchDefinitions, self._onSearchDefinitions)
         act_conn(ui.actionAdvancedSearch, self._onAdvancedSearch)
         act_conn(wv.actionSearchText, self.searchSelectedText)
+        act_conn(wv.actionDownloadAudio, self.downloadSelectedAudio)
         act_conn(ui.actionZoomIn, partial(self.setZoom, 1, relative=True))
         act_conn(ui.actionZoomOut, partial(self.setZoom, -1, relative=True))
         act_conn(ui.actionNormalSize, partial(self.setZoom, 0))

ldoce5viewer/qtgui/ui/custom.py

@@ -192,6 +192,7 @@ def __init__(self, parent):
         self._actionCopyPlain.setShortcut(QKeySequence.Copy)
         self.page().selectionChanged.connect(self.__onSelectionChanged)
         self.__onSelectionChanged()
+        self._actionDownloadAudio = QAction(u'Download mp3',  self)
 
     def _copyAsPlainText(self):
         text = self.selectedText().strip()
@@ -205,6 +206,14 @@ def actionSearchText(self):
     def actionCopyPlain(self):
         return self._actionCopyPlain
 
+    @property
+    def actionDownloadAudio(self):
+        return self._actionDownloadAudio
+
+    @property
+    def audioUrlToDownload(self):
+        return self._audioUrlToDownload
+
     def __onSelectionChanged(self):
         text = self.selectedText()
         self._actionCopyPlain.setEnabled(bool(text))
@@ -213,6 +222,14 @@ def contextMenuEvent(self, event):
         page = self.page()
         menu = page.createStandardContextMenu()
         actions = menu.actions()
+        
+        # inserts the "Download audio" action
+        frame = page.frameAt(event.pos())
+        hit_test_result = frame.hitTestContent(event.pos())
+        if hit_test_result.linkUrl().scheme() == 'audio':
+            self._audioUrlToDownload = hit_test_result.linkUrl()
+            menu.insertAction(actions[0] if actions else None,
+                    self.actionDownloadAudio)
 
         # inserts the "Search for ..." action
         text = page.selectedText().strip().lower()