Source code for feynml.head

import logging
import warnings
from dataclasses import dataclass, field
from importlib.metadata import version
from typing import List, Optional

import cssutils
import smpl_io.io as io

from feynml.link import Link
from feynml.meta import Meta as alias_meta
from feynml.sheet import SheetHandler
from feynml.styled import CSSSheet, Styled


@dataclass
class Head(SheetHandler):
    class Meta:
        name = "head"

    metas: List[alias_meta] = field(
        default_factory=list,
        metadata={"name": "meta", "namespace": ""},
    )
    links: List[Link] = field(
        default_factory=list,
        metadata={"name": "link", "namespace": ""},
    )
    cached_links = {}
    root = "."  # root directory for relative links
    title: Optional[str] = field(default=None, metadata={"type": "Element"})

    # description: Optional[str] = field(default="", metadata={"type": "Element"})

    # style: Optional[str] = field(default="", metadata={"type": "Element"})
    style: CSSSheet = field(
        default_factory=lambda: cssutils.parseString(""),
        metadata={
            "name": "style",
            "type": "Element",
        },
    )

    def get_link_dict(self, cached=True, cache=True):
        """
        Return a dictionary of resolved links.

        If cached is True, then the cached resolved links are returned.
        """
        ret = {}
        for m in self.links:
            if m.rel in self.cached_links and cached:
                ret[m.rel] = self.cached_links[m.rel]
            else:
                # handle absolute or relative
                if m.href.startswith("http"):
                    print("Direct")
                    ret[m.rel] = io.read(m.href)
                    print(ret[m.rel])
                elif m.href.startswith("/"):
                    ret[m.rel] = io.read(m.href)
                else:
                    ret[m.rel] = io.read(self.root + m.href)
                if m.rel == "stylesheet":
                    ret[m.rel] = cssutils.parseString(ret[m.rel])
                if cache:
                    self.cached_links[m.rel] = ret[m.rel]
        return ret

    def get_meta_dict(self):
        """
        Return a dictionary of meta tags.
        """
        return {m.name: m.content for m in self.metas}

    def get_sheets(self):
        sheets = []
        sheets += super().get_sheets()
        for k, v in self.get_link_dict().items():
            if k == "stylesheet":
                sheets += [v]
        sheets += [self.style]
        return sheets

    def get_sheet(self):
        return self.style

    def with_sheet(self, sheet):
        self.style = sheet
        return self