| import ast | |
| import json | |
| import os | |
| import importlib | |
| import logging | |
| logger = logging.getLogger(__name__) | |
| folder = os.path.dirname(os.path.abspath(__file__)) | |
| folder = os.path.dirname(folder) | |
| folder = os.path.dirname(folder) | |
| folder = os.path.join(folder, "assets", "themes") | |
| import sys | |
| sys.path.append(folder) | |
| def get_class(file_name, class_name): | |
| with open(file_name, 'r') as file: | |
| content = file.read() | |
| syntax_tree = ast.parse(content) | |
| for node in ast.walk(syntax_tree): | |
| if isinstance(node, ast.ClassDef) and node.name == class_name: | |
| return node | |
| return None | |
| def get_list(): | |
| themes_list = [ | |
| os.path.splitext(name)[0] | |
| for root, _, files in os.walk(folder, topdown=False) | |
| for name in files | |
| if name.endswith(".py") and root == folder | |
| ] | |
| return themes_list | |
| def select_theme(name): | |
| selected_file = name + ".py" | |
| class_name = name | |
| full_path = os.path.join(folder, selected_file) | |
| class_found = get_class(full_path, class_name) | |
| if class_found: | |
| with open(os.path.join(folder, 'theme.json'), 'w') as json_file: | |
| json.dump({"file": selected_file, "class": class_name}, json_file) | |
| logger.info(f"Theme {class_name} successfully selected, restart applio.") | |
| else: | |
| logger.warn(f"Theme {class_name} was not found.") | |
| def read_json(): | |
| json_file_name = os.path.join(folder, 'theme.json') | |
| try: | |
| with open(json_file_name, 'r') as json_file: | |
| data = json.load(json_file) | |
| selected_file = data.get("file") | |
| class_name = data.get("class") | |
| if selected_file and class_name: | |
| return class_name | |
| else: | |
| return "" | |
| except: | |
| return "applio" | |
| def load_json(): | |
| json_file_name = os.path.join(folder, 'theme.json') | |
| try: | |
| with open(json_file_name, 'r') as json_file: | |
| data = json.load(json_file) | |
| selected_file = data.get("file") | |
| class_name = data.get("class") | |
| if selected_file and class_name: | |
| module = importlib.import_module(selected_file[:-3]) | |
| obtained_class = getattr(module, class_name) | |
| instance = obtained_class() | |
| logger.info(f"Theme Loaded: {class_name}") | |
| return instance | |
| else: | |
| logger.warn("The theme is incorrect.") | |
| return None | |
| except Exception as e: | |
| logger.warning(f"Error Loading: {str(e)}") | |
| return None | |