Gradience/gradience/utils/preset.py
Kian-Meng Ang 0f297aece3 docs: fix typos
Found via `codespell -S po,builddir`.
2022-11-05 09:37:58 +08:00

141 lines
4.2 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 = {}
repo = "user"
display_name = "New Preset"
filename = "new_preset"
badges = {}
def __init__(self, name=None, repo=None, preset_path=None, text=None, preset=None):
if text: # load from resource
self.load_preset(text=text)
elif preset: # css or dict
self.load_preset(preset=preset)
else:
if name is not None:
self.filename = to_slug_case(name)
if repo is not None:
self.repo = repo
if preset_path is None:
self.preset_path = os.path.join(
presets_dir, repo, self.filename + ".json")
else:
self.preset_path = preset_path
self.load_preset()
def load_preset(self, text=None, preset=None):
try:
if not preset:
if text:
preset_text = text
else:
with open(self.preset_path, "r", encoding="utf-8") as file:
preset_text = file.read()
preset = json.loads(preset_text)
self.display_name = preset["name"]
self.filename = to_slug_case(self.display_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 error:
buglog(error, " -> preset : ", self.preset_path)
def save_preset(self, name=None, plugins_list=None, to=None):
self.display_name = name if name else self.display_name
self.filename = to_slug_case(name) if name else self.filename
if to is None:
self.preset_path = os.path.join(
presets_dir, self.repo, self.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()
print(self.preset_path)
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))
def validate(self):
return True
if __name__ == "__main__":
p = Preset("test", "user")
buglog(p.variables)
buglog(p.palette)