X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/551bb1c9f85382a90abc89ac622bd363dccd36a7..62043b2bafbcb57beae7559235bd6984b6275b55:/glowfic.py?ds=inline diff --git a/glowfic.py b/glowfic.py index e192646..8561cd7 100644 --- a/glowfic.py +++ b/glowfic.py @@ -89,19 +89,33 @@ def makeChunk(chunk_dom: Tag, image_store: ImageStore) -> Chunk: assert tag is None or isinstance(tag, Tag) return tag + def stripHREF(tag: Tag) -> None: + for c in tag.findChildren("a"): + if "href" in c.attrs: + del c.attrs["href"] + + def getMeta(css_class: str) -> Tag | None: + tag = getByClass(css_class) + if tag is None: + return None + stripHREF(tag) + return tag + content = chunk_dom.findChild('div', class_='post-content') assert isinstance(content, Tag) return Chunk(getIcon(), - getByClass('post-character'), - getByClass('post-screenname'), - getByClass('post-author'), + getMeta('post-character'), + getMeta('post-screenname'), + getMeta('post-author'), content) -def renderIcon(icon_path: str | None, image_size: float) -> str | None: - params = f'width={image_size}mm,height={image_size}mm,keepaspectratio' - return f'\\includegraphics[{params}]{{{icon_path}}}' if icon_path else None +def renderIcon(icon_path: str | None, image_size: float) -> bytes | None: + if icon_path is None: + return None + return b'\\includegraphics[width=%fmm,height=%fmm,keepaspectratio]{%s}' % ( + image_size, image_size, icon_path.encode('UTF-8')) class Layout(ABC): @@ -128,7 +142,7 @@ class BelowIconLayout(Layout): def renderChunk(self, chunk: Chunk) -> bytes: icon = renderIcon(chunk.icon, self._image_size) - meta = [icon.encode('UTF-8')] if icon else [] + meta = [icon] if icon else [] meta += [self._texifier.texify(x) for x in [chunk.character, chunk.screen_name, chunk.author] if x is not None] @@ -189,6 +203,6 @@ class BesideIconLayout(Layout): \\strut ''' % ( - icon.encode('UTF-8') if icon else b'', + icon if icon else b'', b'\\\\*'.join(self._texifier.texify(x) for x in meta if x is not None), self._texifier.texify(chunk.content))