#!/usr/bin/env python3 """ OneRoll interactive dice roll program Supports command line parameters and interactive mode. Example of usage: # Direct throw python -m oneroll "3d6 + 2" # Interactive mode python -m oneroll # Statistical Mode python -m oneroll --stats "3d6" --times 100 """ import sys import argparse from typing import List, Dict, Any from rich.console import Console from rich.table import Table from rich.panel import Panel from rich.text import Text from rich.prompt import Prompt, Confirm from rich.progress import Progress, SpinnerColumn, TextColumn from . import OneRoll, roll, roll_simple, roll_multiple, roll_statistics, CommonRolls console = Console() class OneRollCLI: """OneRoll command line interface""" def __init__(self): self.roller = OneRoll() self.history: List[Dict[str, Any]] = [] def print_result(self, result: Dict[str, Any], expression: str = None): """Pretty print the dice roll result""" if expression is None: expression = result.get('expression', 'Unknown') # Create result panel total = result['total'] details = result['details'] rolls = result['rolls'] # Select color based on result value if total >= 15: color = "green" elif total >= 10: color = "yellow" else: color = "red" # Build display text text = Text() text.append(f"🎲 {expression}\n", style="bold blue") text.append(f"总点数: ", style="bold") text.append(f"{total}", style=f"bold {color}") text.append(f"\n详情: {details}", style="white") if rolls: text.append(f"\n投掷结果: ", style="bold") text.append(f"{rolls}", style="cyan") # Display comment comment = result.get("comment", "") if comment: text.append(f"\n注释: ", style="bold") text.append(f"{comment}", style="italic blue") panel = Panel(text, title="Dice Roll Result", border_style=color) console.print(panel) def print_statistics(self, stats: Dict[str, Any], expression: str): """Print statistics information""" table = Table(title=f"统计结果: {expression} (投掷 {stats['count']} 次)") table.add_column("统计项", style="cyan") table.add_column("数值", style="green") table.add_row("最小值", str(stats['min'])) table.add_row("最大值", str(stats['max'])) table.add_row("平均值", f"{stats['mean']:.2f}") table.add_row("总和", str(stats['total'])) table.add_row("投掷次数", str(stats['count'])) console.print(table) def print_history(self): """Print dice roll history""" if not self.history: console.print("暂无投掷历史", style="yellow") return table = Table(title="投掷历史") table.add_column("序号", style="cyan") table.add_column("表达式", style="green") table.add_column("总点数", style="yellow") table.add_column("注释", style="blue") table.add_column("详情", style="white") for i, result in enumerate(self.history[-10:], 1): # only show last 30 times table.add_row( str(i), result.get('expression', 'Unknown'), str(result['total']), result.get('comment', ''), result['details'] ) console.print(table) def show_help(self): """show help information""" help_text = """ 🎲 OneRoll 骰子表达式解析器 支持的表达式格式： • 基本骰子: 3d6, 1d20, 2d10 • 数学运算: 3d6 + 2, 2d6 * 3, (2d6 + 3) * 2 • 修饰符: - ! 爆炸骰子: 2d6! - kh 取高: 4d6kh3 - kl 取低: 4d6kl2 - dh 丢弃高: 5d6dh1 - dl 丢弃低: 5d6dl1 - r 重投: 3d6r1 - ro 条件重投: 4d6ro1 • 注释: 在表达式末尾使用 # 添加注释 常用命令： • help - 显示帮助 • history - 显示投掷历史 • stats <表达式> <次数> - 统计投掷 • clear - 清空历史 • quit/exit - 退出程序 常用表达式： • d20 - 1d20 • advantage - 2d20kh1 (优势) • disadvantage - 2d20kl1 (劣势) • attribute - 4d6kh3 (属性投掷) """ console.print(Panel(help_text, title="帮助信息", border_style="blue")) def interactive_mode(self): """Interactive Mode""" console.print(Panel.fit("🎲 OneRoll 交互式掷骰程序", style="bold blue")) console.print("输入 'help' 查看帮助，输入 'quit' 退出程序\n") while True: try: user_input = Prompt.ask("🎲 请输入骰子表达式").strip() if not user_input: continue # handle special command if user_input.lower() in ['quit', 'exit', 'q']: if Confirm.ask("确定要退出吗？"): break continue if user_input.lower() == 'help': self.show_help() continue if user_input.lower() == 'history': self.print_history() continue if user_input.lower() == 'clear': self.history.clear() console.print("历史已清空", style="green") continue if user_input.lower().startswith('stats '): parts = user_input.split() if len(parts) >= 3: expression = parts[1] try: times = int(parts[2]) if times > 1000: console.print("统计次数不能超过1000次", style="red") continue with Progress( SpinnerColumn(), TextColumn("[progress.description]{task.description}"), console=console ) as progress: task = progress.add_task(f"正在统计 {expression}...", total=None) stats = roll_statistics(expression, times) progress.stop() self.print_statistics(stats, expression) except ValueError: console.print("统计次数必须是数字", style="red") except Exception as e: console.print(f"统计错误: {e}", style="red") else: console.print("用法: stats <表达式> <次数>", style="red") continue # resolve regular expression's alias expression = self._resolve_expression_alias(user_input) # execute roll try: result = roll(expression) self.history.append(result) self.print_result(result, user_input) except Exception as e: console.print(f"错误: {e}", style="red") except KeyboardInterrupt: if Confirm.ask("\n确定要退出吗？"): break except EOFError: break def _resolve_expression_alias(self, user_input: str) -> str: """resolve regular expression's alias""" aliases = { 'd20': CommonRolls.D20, 'advantage': CommonRolls.D20_ADVANTAGE, 'disadvantage': CommonRolls.D20_DISADVANTAGE, 'attr': CommonRolls.ATTRIBUTE_ROLL, 'attribute': CommonRolls.ATTRIBUTE_ROLL, } return aliases.get(user_input.lower(), user_input) def run(self, args): """run tui mode""" if args.tui: # start tui try: from .tui import run_tui run_tui() except ImportError: console.print("TUI 模式需要安装 textual: pip install textual", style="red") sys.exit(1) except Exception as e: console.print(f"TUI 启动失败: {e}", style="red") sys.exit(1) elif args.expression: # single roll mode try: result = roll(args.expression) self.print_result(result) except Exception as e: console.print(f"错误: {e}", style="red") sys.exit(1) elif args.stats: # stats mode try: times = args.times or 100 if times > 10000: console.print("统计次数不能超过10000次", style="red") sys.exit(1) with Progress( SpinnerColumn(), TextColumn("[progress.description]{task.description}"), console=console ) as progress: task = progress.add_task(f"正在统计 {args.stats}...", total=None) stats = roll_statistics(args.stats, times) progress.stop() self.print_statistics(stats, args.stats) except Exception as e: console.print(f"错误: {e}", style="red") sys.exit(1) else: # interactive mode self.interactive_mode() def main(): parser = argparse.ArgumentParser( description="OneRoll 骰子表达式解析器", formatter_class=argparse.RawDescriptionHelpFormatter, epilog=""" 使用示例: %(prog)s "3d6 + 2" # 单次投掷 %(prog)s --stats "3d6" --times 100 # 统计模式 %(prog)s --tui # 终端用户界面 %(prog)s # 交互式模式 """ ) parser.add_argument( 'expression', nargs='?', help='骰子表达式，如 "3d6 + 2"' ) parser.add_argument( '--stats', help='统计模式，指定要统计的表达式' ) parser.add_argument( '--times', type=int, default=100, help='统计次数，默认100次' ) parser.add_argument( '--version', action='version', version='OneRoll 0.0.1' ) parser.add_argument( '--tui', action='store_true', help='启动终端用户界面 (TUI)' ) args = parser.parse_args() cli = OneRollCLI() cli.run(args) if __name__ == '__main__': main()