global s/frontends/applications/

weboob/applications/qvideoob/main_window.py

@@ -0,0 +1,116 @@
+# -*- coding: utf-8 -*-
+
+# Copyright(C) 2010  Romain Bignon
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, version 3 of the License.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+
+
+from PyQt4.QtCore import SIGNAL
+
+from weboob.tools.application.qt import QtMainWindow, QtDo
+
+from weboob.frontends.qvideoob.ui.main_window_ui import Ui_MainWindow
+
+from .video import Video
+from .minivideo import MiniVideo
+
+class MainWindow(QtMainWindow):
+    def __init__(self, config, weboob, parent=None):
+        QtMainWindow.__init__(self, parent)
+        self.ui = Ui_MainWindow()
+        self.ui.setupUi(self)
+
+        self.config = config
+        self.weboob = weboob
+        self.minivideos = []
+
+        self.ui.backendEdit.addItem('All backends', '')
+        for i, backend in enumerate(self.weboob.iter_backends()):
+            self.ui.backendEdit.addItem(backend.name, backend.name)
+            if backend.name == self.config.get('settings', 'backend'):
+                self.ui.backendEdit.setCurrentIndex(i+1)
+        self.ui.sortbyEdit.setCurrentIndex(int(self.config.get('settings', 'sortby')))
+        self.ui.nsfwCheckBox.setChecked(int(self.config.get('settings', 'nsfw')))
+        self.ui.sfwCheckBox.setChecked(int(self.config.get('settings', 'sfw')))
+
+        self.connect(self.ui.searchEdit, SIGNAL("returnPressed()"), self.search)
+        self.connect(self.ui.urlEdit, SIGNAL("returnPressed()"), self.openURL)
+        self.connect(self.ui.nsfwCheckBox, SIGNAL("stateChanged(int)"), self.nsfwChanged)
+        self.connect(self.ui.sfwCheckBox, SIGNAL("stateChanged(int)"), self.sfwChanged)
+
+    def nsfwChanged(self, state):
+        self.config.set('settings', 'nsfw', int(self.ui.nsfwCheckBox.isChecked()))
+        self.updateVideosDisplay()
+
+    def sfwChanged(self, state):
+        self.config.set('settings', 'sfw', int(self.ui.sfwCheckBox.isChecked()))
+        self.updateVideosDisplay()
+
+    def updateVideosDisplay(self):
+        for minivideo in self.minivideos:
+            if (minivideo.video.nsfw and self.ui.nsfwCheckBox.isChecked() or
+                not minivideo.video.nsfw and self.ui.sfwCheckBox.isChecked()):
+                minivideo.show()
+            else:
+                minivideo.hide()
+
+    def search(self):
+        pattern = unicode(self.ui.searchEdit.text())
+        if not pattern:
+            return
+
+        for minivideo in self.minivideos:
+            self.ui.scrollAreaContent.layout().removeWidget(minivideo)
+            minivideo.hide()
+
+        self.minivideos = []
+        self.ui.searchEdit.setEnabled(False)
+
+        backend_name = str(self.ui.backendEdit.itemData(self.ui.backendEdit.currentIndex()).toString())
+
+        self.process = QtDo(self.weboob, self.addVideo)
+        if backend_name:
+            self.process.do_backends(backend_name, 'iter_search_results', pattern, self.ui.sortbyEdit.currentIndex(), nsfw=True)
+        else:
+            self.process.do('iter_search_results', pattern, self.ui.sortbyEdit.currentIndex(), nsfw=True)
+
+    def addVideo(self, backend, video):
+        if not backend:
+            self.ui.searchEdit.setEnabled(True)
+            return
+        minivideo = MiniVideo(backend, video)
+        self.ui.scrollAreaContent.layout().addWidget(minivideo)
+        self.minivideos.append(minivideo)
+        if (video.nsfw and not self.ui.nsfwCheckBox.isChecked() or
+            not video.nsfw and not self.ui.sfwCheckBox.isChecked()):
+            minivideo.hide()
+
+    def openURL(self):
+        url = unicode(self.ui.urlEdit.text())
+        if not url:
+            return
+
+        for backend in self.weboob.iter_backends():
+            video = backend.get_video(url)
+            if video:
+                video_widget = Video(video, self)
+                video_widget.show()
+
+        self.ui.urlEdit.clear()
+
+    def closeEvent(self, ev):
+        self.config.set('settings', 'backend', str(self.ui.backendEdit.itemData(self.ui.backendEdit.currentIndex()).toString()))
+        self.config.set('settings', 'sortby', self.ui.sortbyEdit.currentIndex())
+        self.config.save()
+        ev.accept()