diff options
| author | 2025-10-25 00:30:48 +0800 | |
|---|---|---|
| committer | 2025-10-25 00:30:48 +0800 | |
| commit | cbc653ffd0ea9abf4360623dc7a7651e1a49cc61 (patch) | |
| tree | ea3c396148158077bae3e77eaa9341f8c1990636 /src/conventionalrp/renderers/markdown_renderer.py | |
| parent | 08299b37dfda86e56e4f2b442f68ccd2da7a82e3 (diff) | |
| download | conventional_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.py | 176 |
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. |
