123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112 |
- import configparser
- from typing import Dict, List, IO, Mapping, Optional
- from .default_styles import DEFAULT_STYLES
- from .style import Style, StyleType
- class Theme:
- """A container for style information, used by :class:`~rich.console.Console`.
- Args:
- styles (Dict[str, Style], optional): A mapping of style names on to styles. Defaults to None for a theme with no styles.
- inherit (bool, optional): Inherit default styles. Defaults to True.
- """
- styles: Dict[str, Style]
- def __init__(
- self, styles: Optional[Mapping[str, StyleType]] = None, inherit: bool = True
- ):
- self.styles = DEFAULT_STYLES.copy() if inherit else {}
- if styles is not None:
- self.styles.update(
- {
- name: style if isinstance(style, Style) else Style.parse(style)
- for name, style in styles.items()
- }
- )
- @property
- def config(self) -> str:
- """Get contents of a config file for this theme."""
- config = "[styles]\n" + "\n".join(
- f"{name} = {style}" for name, style in sorted(self.styles.items())
- )
- return config
- @classmethod
- def from_file(
- cls, config_file: IO[str], source: Optional[str] = None, inherit: bool = True
- ) -> "Theme":
- """Load a theme from a text mode file.
- Args:
- config_file (IO[str]): An open conf file.
- source (str, optional): The filename of the open file. Defaults to None.
- inherit (bool, optional): Inherit default styles. Defaults to True.
- Returns:
- Theme: A New theme instance.
- """
- config = configparser.ConfigParser()
- config.read_file(config_file, source=source)
- styles = {name: Style.parse(value) for name, value in config.items("styles")}
- theme = Theme(styles, inherit=inherit)
- return theme
- @classmethod
- def read(cls, path: str, inherit: bool = True) -> "Theme":
- """Read a theme from a path.
- Args:
- path (str): Path to a config file readable by Python configparser module.
- inherit (bool, optional): Inherit default styles. Defaults to True.
- Returns:
- Theme: A new theme instance.
- """
- with open(path, "rt") as config_file:
- return cls.from_file(config_file, source=path, inherit=inherit)
- class ThemeStackError(Exception):
- """Base exception for errors related to the theme stack."""
- class ThemeStack:
- """A stack of themes.
- Args:
- theme (Theme): A theme instance
- """
- def __init__(self, theme: Theme) -> None:
- self._entries: List[Dict[str, Style]] = [theme.styles]
- self.get = self._entries[-1].get
- def push_theme(self, theme: Theme, inherit: bool = True) -> None:
- """Push a theme on the top of the stack.
- Args:
- theme (Theme): A Theme instance.
- inherit (boolean, optional): Inherit styles from current top of stack.
- """
- styles: Dict[str, Style]
- styles = (
- {**self._entries[-1], **theme.styles} if inherit else theme.styles.copy()
- )
- self._entries.append(styles)
- self.get = self._entries[-1].get
- def pop_theme(self) -> None:
- """Pop (and discard) the top-most theme."""
- if len(self._entries) == 1:
- raise ThemeStackError("Unable to pop base theme")
- self._entries.pop()
- self.get = self._entries[-1].get
- if __name__ == "__main__": # pragma: no cover
- theme = Theme()
- print(theme.config)
|