Current File : //lib/python3/dist-packages/softwareproperties/gtk/SoftwarePropertiesGtk.py
# GTK+ based frontend to software-properties
#
# Copyright (c) 2004-2020 Canonical Ltd.
# 2004-2005 Michiel Sikkes
#
# Author: Michiel Sikkes <michiel@eyesopened.nl>
# Michael Vogt <mvo@debian.org>
# Sebastian Heinlein <glatzor@ubuntu.com>
# Andrea Azzarone <andrea.azzarone@canonical.com>
# Olivier Tilloy <olivier.tilloy@canonical.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., 59 Temple Place, Suite 330, Boston, MA 02111-1307
# USA
import apt_pkg
import datetime
import dbus
import dbus.glib
import dbus.mainloop.glib
from gettext import gettext as _
import gettext
import os
import subprocess
import logging
import threading
import sys
import time
import gi
gi.require_version("Gdk", "3.0")
gi.require_version("Gtk", "3.0")
gi.require_version("Handy", "1")
gi.require_version("PackageKitGlib", "1.0")
from gi.repository import PackageKitGlib as packagekit
from gi.repository import GObject, Gdk, Gtk, Gio, GLib, Handy
from aptsources.sourceslist import Deb822SourceEntry
from .SimpleGtkbuilderApp import SimpleGtkbuilderApp
from .DialogAdd import DialogAdd
from .DialogMirror import DialogMirror
from .DialogEdit import DialogEdit
from .DialogEditDeb822 import DialogEditDeb822
from .DialogCacheOutdated import DialogCacheOutdated
from .DialogAddSourcesList import DialogAddSourcesList
from .UbuntuProPage import UbuntuProPage
import softwareproperties
import softwareproperties.distro
from softwareproperties.SoftwareProperties import SoftwareProperties
import softwareproperties.SoftwareProperties
from softwareproperties.gtk.utils import (
get_ua_status,
get_ua_service_status,
current_distro,
is_current_distro_lts,
)
from UbuntuDrivers import detect
dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
if GLib.pyglib_version < (3, 9, 1):
GLib.threads_init()
(LIST_MARKUP, LIST_ENABLED, LIST_ENTRY_OBJ) = range(3)
(
COLUMN_ACTIVE,
COLUMN_DESC
) = list(range(2))
RESPONSE_REPLACE = 1
RESPONSE_ADD = 2
# columns of the source_store
(
STORE_ACTIVE,
STORE_DESCRIPTION,
STORE_SOURCE,
STORE_SEPARATOR,
STORE_VISIBLE
) = list(range(5))
# rows for the updates subscription combobox
(
UPDATES_ALL,
UPDATES_RECOMMENDED,
UPDATES_SECURITY,
UPDATES_CUSTOM,
) = range(4)
# names of the pockets
POCKET_SECURITY = "security"
POCKET_UPDATES = "updates"
POCKET_BACKPORTS = "backports"
class UpdateSubscriptions(object):
security = False
updates = False
backports = False
other = False
inconsistent = False
def get_status(self):
if self.other:
return UPDATES_CUSTOM
if self.security and self.updates and self.backports:
return UPDATES_ALL
elif self.security and self.updates and not self.backports:
return UPDATES_RECOMMENDED
elif self.security and not self.updates and not self.backports:
return UPDATES_SECURITY
return UPDATES_CUSTOM
def maybe_log_authentication_canceled_error(dbus_exception):
"""Log an error if the DBus exception was raised because the user canceled
the authentication prompt shown when changing apt sources/configuration.
Return True if it is, so that calling code can do further processing,
False otherwise.
"""
if dbus_exception._dbus_error_name == \
'com.ubuntu.SoftwareProperties.PermissionDeniedByPolicy':
logging.error("Authentication canceled, changes have not been saved")
return True
return False
def error(parent_window, summary, msg):
""" show a error dialog """
dialog = Gtk.MessageDialog(parent=parent_window,
flags=Gtk.DialogFlags.MODAL,
type=Gtk.MessageType.ERROR,
buttons=Gtk.ButtonsType.OK,
message_format=None)
dialog.set_markup("<big><b>%s</b></big>\n\n%s" % (summary, msg))
dialog.run()
dialog.destroy()
return False
class SoftwarePropertiesGtk(SoftwareProperties, SimpleGtkbuilderApp):
def __init__(self, datadir=None, options=None, file=None, parent=None):
""" Provide a GTK based graphical user interface to configure
the used software repositories, corresponding authentication keys
and update automation """
SoftwareProperties.__init__(self, options=options, datadir=datadir,
deb822=True)
Gtk.Window.set_default_icon_name("software-properties")
# Support GNOME 42 dark mode
Handy.init()
style_manager = Handy.StyleManager.get_default()
style_manager.set_color_scheme(Handy.ColorScheme.PREFER_LIGHT)
SimpleGtkbuilderApp.__init__(self,
os.path.join(datadir, "gtkbuilder", "main.ui"),
domain="software-properties")
if parent:
self.window_main.set_type_hint(Gdk.WindowTypeHint.DIALOG)
self.window_main.show()
try:
self.window_main.set_transient_for(parent)
except:
pass
# If externally called, reparent to external application.
self.options = options
if options and options.toplevel != None:
self.window_main.set_type_hint(Gdk.WindowTypeHint.DIALOG)
self.window_main.show()
try:
toplevel = Gdk.window_foreign_new(int(options.toplevel))
except AttributeError:
toplevel = None
if (toplevel):
try:
self.window_main.set_transient_for(toplevel)
except:
pass
# gsettings
all_schemas = Gio.Settings.list_schemas()
if "com.ubuntu.update-notifier" in all_schemas:
self.settings = Gio.Settings.new("com.ubuntu.update-notifier")
# we need this for reverting
self.initial_auto_launch = self.settings.get_int("regular-auto-launch-interval")
else:
self.settings = None
self.initial_auto_launch = 0
self.combobox_other_updates.set_sensitive(False)
# get the dbus backend
bus = dbus.SystemBus()
proxy = bus.get_object("com.ubuntu.SoftwareProperties", "/")
self.backend = dbus.Interface(proxy, "com.ubuntu.SoftwareProperties")
self.backend.connect_to_signal(
"SourcesListModified", self.on_sources_list_modified)
self.backend.connect_to_signal(
"ConfigModified", self.on_config_modified)
self.backend.connect_to_signal(
"KeysModified", self.on_keys_modified)
self.backend.connect_to_signal(
"AuthFailed", self.on_auth_failed)
self.backend.connect_to_signal(
"CdromScanFailed", self.on_cdrom_scan_failed)
# Reload dbus backend, for if there are any sources changed.
self.backend.Reload();
# Show what we have early
self.window_main.show()
# used to store the handlers of callbacks
self.handlers = {}
# Initialise and store the apt cache
self.init_apt_cache()
self.pk_task = None
# Put some life into the user interface:
self.init_auto_update()
self.init_release_upgrades()
self.show_auto_update_level()
# Setup the key list
self.init_keys()
self.show_keys()
# Setup the ISV sources list
self.init_isv_sources()
self.show_isv_sources()
self.show_cdrom_sources()
# Setup and show the distro elements
self.init_distro()
self.show_distro()
# Setup and show the Additonal Drivers tab
self.init_drivers()
# Setup and show the Ubuntu Pro tab if the serie is a LTS
if is_current_distro_lts():
self.init_ubuntu_pro()
# Connect to switch-page before setting initial tab. Otherwise the
# first switch goes unnoticed.
self.notebook_main.connect("switch-page", self.on_main_notebook_page_switched)
if options and options.open_tab:
self.notebook_main.set_current_page(int(options.open_tab))
# Show the import/replace sources.list dialog if a file different
# to the default sources.list was specified
# NOTE: If the file path points to the default sources.list the user
# perhaps assumed that s-p would act like a normal editor.
# We have got some bug reports from users calling
# "sudo software-properties-gtk /etc/apt/sources.list" from the
# command line.
if (file != None and
os.path.abspath(file) != "%s%s" % (apt_pkg.config.find_dir("Dir::Etc"),
apt_pkg.config.find("Dir::Etc::sourcelist"))):
self.open_file(file)
# Make sure window is as small as possible initially.
self.window_main.resize(1, 1)
def on_main_notebook_page_switched(self, notebook, page, page_num):
# On the additional drivers page, don't show the backend revert button.
if page == self.vbox_drivers:
self.button_revert.set_visible(False)
if not self.detect_called:
GLib.idle_add(lambda: threading.Thread(target=self.detect_drivers).start())
else:
self.button_revert.set_visible(True)
def update_interface(self):
""" abstract interface to keep the UI alive """
while Gtk.events_pending():
Gtk.main_iteration()
def init_release_upgrades(self):
" setup the widgets that allow configuring the release upgrades "
i = self.get_release_upgrades_policy()
self.combobox_release_upgrades.set_active(i)
self.handlers[self.combobox_release_upgrades] = \
self.combobox_release_upgrades.connect("changed",
self.on_combobox_release_upgrades_changed)
def init_apt_cache(self):
apt_pkg.init_config()
apt_pkg.init_system()
self.apt_cache = apt_pkg.Cache(None)
self.depcache = apt_pkg.DepCache(self.apt_cache)
self.records = apt_pkg.PackageRecords(self.apt_cache)
def update_apt_cache(self):
self.apt_cache = apt_pkg.Cache(None)
self.depcache = apt_pkg.DepCache(self.apt_cache)
self.records = apt_pkg.PackageRecords(self.apt_cache)
def init_auto_update(self):
""" Set up the widgets that allow to configure the update automation """
self.combobox_update_interval.show()
# normal updates
# this maps the key (combo-box-index) to the auto-update-interval value
# we build it dynamically from the model
model = self.combobox_update_interval.get_model()
self.combobox_interval_mapping = {}
for (i, row) in enumerate(model):
# column 1 is the update interval in days
value = model.get_value(row.iter, 1)
self.combobox_interval_mapping[i] = value
# normal updates
update_days = self.get_update_interval()
# If a custom period is defined add a corresponding entry
if not update_days in self.combobox_interval_mapping.values():
if update_days > 0:
self.combobox_update_interval.append_text(_("Every %s days")
% update_days)
self.combobox_interval_mapping[-1] = update_days
for key in self.combobox_interval_mapping:
if self.combobox_interval_mapping[key] == update_days:
self.combobox_update_interval.set_active(key)
break
self.handlers[self.combobox_update_interval] = \
self.combobox_update_interval.connect("changed",
self.on_combobox_update_interval_changed)
def show_auto_update_level(self):
"""Represent the level of update automation in the user interface"""
# Security Updates
self.set_security_update_level()
# Other Updates
if self.settings:
level_other = self.settings.get_int("regular-auto-launch-interval")
model = self.combobox_other_updates.get_model()
for (i, row) in enumerate(model):
level = model.get_value(row.iter, 1)
if level_other == level:
self.combobox_other_updates.set_active(i)
break
self.handlers[self.combobox_security_updates] = \
self.combobox_security_updates.connect("changed",
self.set_sec_update_automation_level)
self.handlers[self.combobox_other_updates] = \
self.combobox_other_updates.connect("changed",
self.set_other_update_automation_level)
def set_security_update_level(self):
"""Fetch the security level, Enable/Disable and set the value appropriately"""
# Security Updates
level_sec = self.get_update_automation_level()
if level_sec == None:
self.combobox_security_updates.set_sensitive(False)
else:
self.combobox_security_updates.set_sensitive(True)
if (level_sec == softwareproperties.UPDATE_MANUAL or
level_sec == softwareproperties.UPDATE_NOTIFY):
self.combobox_security_updates.set_active(0) # Display immediately
elif level_sec == softwareproperties.UPDATE_DOWNLOAD:
self.combobox_security_updates.set_active(1) # Download automatically
elif level_sec == softwareproperties.UPDATE_INST_SEC:
self.combobox_security_updates.set_active(2) # Download and install automatically
def init_distro(self):
"""Setup the user interface elements to represent the distro"""
# TRANS: %s stands for the distribution name e.g. Debian or Ubuntu
self.label_dist_name.set_label(_("%s Software") % self.distro.id)
self.handlers[self.checkbutton_source_code] = \
self.checkbutton_source_code.connect("toggled",
self.on_checkbutton_source_code_toggled)
# Setup the checkbuttons for the components
for checkbutton in self.vbox_dist_comps.get_children():
self.vbox_dist_comps.remove(checkbutton)
for comp in self.distro.source_template.components:
# TRANSLATORS: Label for the components in the Internet section
# first %s is the description of the component
# second %s is the code name of the comp, eg main, universe
label = _("%s (%s)") % (comp.get_description(), comp.name)
checkbox = Gtk.CheckButton(label=label)
checkbox.comp = comp
# setup the callback and show the checkbutton
self.handlers[checkbox] = checkbox.connect("toggled",
self.on_component_toggled,
comp.name)
self.vbox_dist_comps.add(checkbox)
checkbox.show()
# Setup the checkbuttons for the child repos
for checkbutton in self.dev_box.get_children():
self.dev_box.remove(checkbutton)
for template in self.distro.source_template.children:
# Do not show source entries in there
if template.type == "deb-src":
continue
if "proposed" in template.name:
checkbox = Gtk.CheckButton(label="%s (%s)" % (template.description,
template.name))
checkbox.template = template
self.handlers[checkbox] = \
checkbox.connect("toggled",
self.on_checkbutton_child_toggled,
template)
self.dev_box.add(checkbox)
checkbox.show()
status = get_ua_status()
if not is_current_distro_lts():
esm_available = False
esm_enabled = False
else:
(infra_available, infra_status) = get_ua_service_status("esm-infra", status=status)
(apps_available, apps_status) = get_ua_service_status("esm-apps", status=status)
esm_available = bool(infra_available or apps_available)
esm_enabled = "enabled" in (infra_status, apps_status)
distro = current_distro()
if esm_enabled:
eol_text = _("Extended Security Maintenance")
# EOL date should probably be UA contract expiry.
# This is probably sooner than ESM EOL for the distro and
# gives software properties dialogs a chance to interact about
# renewals if needed.
try:
# Ignore timezone to simplify formatting python < 3.7
# 3.7 has datetime.fromisoformat()
dt, _sep, _tz = status.get("expires", "").partition("+")
eol_date = datetime.datetime.strptime(
dt, "%Y-%m-%dT%H:%M:%S"
).date()
except ValueError:
print("Unable to determine UA contract expiry")
eol_date = distro.eol
else:
eol_text = _("Basic Security Maintenance")
eol_date = distro.eol
self.label_esm_status.set_markup(eol_text)
esm_url = "https://ubuntu.com/esm" # Non-EOL LTS generic ESM
today = datetime.datetime.now().date()
if today >= eol_date:
if esm_available:
# EOL LTS uses release-specific ESM ubuntu.com/XX-YY
distro_ver = distro.version.replace(' LTS', '')
esm_url = "https://ubuntu.com/%s" % distro_ver.replace(".", "-")
eol_expiry_text = _("Ended %s - extend or upgrade now") % eol_date.strftime("%x")
elif today >= eol_date - datetime.timedelta(days=60):
eol_expiry_text = _("Ends %s - extend or upgrade soon") % eol_date.strftime("%x")
else:
eol_expiry_text = _("Active until %s") % eol_date.strftime("%x")
self.label_eol.set_label(eol_expiry_text)
self.label_esm_subscribe.set_markup(
"<a href=\"%s\">%s</a>" % (esm_url, _("Extend…"))
)
self.label_esm_subscribe.set_visible(
esm_available and not esm_enabled
)
eol_expiry_text = _("Ended %s") % eol_date.strftime("%x")
# Setup the combo box for updates subscriptions
combobox = self.combobox_updates_subscription
self.handlers[combobox] = combobox.connect(
"changed", self.on_combobox_updates_subscription_changed)
# setup the server chooser
cell = Gtk.CellRendererText()
self.combobox_server.pack_start(cell, True)
self.combobox_server.add_attribute(cell, 'text', 0)
self.handlers[self.combobox_server] = self.combobox_server.connect("changed",
self.on_combobox_server_changed)
server_store = Gtk.ListStore(GObject.TYPE_STRING,
GObject.TYPE_STRING,
GObject.TYPE_BOOLEAN)
self.combobox_server.set_model(server_store)
self.combobox_server.set_row_separator_func(self.is_row_separator, 2)
def block_handlers(self):
for widget in self.handlers:
if widget.handler_is_connected(self.handlers[widget]):
widget.handler_block(self.handlers[widget])
def unblock_handlers(self):
for widget in self.handlers:
if widget.handler_is_connected(self.handlers[widget]):
widget.handler_unblock(self.handlers[widget])
def show_distro(self):
"""Fill the distro user interface with life"""
self.block_handlers()
# Enable or disable the child source checkbuttons
for checkbox in self.dev_box.get_children():
(active, inconsistent) = self.get_comp_child_state(checkbox.template)
checkbox.set_active(active)
checkbox.set_inconsistent(inconsistent)
# Enable or disable the component checkbuttons
for checkbox in self.vbox_dist_comps.get_children():
# check if the comp is enabled
(active, inconsistent) = self.get_comp_download_state(checkbox.comp)
checkbox.set_inconsistent(inconsistent)
checkbox.set_active(active)
# Refresh the combo box for updates subscriptions
subscriptions = UpdateSubscriptions()
for template in self.distro.source_template.children:
# Do not show source entries in there
if template.type == "deb-src":
continue
pocket = template.name.split('-')[-1]
(active, inconsistent) = self.get_comp_child_state(template)
subscriptions.inconsistent |= inconsistent
if active:
if pocket == POCKET_SECURITY:
subscriptions.security = True
elif pocket == POCKET_UPDATES:
subscriptions.updates = True
elif pocket == POCKET_BACKPORTS:
subscriptions.backports = True
else:
subscriptions.other = True
status = subscriptions.get_status()
combobox = self.combobox_updates_subscription
model = combobox.get_model()
if status == UPDATES_CUSTOM:
try:
model.get_iter_from_string(str(UPDATES_CUSTOM))
except ValueError:
model.append([_("Custom")])
else:
try:
model.remove(model.get_iter_from_string(str(UPDATES_CUSTOM)))
except ValueError:
pass
combobox.set_active(status)
combobox.set_sensitive(not subscriptions.inconsistent)
# If no components are enabled there will be no need for updates
# and source code
if len(self.distro.enabled_comps) < 1:
self.combobox_updates_subscription.set_sensitive(False)
self.dev_box.set_sensitive(False)
self.checkbutton_source_code.set_sensitive(False)
else:
self.dev_box.set_sensitive(True)
self.checkbutton_source_code.set_sensitive(True)
# Check for source code sources
source_code_state = self.get_source_code_state()
if source_code_state == None:
self.checkbutton_source_code.set_inconsistent(True)
elif source_code_state == True:
self.checkbutton_source_code.set_active(True)
self.checkbutton_source_code.set_inconsistent(False)
else:
self.checkbutton_source_code.set_active(False)
self.checkbutton_source_code.set_inconsistent(False)
# Will show a short explanation if no CDROMs are used
if len(self.get_cdrom_sources()) == 0:
self.scrolledwindow_cd.hide()
self.scrolledwindow_no_cd.show()
else:
self.scrolledwindow_cd.show()
self.scrolledwindow_no_cd.hide()
# provide a list of mirrors
server_store = self.combobox_server.get_model()
server_store.clear()
seen_server_new = []
for (name, uri, active) in self.distro.get_server_list():
server_store.append([name, uri, False])
if [name, uri] in self.seen_server:
self.seen_server.remove([name, uri])
elif uri != None:
seen_server_new.append([name, uri])
if active == True:
self.active_server = len(server_store) - 1
self.combobox_server.set_active(self.active_server)
for [name, uri] in self.seen_server:
server_store.append([name, uri, False])
self.seen_server = seen_server_new
# only add a separator and the option to choose another mirror from
# the list if a mirror set exists, it may not for some architectures
if self.distro.source_template.mirror_set:
server_store.append(["sep", None, True])
server_store.append([_("Other..."), None, False])
# make the interface respond to user interput again
self.unblock_handlers()
# Output a lot of debug stuff
if self.options.debug == True or self.options.massive_debug == True:
print("ENABLED COMPS: %s" % self.distro.enabled_comps)
print("INTERNET COMPS: %s" % self.distro.download_comps)
print("MAIN SOURCES")
for source in self.distro.main_sources:
self.print_source_entry(source)
print("CHILD SOURCES")
for source in self.distro.child_sources:
self.print_source_entry(source)
print("CDROM SOURCES")
for source in self.distro.cdrom_sources:
self.print_source_entry(source)
print("SOURCE CODE SOURCES")
for source in self.distro.source_code_sources:
self.print_source_entry(source)
print("DISABLED SOURCES")
for source in self.distro.disabled_sources:
self.print_source_entry(source)
print("ISV")
for source in self.sourceslist_visible:
self.print_source_entry(source)
def set_sec_update_automation_level(self, widget):
"""Call the backend to set the security update automation level to the given
value"""
index = widget.get_active()
state = -1
if index == 0: # Display immediately
state = softwareproperties.UPDATE_NOTIFY
elif index == 1: # Download automatically
state = softwareproperties.UPDATE_DOWNLOAD
elif index == 2: # Download and install automatically
state = softwareproperties.UPDATE_INST_SEC
# only set if something actually changed
try:
if state != self.get_update_automation_level():
self.backend.SetUpdateAutomationLevel(state)
except dbus.DBusException as e:
if maybe_log_authentication_canceled_error(e):
combo_handler = self.handlers[self.combobox_security_updates]
self.combobox_security_updates.handler_block(combo_handler)
self.set_security_update_level()
self.combobox_security_updates.handler_unblock(combo_handler)
def set_other_update_automation_level(self, widget):
"""Set the other update automation level to the given value via gconf"""
index = widget.get_active()
model = self.combobox_other_updates.get_model()
# the second column is the update interval days
days = model[index][1]
self.settings.set_int("regular-auto-launch-interval", days)
def is_row_separator(self, model, iter, column=0):
''' Check if a given row is a separator '''
return model.get_value(iter, column)
def on_combobox_release_upgrades_changed(self, combobox):
""" set the release upgrades policy """
#print("on_combobox_release_upgrades_changed()")
i = combobox.get_active()
try:
self.backend.SetReleaseUpgradesPolicy(i)
except dbus.DBusException as e:
if maybe_log_authentication_canceled_error(e):
combo_handler = self.handlers[self.combobox_release_upgrades]
self.combobox_release_upgrades.handler_block(combo_handler)
i = self.get_release_upgrades_policy()
self.combobox_release_upgrades.set_active(i)
self.combobox_release_upgrades.handler_unblock(combo_handler)
def on_combobox_updates_subscription_changed(self, combobox):
index = combobox.get_active()
if index != UPDATES_CUSTOM:
model = combobox.get_model()
try:
model.remove(model.get_iter_from_string(str(UPDATES_CUSTOM)))
except ValueError:
pass
def _toggle_child_source(pocket, enabled):
source = "{}-{}".format(self.distro.codename, pocket)
if enabled:
logging.info("enabling {}".format(source))
self.backend.EnableChildSource(source)
else:
logging.info("disabling {}".format(source))
self.backend.DisableChildSource(source)
try:
if index == UPDATES_ALL:
_toggle_child_source(POCKET_SECURITY, True)
_toggle_child_source(POCKET_UPDATES, True)
_toggle_child_source(POCKET_BACKPORTS, True)
elif index == UPDATES_RECOMMENDED:
_toggle_child_source(POCKET_SECURITY, True)
_toggle_child_source(POCKET_UPDATES, True)
_toggle_child_source(POCKET_BACKPORTS, False)
elif index == UPDATES_SECURITY:
_toggle_child_source(POCKET_SECURITY, True)
_toggle_child_source(POCKET_UPDATES, False)
_toggle_child_source(POCKET_BACKPORTS, False)
for template in self.distro.source_template.children:
if template.type == "deb-src":
continue
pocket = template.name.split('-')[-1]
if pocket not in (POCKET_SECURITY, POCKET_UPDATES, POCKET_BACKPORTS):
_toggle_child_source(pocket, False)
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def on_combobox_server_changed(self, combobox):
"""
Replace the servers used by the main and update sources with
the selected one
"""
if combobox.get_active() == self.active_server:
return
server_store = combobox.get_model()
iter = combobox.get_active_iter()
uri = server_store.get_value(iter, 1)
name = server_store.get_value(iter, 0)
try:
if name == _("Other..."):
dialog = DialogMirror(self.window_main,
self.datadir,
self.distro,
self.custom_mirrors)
res = dialog.run()
if res != None:
self.backend.ChangeMainDownloadServer(res)
else:
combobox.set_active(self.active_server)
elif uri != None and len(self.distro.used_servers) > 0:
self.active_server = combobox.get_active()
self.backend.ChangeMainDownloadServer(uri)
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
# mvo: is this still needed?
#else:
# self.distro.default_server = uri
def on_component_toggled(self, checkbutton, comp):
"""
Sync the components of all main sources (excluding cdroms),
child sources and source code sources
"""
try:
if checkbutton.get_active() == True:
self.backend.EnableComponent(comp)
else:
self.backend.DisableComponent(comp)
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def on_checkbutton_child_toggled(self, checkbutton, template):
"""
Enable or disable a child repo of the distribution main repository
"""
try:
if checkbutton.get_active() == False:
self.backend.DisableChildSource(template.name)
else:
self.backend.EnableChildSource(template.name)
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def on_checkbutton_source_code_toggled(self, checkbutton):
""" Disable or enable the source code for all sources """
try:
if checkbutton.get_active() == True:
self.backend.EnableSourceCodeSources()
else:
self.backend.DisableSourceCodeSources()
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def on_checkbutton_popcon_toggled(self, widget):
""" The user clicked on the popcon paritipcation button """
# only trigger the backend if something actually changed
do_popcon = self.get_popcon_participation()
if widget.get_active() != do_popcon:
try:
self.backend.SetPopconPariticipation(widget.get_active())
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def open_file(self, file):
"""Show a confirmation for adding the channels of the specified file"""
try:
dialog = DialogAddSourcesList(self.window_main,
self.sourceslist,
self.render_source,
self.get_comparable,
self.datadir,
file)
(res, new_sources) = dialog.run()
if res == RESPONSE_REPLACE:
self.sourceslist.list = []
if res in (RESPONSE_ADD, RESPONSE_REPLACE):
for source in new_sources:
self.backend.AddSourceFromLine(str(source))
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def on_sources_drag_data_received(self, widget, context, x, y,
selection, target_type, timestamp):
"""Extract the dropped file pathes and open the first file, only"""
uri = selection.data.strip()
uri_splitted = uri.split()
if len(uri_splitted) > 0:
self.open_file(uri_splitted[0])
def hide(self):
self.window_main.hide()
def source_search_function(self, store, column, key, iter, data=None):
# this needs to return False on a match
return store.get_value(iter, COLUMN_DESC).lower().find(key.lower()) == -1
def init_isv_sources(self):
"""
Read all valid sources into our ListStore
"""
# STORE_ACTIVE - is the source enabled or disabled
# STORE_DESCRIPTION - description of the source entry
# STORE_SOURCE - the source entry object
# STORE_SEPARATOR - if the entry is a separator
# STORE_VISIBLE - if the entry is shown or hidden
self.cdrom_store = Gtk.ListStore(GObject.TYPE_BOOLEAN,
GObject.TYPE_STRING,
GObject.TYPE_PYOBJECT,
GObject.TYPE_BOOLEAN,
GObject.TYPE_BOOLEAN)
self.treeview_cdroms.set_model(self.cdrom_store)
self.source_store = Gtk.ListStore(GObject.TYPE_BOOLEAN,
GObject.TYPE_STRING,
GObject.TYPE_PYOBJECT,
GObject.TYPE_BOOLEAN,
GObject.TYPE_BOOLEAN)
self.treeview_sources.set_model(self.source_store)
self.treeview_sources.set_row_separator_func(self.is_separator,
STORE_SEPARATOR)
cell_desc = Gtk.CellRendererText()
cell_desc.set_property("xpad", 2)
cell_desc.set_property("ypad", 2)
col_desc = Gtk.TreeViewColumn(_("Software Sources"), cell_desc,
markup=COLUMN_DESC)
col_desc.set_max_width(1000)
cell_toggle = Gtk.CellRendererToggle()
cell_toggle.set_property("xpad", 2)
cell_toggle.set_property("ypad", 2)
self.handlers[cell_toggle] = cell_toggle.connect('toggled',
self.on_isv_source_toggled,
self.cdrom_store)
col_active = Gtk.TreeViewColumn(_("Active"), cell_toggle,
active=COLUMN_ACTIVE)
self.treeview_cdroms.append_column(col_active)
self.treeview_cdroms.append_column(col_desc)
cell_desc = Gtk.CellRendererText()
cell_desc.set_property("xpad", 2)
cell_desc.set_property("ypad", 2)
col_desc = Gtk.TreeViewColumn(_("Software Sources"), cell_desc,
markup=COLUMN_DESC)
col_desc.set_max_width(1000)
cell_toggle = Gtk.CellRendererToggle()
cell_toggle.set_property("xpad", 2)
cell_toggle.set_property("ypad", 2)
self.handlers[cell_toggle] = cell_toggle.connect('toggled',
self.on_isv_source_toggled,
self.source_store)
col_active = Gtk.TreeViewColumn(_("Active"), cell_toggle,
active=COLUMN_ACTIVE)
self.treeview_sources.append_column(col_active)
self.treeview_sources.append_column(col_desc)
# typeahead support for the sources list using case insensitive matching
self.treeview_sources.set_search_column(COLUMN_DESC)
self.treeview_sources.set_search_equal_func(self.source_search_function, None)
# drag and drop support for sources.list
try:
Gtk.drag_dest_set(self.treeview_sources, Gtk.DestDefaults.ALL,
[Gtk.TargetEntry.new('text/uri-list', 0, 0)],
Gdk.DragAction.COPY)
self.treeview_sources.connect("drag_data_received",
self.on_sources_drag_data_received)
except AttributeError:
# does not work with Gtk2/GI
pass
def on_isv_source_activate(self, treeview, path, column):
"""Open the edit dialog if a channel was double clicked"""
self.on_edit_clicked(treeview)
def on_isv_source_toggled(self, cell_toggle, path, store):
"""Enable or disable the selected channel"""
#FIXME Gtk is doing something that takes a lock while processing
# the events, the polkit auth dialog can't be displayed in return,
# we don't have a proper fix and it might require GTK changes,
# meanwhile we want a working interface so workaround with a sleep
# https://launchpad.net/bugs/1727908
time.sleep(0.3)
#FIXME cdroms need to disable the comps in the childs and sources
iter = store.get_iter((int(path),))
source_entry = store.get_value(iter, STORE_SOURCE)
try:
self.backend.ToggleSourceUse(str(source_entry))
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def init_keys(self):
"""Setup the user interface parts needed for the key handling"""
self.keys_store = Gtk.ListStore(str)
self.treeview_auth.set_model(self.keys_store)
tr = Gtk.CellRendererText()
keys_col = Gtk.TreeViewColumn(_("Key"), tr, text=0)
self.treeview_auth.append_column(keys_col)
try:
self.treeview_auth.enable_model_drag_dest(
[('text/plain', 0, 0)], Gdk.DragAction.COPY)
self.treeview_auth.connect("drag_data_received",
self.on_auth_drag_data_received)
except AttributeError:
# Does not work with GTK 2/GI
pass
self.treeview_auth.connect("button-press-event",
self.show_auth_context_menu)
def show_auth_context_menu(self, widget, event):
if event.type == Gdk.EventType.BUTTON_PRESS and event.button == 3:
menu = Gtk.Menu()
item_paste = Gtk.MenuItem(label=_("_Add key from paste data"))
item_paste.connect("activate", self.on_auth_add_key_from_paste)
menu.append(item_paste)
menu.show_all()
menu.popup(None, None, None, None, event.button, event.time)
return True
def on_auth_add_key_from_paste(self, widget):
keydata = Gtk.Clipboard().wait_for_text()
if not keydata:
return
if not self.add_key_from_data(keydata):
error(self.window_main,
_("Error importing key"),
_("The selected data may not be a GPG key file "
"or it might be corrupt."))
self.show_keys()
def on_auth_drag_data_received(self, widget, context, x, y,
selection, target_type, timestamp):
"""Extract the dropped key and add it to the keyring"""
keydata = selection.get_data().strip()
if not self.add_key_from_data(keydata):
error(self.window_main,
_("Error importing key"),
_("The selected data may not be a GPG key file "
"or it might be corrupt."))
self.show_keys()
def add_key_from_data(self, keydata):
return self.backend.AddKeyFromData(keydata)
def on_button_revert_clicked(self, button):
"""Restore the source list from the startup of the dialog"""
try:
self.backend.Revert()
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
if self.settings:
self.settings.set_int("regular-auto-launch-interval", self.initial_auto_launch)
self.show_auto_update_level()
self.button_revert.set_sensitive(False)
self.modified_sourceslist = False
# dbus events
def on_config_modified(self):
"""The config was changed and now needs to be saved and reloaded"""
apt_pkg.init_config()
self.button_revert.set_sensitive(True)
def on_keys_modified(self):
""" The apt keys have changed and need to be redisplayed """
self.show_keys()
def on_sources_list_modified(self):
"""The sources list was changed and now needs to be saved and reloaded"""
self.reload_sourceslist()
self.show_distro()
self.show_isv_sources()
self.show_cdrom_sources()
self.button_revert.set_sensitive(True)
self.modified_sourceslist = True
def on_auth_failed(self):
""" send when the authentication failed """
# reread the current config
self.on_sources_list_modified()
self.on_config_modified()
self.on_keys_modified()
def on_cdrom_scan_failed(self):
error(self.window_main,
_("Error scanning the CD"),
_("Could not find a suitable CD."))
# helpers
def show_isv_sources(self):
""" Show the repositories of independent software vendors in the
third-party software tree view """
self.source_store.clear()
for source in self.get_isv_sources():
contents = self.render_source(source)
self.source_store.append([not source.disabled, contents,
source, False, True])
(path_x, path_y) = self.treeview_sources.get_cursor()
if len(self.source_store) < 1 or path_x is None:
self.button_remove.set_sensitive(False)
self.button_edit.set_sensitive(False)
def show_cdrom_sources(self):
""" Show CD-ROM/DVD based repositories of the currently used distro in
the CDROM based sources list """
self.cdrom_store.clear()
for source in self.get_cdrom_sources():
contents = self.render_source(source)
self.cdrom_store.append([not source.disabled, contents,
source, False, True])
def is_separator(self, model, iter, column):
""" Return true if the selected row is a separator """
try:
return model.get_value(iter, column)
except Exception as e:
print("is_seperator returned '%s' " % e)
return False
def show_keys(self):
self.keys_store.clear()
for key in self.apt_key.list():
self.keys_store.append([key])
def on_combobox_update_interval_changed(self, widget):
"""Set the update automation interval to the chosen one"""
i = self.combobox_update_interval.get_active()
if i != -1:
value = self.combobox_interval_mapping[i]
try:
self.backend.SetUpdateInterval(value)
except dbus.DBusException as e:
if maybe_log_authentication_canceled_error(e):
update_days = self.get_update_interval()
combo_handler = self.handlers[self.combobox_update_interval]
for key in self.combobox_interval_mapping:
if self.combobox_interval_mapping[key] == update_days:
self.combobox_update_interval.handler_block(combo_handler)
self.combobox_update_interval.set_active(key)
self.combobox_update_interval.handler_unblock(combo_handler)
break
def on_add_clicked(self, widget):
"""Show a dialog that allows to enter the apt line of a to be used repo"""
dialog = DialogAdd(self.window_main, self.sourceslist,
self.datadir, self.distro)
line = dialog.run()
if line != None:
try:
self.backend.AddSourceFromLine(line)
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def on_edit_clicked(self, widget):
"""Show a dialog to edit an ISV source"""
sel = self.treeview_sources.get_selection()
(model, iter) = sel.get_selected()
if not iter:
return
old_source_entry = model.get_value(iter, LIST_ENTRY_OBJ)
if isinstance(old_source_entry, Deb822SourceEntry):
dialog = DialogEditDeb822(self.window_main, self.sourceslist,
old_source_entry, self.datadir)
else:
dialog = DialogEdit(self.window_main, self.sourceslist,
old_source_entry, self.datadir)
if dialog.run() == Gtk.ResponseType.OK:
try:
self.backend.ReplaceSourceEntry(str(old_source_entry),
str(dialog.new_source_entry))
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
# FIXME:outstanding from merge
def on_isv_source_activated(self, treeview, path, column):
"""Open the edit dialog if a channel was double clicked"""
# check if the channel can be edited
if self.button_edit.get_property("sensitive") == True:
self.on_edit_clicked(treeview)
def on_treeview_sources_cursor_changed(self, treeview):
"""set the sensitiveness of the edit and remove button
corresponding to the selected channel"""
sel = self.treeview_sources.get_selection()
if not sel:
return
(model, iter) = sel.get_selected()
if not iter:
# No channel is selected, so disable edit and remove
self.button_edit.set_sensitive(False)
self.button_remove.set_sensitive(False)
return
# allow to remove the selected channel
self.button_remove.set_sensitive(True)
# disable editing of cdrom sources
source_entry = model.get_value(iter, LIST_ENTRY_OBJ)
if source_entry.uri.startswith("cdrom:"):
self.button_edit.set_sensitive(False)
else:
self.button_edit.set_sensitive(True)
def on_remove_clicked(self, widget):
"""Remove the selected source"""
model = self.treeview_sources.get_model()
(path, column) = self.treeview_sources.get_cursor()
iter = model.get_iter(path)
if iter:
source_entry = model.get_value(iter, LIST_ENTRY_OBJ)
try:
self.backend.RemoveSource(str(source_entry))
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def add_key_clicked(self, widget):
"""Provide a file chooser that allows to add the gnupg of a trusted
software vendor"""
chooser = Gtk.FileChooserDialog(title=_("Import key"),
parent=self.window_main,
buttons=(Gtk.STOCK_CANCEL,
Gtk.ResponseType.REJECT,
Gtk.STOCK_OK, Gtk.ResponseType.ACCEPT))
if "SUDO_USER" in os.environ:
home = os.path.expanduser("~%s" % os.environ["SUDO_USER"])
chooser.set_current_folder(home)
res = chooser.run()
chooser.hide()
if res == Gtk.ResponseType.ACCEPT:
try:
if not self.backend.AddKey(chooser.get_filename()):
error(self.window_main,
_("Error importing selected file"),
_("The selected file may not be a GPG key file "
"or it might be corrupt."))
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def remove_key_clicked(self, widget):
"""Remove a trusted software vendor key"""
selection = self.treeview_auth.get_selection()
(model, a_iter) = selection.get_selected()
if a_iter == None:
return
key = model.get_value(a_iter, 0)
try:
if not self.backend.RemoveKey(key[:16]):
error(self.main,
_("Error removing the key"),
_("The key you selected could not be removed. "
"Please report this as a bug."))
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def on_restore_clicked(self, widget):
"""Restore the original keys"""
try:
self.backend.UpdateKeys()
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def on_delete_event(self, widget, args):
"""Close the window if requested"""
self.on_close_button(widget)
return False
def on_close_button(self, widget):
"""Show a dialog that a reload of the channel information is required
only if there is no parent defined"""
if (self.modified_sourceslist == True and
self.options.no_update == False):
d = DialogCacheOutdated(self.window_main,
self.datadir)
d.run()
self.quit()
def on_button_add_cdrom_clicked(self, widget):
""" when a cdrom is requested for adding """
try:
self.backend.AddCdromSource()
except dbus.DBusException as e:
maybe_log_authentication_canceled_error(e)
def on_driver_changes_progress(self, progress, ptype, data=None):
#print(progress)
self.button_driver_revert.set_visible(False)
self.button_driver_apply.set_visible(False)
self.button_driver_restart.set_visible(False)
self.button_driver_cancel.set_visible(True)
self.progress_bar.set_visible(True)
self.progress_bar.set_visible(True)
self.label_driver_action.set_label(_("Applying changes..."))
if ptype == packagekit.ProgressType.PERCENTAGE:
prog_value = progress.get_property('percentage')
self.progress_bar.set_fraction(prog_value / 100.0)
def on_driver_changes_finish(self, source, result, installs_pending):
try:
self.pk_task.generic_finish(result)
except Exception as e:
self.on_driver_changes_revert()
error(self.window_main, _("Error while applying changes"), str(e))
if not installs_pending:
self.progress_bar.set_visible(False)
self.clear_changes()
self.update_apt_cache()
self.set_driver_action_status()
self.update_label_and_icons_from_status()
self.button_driver_revert.set_visible(True)
self.button_driver_apply.set_visible(True)
self.button_driver_cancel.set_visible(False)
self.scrolled_window_drivers.set_sensitive(True)
def on_driver_changes_apply(self, button):
self.pk_task = packagekit.Task()
installs = []
removals = []
has_nvidia = False
for pkg in self.driver_changes:
if pkg.current_ver:
removals.append(self.get_package_id(self.apt_cache, pkg))
# The main NVIDIA package is only a metapackage.
# We need to collect its dependencies, so that
# we can uninstall the driver properly.
if 'nvidia' in pkg.name:
has_nvidia = True
for dep in self.get_dependencies(self.apt_cache, pkg, 'nvidia'):
if dep.current_ver:
removals.append(self.get_package_id(self.apt_cache, dep))
else:
installs.append(self.get_package_id(self.apt_cache, pkg))
self.cancellable = Gio.Cancellable()
try:
if removals:
installs_pending = False
if installs:
installs_pending = True
self.pk_task.remove_packages_async(removals,
False, # allow deps
True, # autoremove
self.cancellable, # cancellable
self.on_driver_changes_progress,
(None, ), # progress data
self.on_driver_changes_finish, # callback ready
installs_pending # callback data
)
if installs:
if has_nvidia:
to_install = []
for item in installs:
name = item.split(';')[0]
to_install.append(name)
self.pk_task.install_packages_async(installs,
self.cancellable, # cancellable
self.on_driver_changes_progress,
(None, ), # progress data
self.on_driver_changes_finish, # GAsyncReadyCallback
False # ready data
)
self.button_driver_revert.set_sensitive(False)
self.button_driver_apply.set_sensitive(False)
self.scrolled_window_drivers.set_sensitive(False)
except Exception as e:
print("Warning: install transaction not completed successfully: {}".format(e))
def on_driver_changes_revert(self, button_revert=None):
# HACK: set all the "Do not use" first; then go through the list of the
# actually selected drivers.
for button in self.no_drv:
button.set_active(True)
for alias in self.orig_selection:
button = self.orig_selection[alias]
button.set_active(True)
self.clear_changes()
self.button_driver_revert.set_sensitive(False)
self.button_driver_apply.set_sensitive(False)
def on_driver_changes_cancel(self, button_cancel):
self.cancellable.cancel()
self.clear_changes()
def on_driver_restart_clicked(self, button_restart):
if 'XDG_CURRENT_DESKTOP' in os.environ:
desktop = os.environ['XDG_CURRENT_DESKTOP']
else:
desktop = 'Unknown'
if (desktop == 'ubuntu:GNOME' and
os.path.exists('/usr/bin/gnome-session-quit')):
# argument presents a dialog to cancel reboot
subprocess.call(['gnome-session-quit', '--reboot'])
elif (desktop == 'XFCE' and
os.path.exists('/usr/bin/xfce4-session-logout')):
subprocess.call(['xfce4-session-logout'])
elif (desktop == 'LXDE' and
os.path.exists('/usr/bin/lubuntu-logout')):
subprocess.call(['lubuntu-logout'])
elif (desktop == 'LXQt' and
os.path.exists('/usr/bin/lxqt-leave')):
subprocess.call(['lxqt-leave'])
def clear_changes(self):
self.orig_selection = {}
self.driver_changes = []
def init_drivers(self):
"""Additional Drivers tab"""
self.button_driver_revert = Gtk.Button(label=_("Re_vert"), use_underline=True)
self.button_driver_revert.connect("clicked", self.on_driver_changes_revert)
self.button_driver_apply = Gtk.Button(label=_("_Apply Changes"), use_underline=True)
self.button_driver_apply.connect("clicked", self.on_driver_changes_apply)
self.button_driver_cancel = Gtk.Button(label=_("_Cancel"), use_underline=True)
self.button_driver_cancel.connect("clicked", self.on_driver_changes_cancel)
self.button_driver_restart = Gtk.Button(label=_("_Restart..."), use_underline=True)
self.button_driver_restart.connect("clicked", self.on_driver_restart_clicked)
self.button_driver_revert.set_sensitive(False)
self.button_driver_revert.set_visible(True)
self.button_driver_apply.set_sensitive(False)
self.button_driver_apply.set_visible(True)
self.button_driver_cancel.set_visible(False)
self.button_driver_restart.set_visible(False)
self.box_driver_action.pack_end(self.button_driver_apply, False, False, 0)
self.box_driver_action.pack_end(self.button_driver_revert, False, False, 0)
self.box_driver_action.pack_end(self.button_driver_restart, False, False, 0)
self.box_driver_action.pack_end(self.button_driver_cancel, False, False, 0)
props = {"halign":Gtk.Align.CENTER, "valign":Gtk.Align.CENTER,
"vexpand":True, "visible":True}
self.label_driver_detail = Gtk.Label(label=_("Searching for available "
"drivers..."), **props)
self.box_driver_detail.add(self.label_driver_detail)
self.progress_bar = Gtk.ProgressBar()
self.box_driver_action.pack_end(self.progress_bar, False, False, 0)
self.progress_bar.set_visible(False)
self.devices = {}
self.detect_called = False
self.driver_changes = []
self.orig_selection = {}
# HACK: the case where the selection is actually "Do not use"; is a little
# tricky to implement because you can't check for whether a package is
# installed or any such thing. So let's keep a list of all the
# "Do not use" radios, set those active first, then iterate through
# orig_selection when doing a Reset.
self.no_drv = []
self.nonfree_drivers = 0
self.ui_building = False
def detect_drivers(self):
# WARNING: This is run in a separate thread.
self.detect_called = True
try:
self.devices = detect.system_device_drivers(self.apt_cache)
except:
# Catch all exceptions and feed them to apport.
GLib.idle_add(self.label_driver_detail.set_text,
_("An error occurred while searching for drivers."))
# For apport to catch this exception. See
# http://bugs.python.org/issue1230540
sys.excepthook(*sys.exc_info())
return
GLib.idle_add(self.show_drivers)
def on_driver_selection_changed(self, button, modalias, pkg_name=None):
if self.ui_building:
return
pkg = None
if pkg_name:
pkg = self.apt_cache[pkg_name]
# Add the matching linux modules package when available
try:
modules_package = detect.get_linux_modules_metapackage(self.apt_cache, pkg_name)
modules_package_obj = self.apt_cache[modules_package]
if modules_package and not modules_package_obj.current_ver:
self.driver_changes.append(modules_package_obj)
except (KeyError, TypeError):
pass
if button.get_active():
if pkg in self.driver_changes:
self.driver_changes.remove(pkg)
if (pkg is not None
and modalias in self.orig_selection
and button is not self.orig_selection[modalias]):
self.driver_changes.append(pkg)
else:
if pkg in self.driver_changes:
self.driver_changes.remove(pkg)
# for revert; to re-activate the original radio buttons.
if modalias not in self.orig_selection:
self.orig_selection[modalias] = button
if (pkg is not None
and pkg not in self.driver_changes
and pkg.current_ver):
self.driver_changes.append(pkg)
self.button_driver_revert.set_sensitive(bool(self.driver_changes))
self.button_driver_apply.set_sensitive(bool(self.driver_changes))
def gather_device_data(self, device):
'''Get various device data used to build the GUI.
return a tuple of (overall_status string, icon, drivers dict).
the drivers dict is using this form:
{"recommended/alternative": {pkg_name: {
'selected': True/False
'description': 'description'
'builtin': True/False
}
}}
"manually_installed": {"manual": {'selected': True, 'description': description_string}}
"no_driver": {"no_driver": {'selected': True/False, 'description': description_string}}
Please note that either manually_installed and no_driver are set to None if not applicable
(no_driver isn't present if there are builtins)
'''
possible_overall_status = {
'recommended': (_("This device is using the recommended driver."), "recommended-driver"),
'alternative': (_("This device is using an alternative driver."), "other-driver"),
'manually_installed': (_("This device is using a manually-installed driver."), "other-driver"),
'no_driver': (_("This device is not working."), "disable-device")
}
returned_drivers = {'recommended': {}, 'alternative': {}, 'manually_installed': {}, 'no_driver': {}}
have_builtin = False
one_selected = False
try:
if device['manual_install']:
returned_drivers['manually_installed'] = {True: {'selected': True,
'description': _("Continue using a manually installed driver")}}
except KeyError:
pass
for pkg_driver_name in device['drivers']:
current_driver = device['drivers'][pkg_driver_name]
# get general status
driver_status = 'alternative'
try:
if current_driver['recommended'] and current_driver['from_distro']:
driver_status = 'recommended'
except KeyError:
pass
builtin = False
try:
if current_driver['builtin']:
builtin = True
have_builtin = True
except KeyError:
pass
try:
pkg = self.apt_cache[pkg_driver_name]
installed = pkg.current_ver
candidate = self.depcache.get_candidate_ver(pkg)
if candidate is not None:
self.records.lookup(candidate.file_list[0])
description = _("Using {} from {}").format(self.records.short_desc, pkg_driver_name)
else:
description = _("Using {}").format(pkg_driver_name)
except KeyError:
print("WARNING: a driver ({}) doesn't have any available package associated: {}".format(pkg_driver_name, current_driver))
continue
# gather driver description
if current_driver['free']:
licence = _("open source")
else:
licence = _("proprietary")
if driver_status == 'recommended':
base_string = _("{base_description} ({licence}, tested)")
else:
base_string = _("{base_description} ({licence})")
description = base_string.format(base_description=description, licence=licence)
selected = False
if not builtin and not returned_drivers['manually_installed']:
selected = installed
if installed:
selected = True
one_selected = True
returned_drivers[driver_status].setdefault(pkg_driver_name, {'selected': selected,
'description': description,
'builtin': builtin})
# adjust making the needed addition
if not have_builtin:
selected = False
if not one_selected:
selected = True
returned_drivers["no_driver"] = {True: {'selected': selected,
'description': _("Do not use the device")}}
else:
# we have a builtin and no selection: builtin is the selected one then
if not one_selected:
for section in ('recommended', 'alternative'):
for pkg_name in returned_drivers[section]:
if returned_drivers[section][pkg_name]['builtin']:
returned_drivers[section][pkg_name]['selected'] = True
# compute overall status
for section in returned_drivers:
for keys in returned_drivers[section]:
if returned_drivers[section][keys]['selected']:
(overall_status, icon) = possible_overall_status[section]
return (overall_status, icon, returned_drivers)
def show_drivers(self):
if not self.devices:
# No drivers found.
self.label_driver_detail.set_text(_("No additional drivers available."))
return
else:
self.box_driver_detail.remove(self.label_driver_detail)
self.ui_building = True
self.dynamic_device_status = {}
for device in sorted(self.devices.keys()):
(overall_status, icon, drivers) = self.gather_device_data(self.devices[device])
driver_status = Gtk.Image()
driver_status.set_valign(Gtk.Align.START)
driver_status.set_halign(Gtk.Align.CENTER)
driver_status.set_from_icon_name(icon, Gtk.IconSize.MENU)
device_box = Gtk.Box(spacing=6, orientation=Gtk.Orientation.HORIZONTAL)
device_box.pack_start(driver_status, False, False, 6)
device_detail = Gtk.Box(spacing=6, orientation=Gtk.Orientation.VERTICAL)
device_box.pack_start(device_detail, True, True, 0)
widget = Gtk.Label.new("{}: {}".format(self.devices[device].get('vendor', _('Unknown')), self.devices[device].get('model', _('Unknown'))))
widget.set_halign(Gtk.Align.START)
device_detail.pack_start(widget, True, False, 0)
widget = Gtk.Label.new("<small>{}</small>".format(overall_status))
widget.set_halign(Gtk.Align.START)
widget.set_use_markup(True)
device_detail.pack_start(widget, True, False, 0)
self.dynamic_device_status[device] = (driver_status, widget)
option_group = None
# define the order of introspection
for section in ('recommended', 'alternative', 'manually_installed', 'no_driver'):
for driver in drivers[section]:
radio_button = Gtk.RadioButton.new_with_label(None, drivers[section][driver]['description'])
if option_group:
radio_button.join_group(option_group)
else:
option_group = radio_button
device_detail.pack_start(radio_button, True, False, 0)
radio_button.set_active(drivers[section][driver]['selected'])
if section == 'no_driver':
self.no_drv.append(radio_button)
if section in ('manually_install', 'no_driver') or ('builtin' in drivers[section][driver] and drivers[section][driver]['builtin']):
radio_button.connect("toggled", self.on_driver_selection_changed, device)
else:
radio_button.connect("toggled", self.on_driver_selection_changed, device, driver)
if drivers['manually_installed'] and section != 'manually_installed':
radio_button.set_sensitive(False)
self.box_driver_detail.pack_start(device_box, False, False, 6)
self.ui_building = False
self.box_driver_detail.show_all()
self.set_driver_action_status()
def update_label_and_icons_from_status(self):
'''Update the current label and icon, computing the new device status'''
for device in self.devices:
(overall_status, icon, drivers) = self.gather_device_data(self.devices[device])
(driver_status, widget) = self.dynamic_device_status[device]
driver_status.set_from_icon_name(icon, Gtk.IconSize.MENU)
widget.set_label("<small>{}</small>".format(overall_status))
def set_driver_action_status(self):
# Update the label in case we end up having some kind of proprietary driver in use.
if (os.path.exists('/var/run/reboot-required')):
self.label_driver_action.set_label(_("You need to restart the computer to complete the driver changes."))
self.button_driver_restart.set_visible(True)
self.window_main.set_urgency_hint(True)
return
self.nonfree_drivers = 0
for device in self.devices:
for pkg_name in self.devices[device]['drivers']:
pkg = self.apt_cache[pkg_name]
if not self.devices[device]['drivers'][pkg_name]['free'] and pkg.current_ver:
self.nonfree_drivers = self.nonfree_drivers + 1
if self.nonfree_drivers > 0:
self.label_driver_action.set_label(gettext.ngettext(
"%(count)d proprietary driver in use.",
"%(count)d proprietary drivers in use.",
self.nonfree_drivers)
% {'count': self.nonfree_drivers})
else:
self.label_driver_action.set_label(_("No proprietary drivers are in use."))
def init_ubuntu_pro(self):
self.ubuntu_pro_page = UbuntuProPage(self)
Mini Shell