Merge pull request #10 from dkalachov/master

'Save mp3' item in context menu for audio

Author: ciscorn

README.md

@@ -63,8 +63,17 @@ For Arch Linux, two packages [ldoce5viewer](https://aur.archlinux.org/packages/l
 
 (for advanced users)
 
+*Homebrew*:
+```bash
+$ brew install pyqt
+$ pip install lxml pyobjc-core pyobjc-framework-Cocoa whoosh py2app
+$ # inside ldoce5viewer directory
+$ sudo DISTUTILS_DEBUG=1 python setup.py py2app
+$ open dist/LDOCE5\ Viewer.app/
+```
+
+Or if you are using *MacPorts*:
 <ol>
-  <li><p>Install MacPorts</p></li>
   <li><p>Install the following ports:</p>
     <ul>
       <li>python27 (or python3x)</li>

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()

setup.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python
 
+import subprocess
 from distutils.core import setup
 from ldoce5viewer import __version__
 
@@ -60,6 +61,8 @@ def iter_static():
 except ImportError:
     pass
 else:
+    qt_plugins_path = subprocess.check_output('qmake -query QT_INSTALL_PLUGINS', shell=True)
+    qt_plugins_path = qt_plugins_path[0:len(qt_plugins_path)-1] # remove "\n"
     extra_options.update(dict(
         name='LDOCE5 Viewer',
         app=['ldoce5viewer.py'],
@@ -90,7 +93,7 @@ def iter_static():
             #]
         }},
         data_files=[
-            ('qt_plugins/imageformats', ['/opt/local/share/qt4/plugins/imageformats/libqjpeg.dylib']),
+            ('qt_plugins/imageformats', [qt_plugins_path]),
             ('', ['ldoce5viewer/static']),
         ],
     ))