#!/usr/bin/python import sys,os import gettext import apt from PyQt4.QtCore import * from PyQt4.QtGui import * gettext.textdomain('kaffeine-install-codecs') def _(string): return unicode(gettext.gettext(string), 'utf-8') if len(sys.argv) < 2: sys.exit(1) class InstallerOpProgress(apt.OpProgress): def __init__(self, installer): self.installer = installer def update(self, percent): self.installer.progress.setValue(percent) QApplication.processEvents() def done(self): self.installer.progress.setValue(100) QApplication.processEvents() class CodecInstallerWidget(QDialog, apt.OpProgress): def __init__(self): QDialog.__init__(self) self.setWindowTitle(_('Kaffeine codec installer')) self.setModal(True) self.status = QLabel() self.status.setAlignment(Qt.AlignHCenter | Qt.AlignVCenter) self.progress = QProgressBar() layout = QVBoxLayout() layout.addWidget(self.status) layout.addWidget(self.progress) self.setLayout(layout) self.setStatus(_('Initializing...')) desktop = QApplication.desktop() self.setMinimumWidth(desktop.width() / 3) self.adjustSize() self.move((desktop.width()-self.width()) / 2, (desktop.height()-self.height()) / 2) self.show() self.run() def setStatus(self, text): self.status.setText('<i>' + text + '</i>') QApplication.processEvents() def init_apt_cache(self): self.setStatus(_('Preparing...')) self.apt_cache = apt.Cache(InstallerOpProgress(self)) def run(self): if sys.argv[1] == 'ffmpeg': self.installer = self.install_ffmpeg elif sys.argv[1] == 'dvdcss': self.installer = self.install_dvdcss else: QMessageBox.critical(self, _('Error'), _('Unknown codec type')) QApplication.exit(1) return if not self.installer(): QApplication.exit(1) return os.system('dcop kaffeine KaffeineIface reloadEngine') os.system('dcop kaffeine KaffeineIface play') QApplication.exit(0) def check_already_installed(self, pkg): if pkg and pkg.isInstalled: QMessageBox.critical(self, _('Error'), _('Codec package is already installed')) QApplication.exit(1) return True return False def install_ffmpeg(self): self.init_apt_cache() pkg = self.search_pkg('libxine','-ffmpeg') if self.check_already_installed(pkg): return False confirm = QMessageBox.question(self, _('Codec not found'), _('Kaffeine does not have a codec ' 'installed to play this file. ' 'Do you want to install the codecs?'), QMessageBox.Yes | QMessageBox.No) if (confirm == QMessageBox.No): QApplication.exit(1) return False self.setStatus(_('Installing...')) QApplication.processEvents() self.install_cache_package(pkg) return True def install_cache_package(self, pkg): if os.system('kdesu -d \'adept_batch install ' + pkg.name + '\'') != 0: QApplication.exit(1) return def search_pkg(self, prefix, suffix = ''): for pkg in self.apt_cache: if pkg.name.startswith(prefix) and pkg.name.endswith(suffix): return pkg return None def install_dvdcss(self): self.init_apt_cache() pkg = self.search_pkg('libdvdcss') if self.check_already_installed(pkg): return False answer = QMessageBox.question(self, _('Codec not found'), _('LibDVDCSS (http://www.videolan.org/developers/libdvdcss.html) ' 'is about to be installed, it allows you to watch encrypted DVD ' 'videos. This is illegal in some countries which ' 'require decryption to be authorised by the copyright holder. ' 'Do you want to install this package?'), QMessageBox.Yes | QMessageBox.No) if (answer == QMessageBox.No): QApplication.exit(1) return False self.setStatus(_('Installing...')) QApplication.processEvents() distro_arch = os.popen('dpkg --print-installation-architecture').read() distro_arch = distro_arch[:-1] # remove trailing newline libdvdcss_url = 'http://kubuntu.org/packages/libdvdcss-' + distro_arch + '.deb' os.system('kfmclient exec ' + libdvdcss_url) return True app = QApplication(sys.argv) window = CodecInstallerWidget() window.exec_()