Exemple #1
0
    def __init__(self, parent=None):
        QCoreApplication.setOrganizationName("Hamster Inc.")
        QCoreApplication.setApplicationName("Hamster")
        QtGui.QMainWindow.__init__(self, parent)
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        self.index_thread = False
        self._shutdown_requested = False
        self.downloader = DownloadManager()
        self.downloader.dl_finished.connect(self.update_cover)
        header = ['Movie']

        #self.db = MovieDB("movies")
        tv = self.ui.tableView
        tv.setShowGrid(False)
        #model = MyTableModel(self.db, header, tv)
        #titles = self.db.get_movie_titles()
        self.index = get_user_index()
        self.db = get_user_db()
        results = self.index.list_all()
        self.model = ResultViewModel(results, header, tv)
        #model = QtGui.QStandardItemModel()
        #model.insertRow(0, [QtGui.QStandardItem("hallo")])
        #model.insertRow(0, [QtGui.QStandardItem("sadf")])
        #model.insertRow(0, [QtGui.QStandardItem("pfui")])
        #model.insertRow(0, [QtGui.QStandardItem("warum?")])
        #model.insertRow(0, [QtGui.QStandardItem("haeff")])
        tv.setModel(self.model)
        selectionModel = tv.selectionModel()
        # hide vertical header
        vh = tv.verticalHeader()
        vh.setVisible(False)

        # set horizontal header properties
        hh = tv.horizontalHeader()
        hh.setStretchLastSection(True)

        selectionModel.selectionChanged.connect(self.setCurrentSelection)
        self.ui.search_bar.textChanged.connect(self.update_model)
        self.ui.action_sync_now.triggered.connect(self.sync)
        self.settings = QSettings()
        global MOVIE_DIR
        MOVIE_DIR = self.settings.value("movie_dir", MOVIE_DIR)
        self.settings.setValue("movie_dir", MOVIE_DIR)
Exemple #2
0
class MyForm(QtGui.QMainWindow):
    shutmedown = Signal()
    def __init__(self, parent=None):
        QCoreApplication.setOrganizationName("Hamster Inc.")
        QCoreApplication.setApplicationName("Hamster")
        QtGui.QMainWindow.__init__(self, parent)
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        self.index_thread = False
        self._shutdown_requested = False
        self.downloader = DownloadManager()
        self.downloader.dl_finished.connect(self.update_cover)
        header = ['Movie']

        #self.db = MovieDB("movies")
        tv = self.ui.tableView
        tv.setShowGrid(False)
        #model = MyTableModel(self.db, header, tv)
        #titles = self.db.get_movie_titles()
        self.index = get_user_index()
        self.db = get_user_db()
        results = self.index.list_all()
        self.model = ResultViewModel(results, header, tv)
        #model = QtGui.QStandardItemModel()
        #model.insertRow(0, [QtGui.QStandardItem("hallo")])
        #model.insertRow(0, [QtGui.QStandardItem("sadf")])
        #model.insertRow(0, [QtGui.QStandardItem("pfui")])
        #model.insertRow(0, [QtGui.QStandardItem("warum?")])
        #model.insertRow(0, [QtGui.QStandardItem("haeff")])
        tv.setModel(self.model)
        selectionModel = tv.selectionModel()
        # hide vertical header
        vh = tv.verticalHeader()
        vh.setVisible(False)

        # set horizontal header properties
        hh = tv.horizontalHeader()
        hh.setStretchLastSection(True)

        selectionModel.selectionChanged.connect(self.setCurrentSelection)
        self.ui.search_bar.textChanged.connect(self.update_model)
        self.ui.action_sync_now.triggered.connect(self.sync)
        self.settings = QSettings()
        global MOVIE_DIR
        MOVIE_DIR = self.settings.value("movie_dir", MOVIE_DIR)
        self.settings.setValue("movie_dir", MOVIE_DIR)

    def sync(self):
        self.index_thread = IndexThread(MOVIE_DIR)
        print self.index_thread.finished.connect(self._indexer_closed)
        print self.shutmedown.connect(self.index_thread.set_stopped,
                Qt.QueuedConnection)
        self.index_thread.start()
        L.d("syncer started")

    def update_cover(self, url, filename):
        if self.current_url == url:
            img = QtGui.QImage(filename)
            self.ui.l_img.setPixmap(QtGui.QPixmap.fromImage(img))

    def update_model(self, new_text):
        search_string = str(new_text)
        if search_string:
            titles = self.index.query(new_text)
        else:
            titles = self.index.list_all()
        self.model.setResults(titles)

    def setCurrentSelection(self, newSelection, oldSelection):
        indexes = newSelection.at(0).indexes()
        if not len(indexes) == 1:
            return
        item = indexes[0]
        id = self.model.getIdForRow(item.row())
        movie_doc = self.db.get_doc(id)
        movie = json.loads(movie_doc.content)
        plot_short = movie.get('plot outline', "")
        plot = movie.get('plot', [""])[0]
        genres = movie.get('genres', [""])
        cast = movie.get('cast', [""])
        countries = movie.get('countries', [""])
        runtime = movie.get('runtimes', [""])[0]
        # handle stuff like "USA:107" in runtime array
        try:
            if ":" in runtime:
                runtime = runtime.split(":")[1]
            runtime = humanize_mins(runtime)
        except:
            runtime = movie.get('runtimes', [""])[0] # reload
            print "could not humanize", runtime, "for", id
        imdb_rating = str(movie.get('rating', "-"))
        votes = movie.get('votes', "-")
        rating = RICHTEXT_RATING % (imdb_rating, votes)
        director = movie.get('director', ["-"])[0]["name"]
        title = movie['long imdb title']
        title = '<span style=" font-size:16pt; font-weight:600;"> '+ title + '</span>'
        self.ui.l_title.setText(title)
        self.ui.l_plot.setText(plot)
        self.ui.l_plot_short.setText(plot_short)
        self.ui.l_rating.setText(rating)
        self.ui.l_director.setText(director)
        self.ui.l_runtime.setText(runtime)
        self.ui.l_genres.setText(", ".join(genres))
        self.ui.l_genres.setCursorPosition(0)
        self.ui.l_countries.setText(", ".join(countries))
        self.ui.l_countries.setCursorPosition(0)
        cast = [c['name'] for c in cast]
        self.ui.l_cast.setText(", ".join(cast[:4]))
        self.ui.l_cast.setCursorPosition(0)
        #url = movie.get('full-size cover url', None)
        url = movie.get('cover url', None)
        self.ui.l_img.setText("-")
        self.current_url = url
        if url:
            #filename, msg = urllib.urlretrieve(url)
            self.downloader.do_download(url)

    def _indexer_closed(self):
        L.d("indexer finished")
        if self._shutdown_requested:
            self.close()

    def closeEvent(self, event):
        L.d("shutdown requested")
        L.d("hiding window")
        self.setVisible(False)
        if self.index_thread and self.index_thread.isRunning():
            event.ignore()
            L.d("setting quit")
            self._shutdown_requested = True
            self.index_thread.set_stopped()
            #self.shutmedown.emit()
            #QTimer.singleShot(10, self.really_close)
        else:
            L.d("goodby")
            event.accept()

    def really_close(self):
        #L.d("stopping indexer thread")
        #self.index_thread.quit()
        #L.d("waiting for threadpool to be done...")
        #from PySide.QtCore import QThreadPool
        #tp = QThreadPool.globalInstance()
        #tp.waitForDone()
        L.d("Threadpool done")
        self.close()