318 lines
11 KiB
Python
318 lines
11 KiB
Python
#!/usr/bin/python3
|
|
|
|
## system-config-printer
|
|
|
|
## Copyright (C) 2010, 2011, 2012, 2013, 2014 Red Hat, Inc.
|
|
## Author: Tim Waugh <twaugh@redhat.com>
|
|
|
|
## 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; either version 2 of the License, or
|
|
## (at your option) any later version.
|
|
|
|
## 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
|
|
import dbus
|
|
from gi.repository import GObject
|
|
from gi.repository import Gtk
|
|
import cupshelpers
|
|
|
|
import cups
|
|
cups.require ("1.9.52")
|
|
|
|
import asyncconn
|
|
from debug import debugprint
|
|
import config
|
|
import gettext
|
|
gettext.install(domain=config.PACKAGE, localedir=config.localedir)
|
|
|
|
class PPDsLoader(GObject.GObject):
|
|
"""
|
|
1. If PackageKit support is available, and this is a local server,
|
|
try to use PackageKit to install relevant drivers. We do this
|
|
because we can only make the right choice about the "best" driver
|
|
when the full complement of drivers is there to choose from.
|
|
|
|
2. Fetch the list of available drivers from CUPS.
|
|
|
|
3. If Jockey is available, and there is no appropriate driver
|
|
available, try to use Jockey to install one.
|
|
|
|
4. If Jockey was able to install one, fetch the list of available
|
|
drivers again.
|
|
"""
|
|
|
|
__gsignals__ = {
|
|
'finished': (GObject.SignalFlags.RUN_LAST, None, ())
|
|
}
|
|
|
|
def __init__ (self, device_id=None, parent=None, device_uri=None,
|
|
host=None, encryption=None, language=None,
|
|
device_make_and_model=None):
|
|
GObject.GObject.__init__ (self)
|
|
debugprint ("+%s" % self)
|
|
self._device_id = device_id
|
|
self._device_uri = device_uri
|
|
self._device_make_and_model = device_make_and_model
|
|
self._parent = parent
|
|
self._host = host
|
|
self._encryption = encryption
|
|
self._language = language
|
|
|
|
self._installed_files = []
|
|
self._conn = None
|
|
self._ppds = None
|
|
self._exc = None
|
|
|
|
self._ppdsmatch_result = None
|
|
self._jockey_queried = False
|
|
self._jockey_has_answered = False
|
|
self._local_cups = (self._host is None or
|
|
self._host == "localhost" or
|
|
self._host[0] == '/')
|
|
try:
|
|
self._bus = dbus.SessionBus ()
|
|
except:
|
|
debugprint ("Failed to get session bus")
|
|
self._bus = None
|
|
|
|
fmt = _("Searching")
|
|
self._dialog = Gtk.MessageDialog (parent=parent,
|
|
modal=True, destroy_with_parent=True,
|
|
message_type=Gtk.MessageType.INFO,
|
|
buttons=Gtk.ButtonsType.CANCEL,
|
|
text=fmt)
|
|
|
|
self._dialog.format_secondary_text (_("Searching for drivers"))
|
|
|
|
self._dialog.connect ("response", self._dialog_response)
|
|
|
|
def run (self):
|
|
self._dialog.show_all ()
|
|
|
|
if self._device_id:
|
|
self._devid_dict = cupshelpers.parseDeviceID (self._device_id)
|
|
|
|
if (self._local_cups and
|
|
self._device_id and
|
|
self._devid_dict["MFG"] and
|
|
self._devid_dict["MDL"] and
|
|
self._bus):
|
|
self._gpk_device_id = "MFG:%s;MDL:%s;" % (self._devid_dict["MFG"],
|
|
self._devid_dict["MDL"])
|
|
self._query_packagekit ()
|
|
else:
|
|
self._query_cups ()
|
|
|
|
def __del__ (self):
|
|
debugprint ("-%s" % self)
|
|
|
|
def destroy (self):
|
|
debugprint ("DESTROY: %s" % self)
|
|
if self._dialog:
|
|
self._dialog.destroy ()
|
|
self._dialog = None
|
|
|
|
self._parent = None
|
|
|
|
if self._conn:
|
|
self._conn.destroy ()
|
|
self._conn = None
|
|
|
|
def get_installed_files (self):
|
|
return self._installed_files
|
|
|
|
def get_ppds (self):
|
|
return self._ppds
|
|
|
|
def get_ppdsmatch_result (self):
|
|
return self._ppdsmatch_result
|
|
|
|
def get_error (self):
|
|
debugprint ("%s: stored error is %s" % (self, repr (self._exc)))
|
|
return self._exc
|
|
|
|
def _dialog_response (self, dialog, response):
|
|
dialog.destroy ()
|
|
self._dialog = None
|
|
self.emit ('finished')
|
|
|
|
def _query_cups (self):
|
|
debugprint ("Asking CUPS for PPDs")
|
|
if (not self._conn):
|
|
c = asyncconn.Connection (host=self._host,
|
|
encryption=self._encryption,
|
|
reply_handler=self._cups_connect_reply,
|
|
error_handler=self._cups_error)
|
|
self._conn = c
|
|
else:
|
|
self._cups_connect_reply(self._conn, None)
|
|
|
|
def _cups_connect_reply (self, conn, UNUSED):
|
|
conn._begin_operation (_("fetching PPDs"))
|
|
conn.getPPDs2 (reply_handler=self._cups_reply,
|
|
error_handler=self._cups_error)
|
|
|
|
def _cups_reply (self, conn, result):
|
|
ppds = cupshelpers.ppds.PPDs (result, language=self._language)
|
|
self._ppds = ppds
|
|
self._need_requery_cups = False
|
|
if self._device_id:
|
|
fit = ppds.\
|
|
getPPDNamesFromDeviceID (self._devid_dict["MFG"],
|
|
self._devid_dict["MDL"],
|
|
self._devid_dict["DES"],
|
|
self._devid_dict["CMD"],
|
|
self._device_uri,
|
|
self._device_make_and_model)
|
|
|
|
ppdnamelist = ppds.\
|
|
orderPPDNamesByPreference (list(fit.keys ()),
|
|
self._installed_files,
|
|
devid=self._devid_dict,
|
|
fit=fit)
|
|
self._ppdsmatch_result = (fit, ppdnamelist)
|
|
|
|
ppdname = ppdnamelist[0]
|
|
if (self._bus and
|
|
not fit[ppdname].startswith ("exact") and
|
|
not self._jockey_queried and
|
|
self._devid_dict["MFG"] and
|
|
self._devid_dict["MDL"] and
|
|
self._local_cups):
|
|
# Try to install packages using jockey if
|
|
# - there's no appropriate driver (PPD) locally available
|
|
# - we are configuring local CUPS server
|
|
self._jockey_queried = True
|
|
self._query_jockey ()
|
|
return
|
|
|
|
conn.destroy ()
|
|
self._conn = None
|
|
if self._dialog is not None:
|
|
self._dialog.destroy ()
|
|
self._dialog = None
|
|
|
|
self.emit ('finished')
|
|
|
|
def _cups_error (self, conn, exc):
|
|
conn.destroy ()
|
|
self._conn = None
|
|
self._ppds = None
|
|
self._exc = exc
|
|
if self._dialog is not None:
|
|
self._dialog.destroy ()
|
|
self._dialog = None
|
|
|
|
self.emit ('finished')
|
|
|
|
def _query_packagekit (self):
|
|
debugprint ("Asking PackageKit to install drivers")
|
|
try:
|
|
obj = self._bus.get_object ("org.freedesktop.PackageKit",
|
|
"/org/freedesktop/PackageKit")
|
|
proxy = dbus.Interface (obj, "org.freedesktop.PackageKit.Modify")
|
|
resources = [self._gpk_device_id]
|
|
interaction = "hide-finished"
|
|
debugprint ("Calling InstallPrinterDrivers (%s, %s, %s)" %
|
|
(repr (0), repr (resources), repr (interaction)))
|
|
proxy.InstallPrinterDrivers (dbus.UInt32 (0),
|
|
resources, interaction,
|
|
reply_handler=self._packagekit_reply,
|
|
error_handler=self._packagekit_error,
|
|
timeout=3600)
|
|
except Exception as e:
|
|
debugprint ("Failed to talk to PackageKit: %s" % repr (e))
|
|
if self._dialog:
|
|
self._dialog.show_all ()
|
|
self._query_cups ()
|
|
|
|
def _packagekit_reply (self):
|
|
debugprint ("Got PackageKit reply")
|
|
self._need_requery_cups = True
|
|
if self._dialog:
|
|
self._dialog.show_all ()
|
|
self._query_cups ()
|
|
|
|
def _packagekit_error (self, exc):
|
|
debugprint ("Got PackageKit error: %s" % repr (exc))
|
|
if self._dialog:
|
|
self._dialog.show_all ()
|
|
self._query_cups ()
|
|
|
|
def _query_jockey (self):
|
|
debugprint ("Asking Jockey to install drivers")
|
|
try:
|
|
obj = self._bus.get_object ("com.ubuntu.DeviceDriver", "/GUI")
|
|
jockey = dbus.Interface (obj, "com.ubuntu.DeviceDriver")
|
|
r = jockey.search_driver ("printer_deviceid:%s" % self._device_id,
|
|
reply_handler=self._jockey_reply,
|
|
error_handler=self._jockey_error,
|
|
timeout=3600)
|
|
except Exception as e:
|
|
self._jockey_error (e)
|
|
|
|
def _jockey_reply (self, conn, result):
|
|
debugprint ("Got Jockey result: %s" % repr (result))
|
|
self._jockey_has_answered = True
|
|
try:
|
|
self._installed_files = result[1]
|
|
except:
|
|
self._installed_files = []
|
|
self._query_cups ()
|
|
|
|
def _jockey_error (self, exc):
|
|
debugprint ("Got Jockey error: %s" % repr (exc))
|
|
if self._need_requery_cups:
|
|
self._query_cups ()
|
|
else:
|
|
if self._conn is not None:
|
|
self._conn.destroy ()
|
|
self._conn = None
|
|
|
|
if self._dialog is not None:
|
|
self._dialog.destroy ()
|
|
self._dialog = None
|
|
|
|
self.emit ('finished')
|
|
|
|
if __name__ == "__main__":
|
|
class Foo:
|
|
def __init__ (self):
|
|
w = Gtk.Window ()
|
|
b = Gtk.Button.new_with_label ("Go")
|
|
w.add (b)
|
|
b.connect ('clicked', self.go)
|
|
w.connect ('delete-event', Gtk.main_quit)
|
|
w.show_all ()
|
|
self._window = w
|
|
|
|
def go (self, button):
|
|
loader = PPDsLoader (device_id="MFG:MFG;MDL:MDL;",
|
|
parent=self._window)
|
|
loader.connect ('finished', self.ppds_loaded)
|
|
loader.run ()
|
|
|
|
def ppds_loaded (self, ppdsloader):
|
|
self._window.destroy ()
|
|
Gtk.main_quit ()
|
|
exc = ppdsloader.get_error ()
|
|
print(exc)
|
|
ppds = ppdsloader.get_ppds ()
|
|
if ppds is not None:
|
|
print(len (ppds))
|
|
|
|
ppdsloader.destroy ()
|
|
|
|
from debug import set_debugging
|
|
set_debugging (True)
|
|
Foo ()
|
|
Gtk.main ()
|