r3gm's picture
Upload 340 files
3b7b011
raw
history blame
2.57 kB
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