lib/tools/loader_themes.py

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