♻️ simplify paper information formatting and update allowed render types

Author: BalconyJH

nonebot_plugin_paper/command.py

@@ -1,13 +1,12 @@
-from pathlib import Path
-
 from aioarxiv.client.arxiv_client import ArxivClient
 from arclet.alconna import Alconna, Args, Subcommand
 from nonebot.log import logger
 from nonebot.typing import T_State
-from nonebot_plugin_alconna import Image, UniMsg, on_alconna
-from nonebot_plugin_htmlrender import template_to_pic
+from nonebot_plugin_alconna import UniMsg, on_alconna
 from nonebot_plugin_uninfo import Uninfo
 
+from nonebot_plugin_paper.utils import text_paper_info
+
 paper_cmd = on_alconna(
     Alconna(
         "paper",
@@ -37,15 +36,4 @@ async def handle_search(keyword: str, state: T_State, uninfo: Uninfo, unimsg: Un
             f"Search result for {keyword} and get {result.total_result} papers"
         )
         for paper in result.papers:
-            # await paper_cmd.send(text_paper_info(paper))
-            await paper_cmd.send(
-                Image(
-                    raw=(
-                        await template_to_pic(
-                            template_path=str(Path(__file__).parent / "jinja_temple"),
-                            template_name="paper.jinja2",
-                            templates={"paper": paper.model_dump()},
-                        )
-                    )
-                )
-            )
+            await paper_cmd.send(text_paper_info(paper))

nonebot_plugin_paper/config.py

@@ -11,7 +11,7 @@
 
 @custom_validation
 class RenderType(str):
-    ALLOWED_VALUES: ClassVar = ["playwright", "pillow", "plaintext"]
+    ALLOWED_VALUES: ClassVar = ["playwright", "pillow", "plaintext", "skia"]
 
     @classmethod
     def __get_validators__(cls) -> Generator[Callable[..., Any], None, None]:
@@ -21,6 +21,8 @@ def __get_validators__(cls) -> Generator[Callable[..., Any], None, None]:
     def validate(cls, value: str) -> str:
         if value.lower() == "pillow":
             raise NotImplementedError("Pillow render is not implemented yet")
+        if value.lower() == "skia":
+            raise NotImplementedError("Skia render is not implemented yet")
         if value.lower() not in cls.ALLOWED_VALUES:
             raise ValueError(
                 f"Invalid type: {value!r}, must be one of {cls.ALLOWED_VALUES}"

nonebot_plugin_paper/utils.py

@@ -26,22 +26,22 @@ def text_paper_info(paper: Paper) -> str:
     )
 
     template = (
-        f"📄 **Title:** {paper.info.title}\n"
-        f"👥 **Authors:** {authors}\n"
-        f"🏷️ **Categories:** {paper.info.categories.primary.term}\n"
-        f"📅 **Published Date:** {published_date}\n"
-        f"🔄 **Updated Date:** {updated_date}\n"
-        f"📝 **Summary:** {summary}\n"
+        f"📄 Title: {paper.info.title}\n"
+        f"👥 Authors: {authors}\n"
+        f"🏷️ Categories: {paper.info.categories.primary.term}\n"
+        f"📅 Published Date: {published_date}\n"
+        f"🔄 Updated Date: {updated_date}\n"
+        f"📝 Summary: {summary}\n"
     )
 
     if paper.doi:
-        template += f"🔗 **DOI:** {paper.doi}\n"
+        template += f"🔗 DOI: {paper.doi}\n"
     if paper.journal_ref:
-        template += f"📚 **Journal Reference:** {paper.journal_ref}\n"
+        template += f"📚 Journal Reference: {paper.journal_ref}\n"
     if paper.pdf_url:
-        template += f"📥 **PDF Download Link:** {paper.pdf_url}\n"
+        template += f"📥 PDF Download Link: {paper.pdf_url}\n"
     if paper.comment:
-        template += f"💬 **Comment:** {paper.comment}\n"
+        template += f"💬 Comment: {paper.comment}\n"
 
     return template