class ShellThemeTweak(Gtk.Box, Tweak):
THEME_EXT_NAME = "*****@*****.**"
THEME_GSETTINGS_SCHEMA = "org.gnome.shell.extensions.user-theme"
THEME_GSETTINGS_NAME = "name"
THEME_GSETTINGS_DIR = os.path.join(GLib.get_user_data_dir(), "gnome-shell",
"extensions", THEME_EXT_NAME, "schemas")
LEGACY_THEME_DIR = os.path.join(GLib.get_home_dir(), ".themes")
THEME_DIR = os.path.join(GLib.get_user_data_dir(), "themes")
def __init__(self, **options):
Gtk.Box.__init__(self, orientation=Gtk.Orientation.HORIZONTAL)
Tweak.__init__(self, _("Shell"),
_("Install custom or user themes for gnome-shell"),
**options)
#check the shell is running and the usertheme extension is present
error = _("Unknown error")
self._shell = _shell
if self._shell is None:
logging.warning("Shell not running", exc_info=True)
error = _("Shell not running")
else:
try:
extensions = self._shell.list_extensions()
if ShellThemeTweak.THEME_EXT_NAME in extensions and extensions[
ShellThemeTweak.THEME_EXT_NAME]["state"] == 1:
#check the correct gsettings key is present
try:
if os.path.exists(ShellThemeTweak.THEME_GSETTINGS_DIR):
self._settings = GSettingsSetting(
ShellThemeTweak.THEME_GSETTINGS_SCHEMA,
schema_dir=ShellThemeTweak.THEME_GSETTINGS_DIR)
else:
self._settings = GSettingsSetting(
ShellThemeTweak.THEME_GSETTINGS_SCHEMA)
name = self._settings.get_string(
ShellThemeTweak.THEME_GSETTINGS_NAME)
ext = extensions[ShellThemeTweak.THEME_EXT_NAME]
logging.debug("Shell user-theme extension\n%s" %
pprint.pformat(ext))
error = None
except:
logging.warning(
"Could not find user-theme extension in %s" %
','.join(list(extensions.keys())),
exc_info=True)
error = _(
"Shell user-theme extension incorrectly installed")
else:
error = _("Shell user-theme extension not enabled")
except Exception as e:
logging.warning("Could not list shell extensions",
exc_info=True)
error = _("Could not list shell extensions")
if error:
cb = build_combo_box_text(None)
build_label_beside_widget(self.name, cb, warning=error, hbox=self)
self.widget_for_size_group = cb
else:
#include both system, and user themes
#note: the default theme lives in /system/data/dir/gnome-shell/theme
# and not themes/, so add it manually later
dirs = [
os.path.join(d, "themes") for d in GLib.get_system_data_dirs()
]
dirs += [ShellThemeTweak.THEME_DIR]
dirs += [ShellThemeTweak.LEGACY_THEME_DIR]
valid = walk_directories(dirs, lambda d:
os.path.exists(os.path.join(d, "gnome-shell")) and \
os.path.exists(os.path.join(d, "gnome-shell", "gnome-shell.css")))
#the default value to reset the shell is an empty string
valid.extend(("", ))
valid = set(valid)
#build a combo box with all the valid theme options
#manually add Adwaita to represent the default
cb = build_combo_box_text(
self._settings.get_string(
ShellThemeTweak.THEME_GSETTINGS_NAME),
*make_combo_list_with_default(
valid, "", default_text=_("<i>Default</i>")))
cb.connect('changed', self._on_combo_changed)
self._combo = cb
#a filechooser to install new themes
chooser = FileChooserButton(_("Select a theme"), True,
["application/zip"])
chooser.connect("file-set", self._on_file_set)
build_label_beside_widget(self.name, chooser, cb, hbox=self)
self.widget_for_size_group = cb
def _on_file_set(self, chooser):
f = chooser.get_filename()
with zipfile.ZipFile(f, 'r') as z:
try:
fragment = ()
theme_name = None
for n in z.namelist():
if n.endswith("gnome-shell.css"):
fragment = n.split("/")[0:-1]
if n.endswith("gnome-shell/theme.json"):
logging.info("New style theme detected (theme.json)")
#new style theme - extract the name from the json file
tmp = tempfile.mkdtemp()
z.extract(n, tmp)
with open(os.path.join(tmp, n)) as f:
try:
theme_name = json.load(
f)["shell-theme"]["name"]
except:
logging.warning("Invalid theme format",
exc_info=True)
if not fragment:
raise Exception("Could not find gnome-shell.css")
if not theme_name:
logging.info(
"Old style theme detected (missing theme.json)")
#old style themes name was taken from the zip name
if fragment[0] == "theme" and len(fragment) == 1:
theme_name = os.path.basename(f)
else:
theme_name = fragment[0]
theme_members_path = "/".join(fragment)
ok, updated = extract_zip_file(
z, theme_members_path,
os.path.join(ShellThemeTweak.THEME_DIR, theme_name,
"gnome-shell"))
if ok:
if updated:
self.notify_information(
_("%s theme updated successfully") % theme_name)
else:
self.notify_information(
_("%s theme installed successfully") % theme_name)
#I suppose I could rely on updated as indicating whether to add the theme
#name to the combo, but just check to see if it is already there
model = self._combo.get_model()
if theme_name not in [r[0] for r in model]:
model.append((theme_name, theme_name))
else:
self.notify_information(_("Error installing theme"))
except:
#does not look like a valid theme
self.notify_information(_("Invalid theme"))
logging.warning("Error parsing theme zip", exc_info=True)
#set button back to default state
chooser.unselect_all()
def _on_combo_changed(self, combo):
val = combo.get_model().get_value(combo.get_active_iter(), 0)
self._settings.set_string(ShellThemeTweak.THEME_GSETTINGS_NAME, val)
def set_theme(desk_env, t_type, theme):
if desk_env not in SUPPORTED_DESKENVS:
raise Exception('Invalid desktop environment!')
elif not theme:
logger.error('{}\'s {} theme not set '.format(correct_name_case(desk_env), correct_name_case(t_type)))
return
if t_type in ['gtk', 'icons']:
from gi.repository import Gio
if desk_env == 'cinnamon':
gsettings_string = 'org.cinnamon.desktop.interface'
else:
gsettings_string = 'org.gnome.desktop.interface'
gsettings = Gio.Settings.new(gsettings_string)
# Safety fallback, shouldn't happen
else:
gsettings = None
if t_type == 'gtk':
# Easy peasy
# For GNOME and Cinnamon
gsettings['gtk-theme'] = theme
# For XFCE
if desk_env == 'xfce':
from subprocess import run, CalledProcessError
try:
# noinspection SpellCheckingInspection
run(['xfconf-query', '-c', 'xsettings', '-p', '/Net/ThemeName', '-s', theme])
except CalledProcessError:
logger.error('Could not apply GTK theme')
return
# Switching GTK themes in KDE is a little bit more complicated than the others...
# For KDE
elif desk_env == 'kde':
from subprocess import run
import configparser
import fileinput
import sys
from automathemely.autoth_tools.utils import get_bin
# Set GTK3 theme
# This would usually be done with kwriteconfig but since there is no way to notify GTK3 apps that the
# theme has changed in KDE like with GTK2 anyway we might as well do it this way
parser = configparser.ConfigParser(strict=False)
# Prevent changing the key's case
parser.optionxform = lambda option: option
parser.read(PATH_CONSTANTS['kde-gtk-config']['gtk3'])
parser['Settings']['gtk-theme-name'] = theme
with open(PATH_CONSTANTS['kde-gtk-config']['gtk3'], 'w') as f:
parser.write(f, space_around_delimiters=False)
# Search for gtk2 config file in theme dir in PATH_CONSTANTS dirs
# As if it wasn't messy enough already...
match = walk_filter_dirs(PATH_CONSTANTS['general-themes'], lambda parent_dir, t: Path(parent_dir)
.joinpath(t).glob('gtk-2.*/gtkrc') and t.lower() != 'default', return_parent=True)
if not match:
logger.warning('The selected GTK theme does not contain a GTK2 theme, so some applications may '
'look odd')
else:
# If there are several themes with the same name in different directories, only get the first one
# to match according to the dirs hierarchy in PATH_CONSTANTS
theme_parent = match[0][1]
if not Path(PATH_CONSTANTS['kde-gtk-config']['gtk2']).is_file():
logger.warning('GTK2 config file not found, set a theme from System Settings at least once and '
'try again')
# Write GTK2 config
else:
line_replaced, previous_is_autogen_comment = False, False
for line in fileinput.input(PATH_CONSTANTS['kde-gtk-config']['gtk2'], inplace=True):
if line.startswith('# Configs for GTK2 programs'):
previous_is_autogen_comment = True
sys.stdout.write(line)
# Even in kde-config-gtk they don't seem to agree if this may not actually be needed, but
# just in case here it is
elif line.startswith('include'):
print('include "{}"'.format(str(Path(theme_parent).joinpath(theme, 'gtk-2.0', 'gtkrc'))))
# This is the one that matters
elif line.startswith('gtk-theme-name='):
print('gtk-theme-name="{}"'.format(theme))
line_replaced = True
else:
if previous_is_autogen_comment and not line.startswith('# Edited by AutomaThemely'):
print('# Edited by AutomaThemely')
previous_is_autogen_comment = False
sys.stdout.write(line)
if not line_replaced:
logger.warning('GTK2 config file is invalid, set a theme from System Settings at least '
'once and try again')
else:
# Send signal to GTK2 apps to refresh their themes
run([get_bin('kde-refresh-gtk2')])
elif t_type == 'icons':
# For GNOME and Cinnamon
gsettings['icon-theme'] = theme
# For XFCE
if desk_env == 'xfce':
from subprocess import run, CalledProcessError
try:
# noinspection SpellCheckingInspection
run(['xfconf-query', '-c', 'xsettings', '-p', '/Net/IconThemeName', '-s', theme])
except CalledProcessError:
logger.error('Could not apply icons theme')
return
elif t_type == 'shell':
# This is WAY out of my level, I'll just let the professionals handle this one...
try:
import gtweak
except ImportError:
logger.error('GNOME Tweaks not installed')
return
from gtweak.gshellwrapper import GnomeShellFactory
from gtweak.defs import GSETTINGS_SCHEMA_DIR, LOCALE_DIR
# This could probably be implemented in house but since we're already importing from gtweak I guess it'll stay
# this way for now
from gtweak.gsettings import GSettingsSetting
gtweak.GSETTINGS_SCHEMA_DIR = GSETTINGS_SCHEMA_DIR
gtweak.LOCALE_DIR = LOCALE_DIR
shell = GnomeShellFactory().get_shell()
shell_theme_name = '*****@*****.**'
shell_theme_schema = 'org.gnome.shell.extensions.user-theme'
shell_theme_schema_dir = Path(PATH_CONSTANTS['shell-user-extensions']).joinpath(shell_theme_name, 'schemas')
if not shell:
logger.error('GNOME Shell not running')
return
else:
# noinspection PyBroadException
try:
shell_extensions = shell.list_extensions()
except Exception:
logger.error('GNOME Shell extensions could not be loaded')
return
else:
# noinspection PyBroadException
try:
if shell_theme_name in shell_extensions and shell_extensions[shell_theme_name]['state'] == 1:
# If shell user-theme was installed locally e. g. through extensions.gnome.org
if Path(shell_theme_schema_dir).is_dir():
user_shell_settings = GSettingsSetting(shell_theme_schema,
schema_dir=str(shell_theme_schema_dir))
# If it was installed as a system extension
else:
user_shell_settings = GSettingsSetting(shell_theme_schema)
else:
logger.error('GNOME Shell user theme extension not enabled')
return
except Exception:
logger.error('Could not load GNOME Shell user theme extension')
return
else:
# To set the default theme you have to input an empty string, but since that won't work with the
# Setting Manager's ComboBoxes we set it by this placeholder name
if theme == 'default':
theme = ''
# Set the GNOME Shell theme
user_shell_settings.set_string('name', theme)
elif t_type == 'lookandfeel':
from subprocess import run, CalledProcessError
try:
# noinspection SpellCheckingInspection
run(['lookandfeeltool', '-a', '{}'.format(theme)], check=True)
except CalledProcessError:
logger.error('Could not apply Look and Feel theme')
return
elif t_type == 'desktop':
from gi.repository import Gio
cinnamon_settings = Gio.Settings.new('org.cinnamon.theme')
cinnamon_settings['name'] = theme
