@dataclass(frozen=True)
class Chunk:
icon: str | None
- character: str | None
- screen_name: str | None
- author: str | None
+ character: Tag | None
+ screen_name: Tag | None
+ author: Tag | None
content: Tag
# We avoid the name "post" because the Glowfic community uses the term
assert isinstance(icon_img, Tag)
return image_store.get_image(icon_img.attrs['src'])
- def getTextByClass(css_class: str) -> str | None:
- div = chunk_dom.find_next('div', class_=css_class)
- if div is None:
- return None
- return div.text.strip()
+ def getByClass(css_class: str) -> Tag | None:
+ tag = chunk_dom.find_next('div', class_=css_class)
+ assert tag is None or isinstance(tag, Tag)
+ return tag
content = chunk_dom.find_next('div', class_='post-content')
assert isinstance(content, Tag)
return Chunk(getIcon(),
- getTextByClass('post-character'),
- getTextByClass('post-screenname'),
- getTextByClass('post-author'),
+ getByClass('post-character'),
+ getByClass('post-screenname'),
+ getByClass('post-author'),
content)
self._image_size = image_size
def renderChunk(self, chunk: Chunk) -> bytes:
- meta = [
- renderIcon(chunk.icon, self._image_size),
- chunk.character,
- chunk.screen_name,
- chunk.author,
- ]
+ icon = renderIcon(chunk.icon, self._image_size)
+ meta = [icon.encode('UTF-8')] if icon else []
+ meta += [self._texifier.texify(x)
+ for x in [chunk.character, chunk.screen_name, chunk.author]
+ if x is not None]
return b'''\\wrapstuffclear
\\begin{wrapstuff}[l]
\\noindent %s
''' % (
- b'\\\\*'.join(x.encode('UTF-8') for x in meta if x is not None),
+ b'\\\\*'.join(meta),
self._texifier.texify(chunk.content))
''' % (
icon.encode('UTF-8') if icon else b'',
- b'\\\\*'.join(x.encode('UTF-8') for x in meta if x is not None),
+ b'\\\\*'.join(self._texifier.texify(x) for x in meta if x is not None),
self._texifier.texify(chunk.content))
c.icon,
'stored:https://d1anwqy6ci9o1i.cloudfront.net/' +
'users%2F366%2Ficons%2Fxqmypqvflgdy28aorw9ml_shock.png')
- self.assertEqual(c.character, 'Keltham')
- self.assertEqual(c.screen_name, 'lawful chaotic')
- self.assertEqual(c.author, 'Iarwain')
+ assert c.character
+ assert c.screen_name
+ assert c.author
+ self.assertEqual(list(c.character.stripped_strings), ['Keltham'])
+ self.assertEqual(
+ list(c.screen_name.stripped_strings), ['lawful chaotic'])
+ self.assertEqual(list(c.author.stripped_strings), ['Iarwain'])
self.assertEqual(str(c.content),
'<div class="post-content"><p></p></div>')