from panflute import Space, SoftBreak, Str, Math from typing import Union # Import local files from context import Context Whitespace = Union[Space,SoftBreak] class NBSP(Space): pass def bavlna(e: Whitespace, c: Context) -> bool: """Determine if given piece of whitespace should be non-breakable.""" if c.get_metadata("language") == "cs": prev = e.prev if isinstance(e.prev, Str) else (e.prev.content[-1] if hasattr(e.prev, "content") and len(e.prev.content) != 0 else None) next = e.next if isinstance(e.next, Str) else (e.next.content[0] if hasattr(e.next, "content") and len(e.next.content) != 0 else None) if isinstance(prev, Str) and isinstance(next, Str): if prev.text.lower() in ['k', 's', 'v', 'z', 'o', 'u', 'a', 'i']: return True if isinstance(e.prev, Str) and isinstance(e.next, Str): prevC = e.prev.text[-1] nextC = e.next.text[0] numbers = ["0123456789"] operators = ["+-/*^%:"] if prevC in numbers and nextC in numbers: return True if prevC in numbers and nextC in operators: return True if prevC in operators and nextC in numbers: return True if isinstance(e.prev, Math) or isinstance(e.next, Math): return True return False