from insignal.config import path
import yaml
import pathlib
from jinja2 import Markup
from flask import render_template

sub_pages_yml = yaml.load(open(pathlib.Path(path.get_config_path(), "sub_pages.yml")), yaml.Loader)


def get_sub_pages(category, menu):
    sub_page_items = sub_pages_yml.get(category.upper()).get(menu)
    sub_pages = {page_name: Markup(render_template(page_file)) for page_name, page_file in sub_page_items.items()}

    return sub_pages