Gradience/gradience/utils/preset.py
tfuxu 0fad4da087
feat: more readible exceptions in 'preset' module
* add .close() to file objects
* remove __name__ == '__main__' statement
2022-11-20 17:21:23 +01:00

142 lines
4.5 KiB
Python

# preset.py
#
# Change the look of Adwaita, with ease
# Copyright (C) 2022 Gradience Team
#
# 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 3 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, see <https://www.gnu.org/licenses/>.
import json
import os
from gradience.settings_schema import settings_schema
from gradience.utils.utils import buglog, to_slug_case
presets_dir = os.path.join(
os.environ.get("XDG_CONFIG_HOME", os.environ["HOME"] + "/.config"),
"presets",
)
class Preset:
variables = {}
palette = {}
custom_css = {
"gtk4": "",
"gtk3": "",
}
plugins = {}
display_name = "New Preset"
preset_path = "new_preset"
badges = {}
def __init__(self, preset_path=None, text=None, preset=None):
if preset_path:
self.load_preset(preset_path=preset_path)
elif text: # load from resource
self.load_preset(text=text)
elif preset: # css or dict
self.load_preset(preset=preset)
else:
raise Exception("Failed to create a new Preset object: Preset created without content")
def load_preset(self, preset_path=None, text=None, preset=None):
try:
if not preset:
if text:
preset_text = text
elif preset_path:
self.preset_path = preset_path
with open(self.preset_path, "r", encoding="utf-8") as file:
preset_text = file.read()
file.close()
else:
raise Exception("load_preset must be called with a path, text, or preset")
preset = json.loads(preset_text)
self.display_name = preset["name"]
self.variables = preset["variables"]
self.palette = preset["palette"]
if "badges" in preset:
self.badges = preset["badges"]
else:
self.badges = {}
if "custom_css" in preset:
self.custom_css = preset["custom_css"]
else:
for app_type in settings_schema["custom_css_app_types"]:
self.custom_css[app_type] = ""
except Exception as e:
if self.preset_path:
buglog(f"Failed to load preset {self.preset_path}. Exc: {e}")
else:
buglog(f"Failed to load preset with unknown path. Exc: {e}")
# Rename an existing preset
def rename_preset(self, name):
self.display_name = name
old_path = self.preset_path
self.preset_path = os.path.join(
os.path.dirname(self.preset_path),
to_slug_case(name) + ".json")
self.save_preset(to=self.preset_path)
os.remove(old_path)
# Save a new user preset (or overwrite one)
def save_preset(self, name=None, plugins_list=None, to=None):
self.display_name = name if name else self.display_name
if to is None:
filename = to_slug_case(name) if name else "new_preset"
self.preset_path = os.path.join(
presets_dir, "user", filename + ".json")
else:
self.preset_path = to
if not os.path.exists(
os.path.join(
presets_dir,
"user",
)
):
os.makedirs(
os.path.join(
presets_dir,
"user",
)
)
if plugins_list is None:
plugins_list = {}
else:
plugins_list = plugins_list.save()
with open(self.preset_path, "w", encoding="utf-8") as file:
object_to_write = {
"name": self.display_name,
"variables": self.variables,
"palette": self.palette,
"custom_css": self.custom_css,
"plugins": plugins_list,
}
file.write(json.dumps(object_to_write, indent=4))
file.close()
def validate(self):
return True