aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/src/conventionalrp/renderers/markdown_renderer.py
diff options
context:
space:
mode:
author简律纯 <i@jyunko.cn>2025-10-25 00:30:48 +0800
committer简律纯 <i@jyunko.cn>2025-10-25 00:30:48 +0800
commitcbc653ffd0ea9abf4360623dc7a7651e1a49cc61 (patch)
treeea3c396148158077bae3e77eaa9341f8c1990636 /src/conventionalrp/renderers/markdown_renderer.py
parent08299b37dfda86e56e4f2b442f68ccd2da7a82e3 (diff)
downloadconventional_role_play-cbc653ffd0ea9abf4360623dc7a7651e1a49cc61.tar.gz
conventional_role_play-cbc653ffd0ea9abf4360623dc7a7651e1a49cc61.zip
feat: Implement plugin system with combat tracker and dice analyzer
- Added `plugin_system_demo.py` to demonstrate basic plugin usage, processing, and analysis. - Created `CombatTrackerPlugin` for tracking combat statistics including damage and healing. - Developed `DiceAnalyzerPlugin` for analyzing dice rolls and calculating success rates. - Introduced `renderer_demo.py` for rendering output in HTML, Markdown, and JSON formats. - Implemented `rule_system_demo.py` to showcase rule engine capabilities with various examples. - Established core rule engine functionality in `rules.py` with support for conditions and actions. - Enhanced base plugin structure in `base.py` to support different plugin types (Processor, Renderer, Analyzer). - Added custom exception handling in `exceptions.py` for better error management. - Configured logging setup in `logging_config.py` for improved logging capabilities. - Created unit tests in `test_rust_core.py` to validate core functionalities and performance.
Diffstat (limited to 'src/conventionalrp/renderers/markdown_renderer.py')
-rw-r--r--src/conventionalrp/renderers/markdown_renderer.py176
1 files changed, 139 insertions, 37 deletions
diff --git a/src/conventionalrp/renderers/markdown_renderer.py b/src/conventionalrp/renderers/markdown_renderer.py
index 9df59a2..7080913 100644
--- a/src/conventionalrp/renderers/markdown_renderer.py
+++ b/src/conventionalrp/renderers/markdown_renderer.py
@@ -1,18 +1,19 @@
+"""
+Markdown 渲染器
+"""
+
+from typing import Any, Dict, List, Union
from .base import BaseRenderer
-from typing import List, Dict, Any, Union
class MarkdownRenderer(BaseRenderer):
+ def __init__(self, enable_syntax_hints: bool = True, enable_emoji: bool = True):
+ super().__init__()
+ self.enable_syntax_hints = enable_syntax_hints
+ self.enable_emoji = enable_emoji
+ self.style = {}
+
def render(self, data: Union[List[Dict[str, Any]], Dict[str, Any]]) -> str:
- """
- Renders the given data in Markdown format.
-
- Args:
- data: The data to render (can be list or dict).
-
- Returns:
- str: The rendered Markdown string.
- """
if isinstance(data, list):
return self._render_list(data)
elif isinstance(data, dict):
@@ -21,40 +22,141 @@ class MarkdownRenderer(BaseRenderer):
return str(data)
def _render_list(self, data: List[Dict[str, Any]]) -> str:
- """渲染列表数据为 Markdown"""
- markdown_output = "# TRPG Log\n\n"
-
- for i, entry in enumerate(data, 1):
- if entry.get("type") == "metadata":
- markdown_output += f"## Entry {i}\n\n"
- markdown_output += f"**Timestamp**: {entry.get('timestamp', 'N/A')} \n"
- markdown_output += f"**Speaker**: {entry.get('speaker', 'N/A')} \n\n"
-
- content_items = entry.get("content", [])
- if content_items:
- markdown_output += "**Content**:\n\n"
- for content in content_items:
- content_type = content.get("type", "unknown")
- content_text = content.get("content", "")
- markdown_output += f"- [{content_type}] {content_text}\n"
- markdown_output += "\n"
- else:
- markdown_output += f"- {entry}\n"
+ if data and not isinstance(data[0], dict):
+ data = [{"type": "text", "content": str(item)} for item in data]
- return markdown_output
+ stats = self._calculate_stats(data)
+
+ emoji_prefix = "🎲 " if self.enable_emoji else ""
+ md = f"# {emoji_prefix}\n\n"
+
+ md += "## Statistics\n\n"
+ md += "| Type | Count |\n"
+ md += "|------|------|\n"
+ md += f"| Total | {stats['total']} |\n"
+ md += f"| Dialogue | {stats['dialogue']} |\n"
+ md += f"| Dice | {stats['dice']} |\n"
+ md += f"| Narration | {stats['narration']} |\n"
+ md += f"| System | {stats['system']} |\n"
+ md += f"| Metadata | {stats['metadata']} |\n\n"
+
+ md += "---\n\n"
+ md += "## Detailed Log\n\n"
+
+ # Render each token
+ for i, item in enumerate(data, 1):
+ md += self._render_token(item, i)
+ md += "\n"
+
+ md += "---\n\n"
+ md += "_Generated by ConventionalRP Markdown Renderer_\n"
+
+ return md
+
+ def _render_token(self, token: Dict[str, Any], index: int) -> str:
+ token_type = token.get("type", "unknown")
+
+ # 处理元数据类型(向后兼容)
+ if token_type == "metadata":
+ return self._render_metadata_token(token, index)
+
+ speaker = token.get("speaker", "")
+ content = str(token.get("content", ""))
+
+ type_emojis = {
+ "dialogue": "💬",
+ "dice": "🎲",
+ "narration": "📖",
+ "system": "⚙️",
+ "success": "✅",
+ "failure": "❌",
+ "text": "📄",
+ "unknown": "❓",
+ }
+
+ emoji = type_emojis.get(token_type, "•") if self.enable_emoji else "-"
+
+ md = f"### {emoji} [{index}] {token_type.upper()}\n\n"
+
+ if speaker:
+ md += f"**说话者:** {speaker}\n\n"
+
+ if token_type == "dice" and self.enable_syntax_hints:
+ md += f"```dice\n{content}\n```\n\n"
+
+ if "result" in token:
+ md += f"**结果:** `{token['result']}`\n\n"
+ else:
+ md += f"{content}\n\n"
+
+ metadata_lines = []
+
+ if "timestamp" in token:
+ metadata_lines.append(f"Timestamp: `{token['timestamp']}`")
+
+ if "tags" in token and token["tags"]:
+ tags_str = " ".join([f"`{tag}`" for tag in token["tags"]])
+ metadata_lines.append(f"Tags: {tags_str}")
+
+ if "combat_data" in token:
+ combat = token["combat_data"]
+ if combat.get("type") == "damage":
+ metadata_lines.append(f"Damage: **{combat['amount']}** (Total: {combat.get('total_damage', '?')})")
+ elif combat.get("type") == "healing":
+ metadata_lines.append(f"Healing: **{combat['amount']}** (Total: {combat.get('total_healing', '?')})")
+
+ if metadata_lines:
+ md += "> " + "\n> ".join(metadata_lines) + "\n\n"
+
+ return md
+
+ def _render_metadata_token(self, token: Dict[str, Any], index: int) -> str:
+ """渲染元数据类型的 token(向后兼容)"""
+ md = f"### Entry {index}\n\n"
+ md += f"**Timestamp**: {token.get('timestamp', 'N/A')} \n"
+ md += f"**Speaker**: {token.get('speaker', 'N/A')} \n\n"
+
+ content_items = token.get("content", [])
+ if content_items:
+ md += "**Content**:\n\n"
+ for content in content_items:
+ content_type = content.get("type", "unknown")
+ content_text = content.get("content", "")
+ md += f"- [{content_type}] {content_text}\n"
+ md += "\n"
+
+ return md
def _render_dict(self, data: Dict[str, Any]) -> str:
- """渲染字典数据为 Markdown"""
markdown_output = ""
for key, value in data.items():
markdown_output += f"## {key}\n\n{value}\n\n"
return markdown_output
+
+ def _calculate_stats(self, data: List[Dict[str, Any]]) -> Dict[str, int]:
+ stats = {
+ "total": len(data),
+ "dialogue": 0,
+ "dice": 0,
+ "narration": 0,
+ "system": 0,
+ "metadata": 0,
+ }
+
+ for item in data:
+ token_type = item.get("type", "unknown")
+ if token_type in stats:
+ stats[token_type] += 1
+
+ return stats
def set_style(self, style):
- """
- Sets the style for the Markdown renderer.
+ self.style = style
+
+ # 从 style 中提取设置
+ if isinstance(style, dict):
+ if "emoji" in style:
+ self.enable_emoji = style["emoji"]
+ if "syntax_hints" in style:
+ self.enable_syntax_hints = style["syntax_hints"]
- Args:
- style (dict): A dictionary of style options.
- """
- self.style = style # Currently, Markdown does not support styling, but this can be extended.