🐛 为各主题添加 repost 的渲染

MountainDash · Mar 20, 2024 · e320646 · e320646
1 parent c915b36
commit e320646
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 2 deletions.
diff --git a/nonebot_bison/theme/themes/basic/build.py b/nonebot_bison/theme/themes/basic/build.py
@@ -25,11 +25,22 @@ async def render(self, post: "Post") -> list[MessageSegmentFactory]:
 
         text += post.content if len(post.content) < 500 else f"{post.content[:500]}..."
 
+        if rp := post.repost:
+            text += f"\n--------------\n转发自 {rp.nickname or ''}:\n"
+            text += f"{rp.title}\n"
+            text += rp.content if len(rp.content) < 500 else f"{rp.content[:500]}..."
+
         text += f"\n来源: {post.platform.name} {post.nickname or ''}\n"
 
+        urls: list[str] = []
+        if rp and rp.url:
+            urls += f"转发详情：{rp.url}"
         if post.url:
             text += f"详情: {post.url}"
 
+        if urls:
+            text += "\n".join(urls)
+
         msgs: list[MessageSegmentFactory] = [Text(text)]
         if post.images:
             pics = post.images

diff --git a/nonebot_bison/theme/themes/brief/build.py b/nonebot_bison/theme/themes/brief/build.py
@@ -18,7 +18,7 @@ async def render(self, post: "Post") -> list[MessageSegmentFactory]:
         if not post.title:
             raise ThemeRenderUnsupportError("Post has no title")
         text = f"{post.title}\n\n"
-        text += f"来源: {post.platform.name} {post.nickname or ''}\n"
+        text += f"来源: {post.platform.name} {post.nickname or ''}{' 的转发' if post.repost else ''}\n"
         if post.url:
             text += f"详情: {post.url}"
 

diff --git a/nonebot_bison/theme/themes/ht2i/build.py b/nonebot_bison/theme/themes/ht2i/build.py
@@ -35,12 +35,24 @@ async def render(self, post: "Post"):
 
         _text += post.content if len(post.content) < 500 else f"{post.content[:500]}..."
 
+        if rp := post.repost:
+            _text += f"\n--------------\n转发自 {rp.nickname or ''}:\n"
+            _text += f"{rp.title}\n"
+            _text += rp.content if len(rp.content) < 500 else f"{rp.content[:500]}..."
+
         _text += f"\n来源: {post.platform.name} {post.nickname or ''}\n"
 
         msgs: list[MessageSegmentFactory] = [await self._text_render(_text)]
 
+        urls: list[str] = []
+        if rp and rp.url:
+            urls += f"转发详情：{rp.url}"
         if post.url:
-            msgs.append(Text(f"详情: {post.url}"))
+            urls += f"详情: {post.url}"
+
+        if urls:
+            msgs.append(Text("\n".join(urls)))
+
         if post.images:
             pics = post.images
             if is_pics_mergable(pics):