chartflow 0.1__py3-none-any.whl

chartflow/edit/syntax_highlighter.py

@@ -0,0 +1,291 @@
+"""
+Python语法高亮器
+基于QSyntaxHighlighter实现
+"""
+
+from PyQt6.QtCore import QRegularExpression
+from PyQt6.QtGui import (
+    QSyntaxHighlighter, QTextCharFormat, QColor,
+    QFont, QTextDocument
+)
+from typing import List, Tuple, Pattern
+import re
+
+
+class PythonSyntaxHighlighter(QSyntaxHighlighter):
+    """Python语法高亮器 - 浅色主题"""
+    
+    # 浅色语法配色
+    COLORS = {
+        'keyword': '#0000FF',      # 蓝色 - 关键字
+        'string': '#008000',       # 绿色 - 字符串
+        'comment': '#808080',      # 灰色 - 注释
+        'function': '#795E26',     # 棕色 - 函数名
+        'class': '#267F99',        # 青色 - 类名
+        'number': '#098658',       # 深绿色 - 数字
+        'operator': '#000000',     # 黑色 - 运算符
+        'builtin': '#0000FF',      # 蓝色 - 内置函数
+        'decorator': '#795E26',    # 棕色 - 装饰器
+        'self': '#0000FF',         # 蓝色 - self
+        'f_string_expr': '#001080', # 深蓝色 - f-string表达式
+    }
+    
+    # Python关键字
+    KEYWORDS = [
+        'and', 'as', 'assert', 'async', 'await', 'break', 'class', 'continue',
+        'def', 'del', 'elif', 'else', 'except', 'False', 'finally', 'for',
+        'from', 'global', 'if', 'import', 'in', 'is', 'lambda', 'None',
+        'nonlocal', 'not', 'or', 'pass', 'raise', 'return', 'True', 'try',
+        'while', 'with', 'yield'
+    ]
+    
+    # Python内置函数
+    BUILTINS = [
+        'abs', 'all', 'any', 'bin', 'bool', 'bytearray', 'bytes', 'callable',
+        'chr', 'classmethod', 'compile', 'complex', 'delattr', 'dict', 'dir',
+        'divmod', 'enumerate', 'eval', 'exec', 'filter', 'float', 'format',
+        'frozenset', 'getattr', 'globals', 'hasattr', 'hash', 'help', 'hex',
+        'id', 'input', 'int', 'isinstance', 'issubclass', 'iter', 'len',
+        'list', 'locals', 'map', 'max', 'memoryview', 'min', 'next', 'object',
+        'oct', 'open', 'ord', 'pow', 'print', 'property', 'range', 'repr',
+        'reversed', 'round', 'set', 'setattr', 'slice', 'sorted', 'staticmethod',
+        'str', 'sum', 'super', 'tuple', 'type', 'vars', 'zip', '__import__'
+    ]
+    
+    def __init__(self, parent: QTextDocument = None):
+        super().__init__(parent)
+        self._setup_formats()
+        self._setup_rules()
+    
+    def _setup_formats(self):
+        """设置文本格式"""
+        # 关键字格式
+        self.keyword_format = QTextCharFormat()
+        self.keyword_format.setForeground(QColor(self.COLORS['keyword']))
+        self.keyword_format.setFontWeight(QFont.Weight.Bold)
+        
+        # 字符串格式
+        self.string_format = QTextCharFormat()
+        self.string_format.setForeground(QColor(self.COLORS['string']))
+        
+        # 多行字符串格式
+        self.multiline_string_format = QTextCharFormat()
+        self.multiline_string_format.setForeground(QColor(self.COLORS['string']))
+        
+        # 注释格式
+        self.comment_format = QTextCharFormat()
+        self.comment_format.setForeground(QColor(self.COLORS['comment']))
+        self.comment_format.setFontItalic(True)
+        
+        # 函数定义格式
+        self.function_format = QTextCharFormat()
+        self.function_format.setForeground(QColor(self.COLORS['function']))
+        
+        # 类定义格式
+        self.class_format = QTextCharFormat()
+        self.class_format.setForeground(QColor(self.COLORS['class']))
+        
+        # 数字格式
+        self.number_format = QTextCharFormat()
+        self.number_format.setForeground(QColor(self.COLORS['number']))
+        
+        # 运算符格式
+        self.operator_format = QTextCharFormat()
+        self.operator_format.setForeground(QColor(self.COLORS['operator']))
+        
+        # 内置函数格式
+        self.builtin_format = QTextCharFormat()
+        self.builtin_format.setForeground(QColor(self.COLORS['builtin']))
+        
+        # 装饰器格式
+        self.decorator_format = QTextCharFormat()
+        self.decorator_format.setForeground(QColor(self.COLORS['decorator']))
+        
+        # self格式
+        self.self_format = QTextCharFormat()
+        self.self_format.setForeground(QColor(self.COLORS['self']))
+        self.self_format.setFontWeight(QFont.Weight.Bold)
+        
+        # f-string表达式格式
+        self.f_string_expr_format = QTextCharFormat()
+        self.f_string_expr_format.setForeground(QColor(self.COLORS['f_string_expr']))
+        self.f_string_expr_format.setFontWeight(QFont.Weight.Bold)
+    
+    def _setup_rules(self):
+        """设置高亮规则"""
+        self.rules: List[Tuple[QRegularExpression, QTextCharFormat]] = []
+        
+        # 关键字规则
+        keyword_pattern = r'\b(' + '|'.join(self.KEYWORDS) + r')\b'
+        self.rules.append((QRegularExpression(keyword_pattern), self.keyword_format))
+        
+        # self规则
+        self.rules.append((QRegularExpression(r'\bself\b'), self.self_format))
+        
+        # 内置函数规则
+        builtin_pattern = r'\b(' + '|'.join(self.BUILTINS) + r')(?=\s*\()'
+        self.rules.append((QRegularExpression(builtin_pattern), self.builtin_format))
+        
+        # 函数定义规则
+        self.rules.append((
+            QRegularExpression(r'\bdef\s+(\w+)'),
+            self.function_format
+        ))
+        
+        # 类定义规则
+        self.rules.append((
+            QRegularExpression(r'\bclass\s+(\w+)'),
+            self.class_format
+        ))
+        
+        # 装饰器规则
+        self.rules.append((
+            QRegularExpression(r'@[\w.]+'),
+            self.decorator_format
+        ))
+        
+        # 数字规则
+        number_pattern = r'\b\d+\.?\d*([eE][+-]?\d+)?\b|\b0[xX][0-9a-fA-F]+\b|\b0[oO]?[0-7]+\b|\b0[bB][01]+\b'
+        self.rules.append((QRegularExpression(number_pattern), self.number_format))
+        
+        # 双引号字符串规则
+        self.rules.append((
+            QRegularExpression(r'"[^"\\]*(\\.[^"\\]*)*"'),
+            self.string_format
+        ))
+        
+        # 单引号字符串规则
+        self.rules.append((
+            QRegularExpression(r"'[^'\\]*(\\.[^'\\]*)*'"),
+            self.string_format
+        ))
+        
+        # f-string规则 (简化处理)
+        self.rules.append((
+            QRegularExpression(r'f"[^"\\]*(\\.[^"\\]*)*"'),
+            self.string_format
+        ))
+        self.rules.append((
+            QRegularExpression(r"f'[^'\\]*(\\.[^'\\]*)*'"),
+            self.string_format
+        ))
+        
+        # 单行注释规则
+        self.rules.append((
+            QRegularExpression(r'#[^\n]*'),
+            self.comment_format
+        ))
+    
+    def highlightBlock(self, text: str):
+        """高亮文本块"""
+        # 应用基本规则
+        for pattern, fmt in self.rules:
+            match_iterator = pattern.globalMatch(text)
+            while match_iterator.hasNext():
+                match = match_iterator.next()
+                # 对于函数和类定义，只高亮名称部分
+                if pattern.pattern().startswith(r'\bdef\s+') or \
+                   pattern.pattern().startswith(r'\bclass\s+'):
+                    # 获取捕获组（函数/类名）
+                    if match.lastCapturedIndex() >= 1:
+                        start = match.capturedStart(1)
+                        length = match.capturedLength(1)
+                    else:
+                        start = match.capturedStart()
+                        length = match.capturedLength()
+                else:
+                    start = match.capturedStart()
+                    length = match.capturedLength()
+                
+                self.setFormat(start, length, fmt)
+        
+        # 处理多行字符串
+        self._highlight_multiline_strings(text)
+        
+        # 设置当前块状态用于多行字符串跟踪
+        self.setCurrentBlockState(0)
+    
+    def _highlight_multiline_strings(self, text: str):
+        """高亮多行字符串"""
+        # 三引号字符串检测
+        triple_double = '"""'
+        triple_single = "'''"
+        
+        # 检查是否在多行字符串中
+        in_multiline = self.previousBlockState() == 1
+        
+        i = 0
+        while i < len(text):
+            if not in_multiline:
+                # 查找三引号开始
+                dd_pos = text.find(triple_double, i)
+                ds_pos = text.find(triple_single, i)
+                
+                start_pos = -1
+                is_double = True
+                
+                if dd_pos != -1 and (ds_pos == -1 or dd_pos < ds_pos):
+                    start_pos = dd_pos
+                    is_double = True
+                elif ds_pos != -1:
+                    start_pos = ds_pos
+                    is_double = False
+                
+                if start_pos != -1:
+                    # 查找结束
+                    end_marker = triple_double if is_double else triple_single
+                    end_pos = text.find(end_marker, start_pos + 3)
+                    
+                    if end_pos != -1:
+                        # 单行多行字符串
+                        length = end_pos - start_pos + 3
+                        self.setFormat(start_pos, length, self.multiline_string_format)
+                        i = end_pos + 3
+                    else:
+                        # 多行开始
+                        length = len(text) - start_pos
+                        self.setFormat(start_pos, length, self.multiline_string_format)
+                        in_multiline = True
+                        self.setCurrentBlockState(1)
+                        break
+                else:
+                    break
+            else:
+                # 在多行字符串中，查找结束
+                end_pos_dd = text.find(triple_double, i)
+                end_pos_ds = text.find(triple_single, i)
+                
+                # 需要知道是哪种三引号
+                # 简化处理：检查哪个先出现
+                end_pos = -1
+                if end_pos_dd != -1 and end_pos_ds != -1:
+                    end_pos = min(end_pos_dd, end_pos_ds)
+                elif end_pos_dd != -1:
+                    end_pos = end_pos_dd
+                elif end_pos_ds != -1:
+                    end_pos = end_pos_ds
+                
+                if end_pos != -1:
+                    # 多行字符串结束
+                    length = end_pos + 3
+                    self.setFormat(0, length, self.multiline_string_format)
+                    in_multiline = False
+                    self.setCurrentBlockState(0)
+                    i = end_pos + 3
+                else:
+                    # 仍在多行字符串中
+                    self.setFormat(0, len(text), self.multiline_string_format)
+                    self.setCurrentBlockState(1)
+                    break
+
+
+class HighlightRule:
+    """高亮规则辅助类"""
+    def __init__(self, pattern: str, color: str, bold: bool = False, italic: bool = False):
+        self.pattern = QRegularExpression(pattern)
+        self.format = QTextCharFormat()
+        self.format.setForeground(QColor(color))
+        if bold:
+            self.format.setFontWeight(QFont.Weight.Bold)
+        if italic:
+            self.format.setFontItalic(True)

chartflow/edit/test_editor.py

@@ -0,0 +1,190 @@
+"""
+PyEdit测试脚本
+"""
+
+from chartflow.edit import PythonEditor, PythonEditorFactory
+from chartflow.edit import CodeEditor
+from chartflow.edit import PythonSyntaxHighlighter
+from chartflow.edit import CodeCompleter, SmartCompleter, CompletionItem
+from chartflow.edit import ContextAnalyzer, CodeContext
+from chartflow.edit import CodeFolder, FoldRegion
+from chartflow.edit import LineNumberArea
+from chartflow.edit import MinimapPanel, MinimapWidget
+
+
+import sys
+from PyQt6.QtWidgets import QApplication, QWidget, QVBoxLayout, QPushButton, QLabel
+from PyQt6.QtCore import Qt
+
+# 测试导入
+try:
+    print("✓ PythonEditor 导入成功")
+except ImportError as e:
+    print(f"✗ PythonEditor 导入失败: {e}")
+    sys.exit(1)
+
+try:
+    print("✓ CodeEditor 导入成功")
+except ImportError as e:
+    print(f"✗ CodeEditor 导入失败: {e}")
+
+try:
+    print("✓ PythonSyntaxHighlighter 导入成功")
+except ImportError as e:
+    print(f"✗ PythonSyntaxHighlighter 导入失败: {e}")
+
+try:
+    print("✓ Completer 导入成功")
+except ImportError as e:
+    print(f"✗ Completer 导入失败: {e}")
+
+try:
+    print("✓ ContextAnalyzer 导入成功")
+except ImportError as e:
+    print(f"✗ ContextAnalyzer 导入失败: {e}")
+
+try:
+    print("✓ CodeFolder 导入成功")
+except ImportError as e:
+    print(f"✗ CodeFolder 导入失败: {e}")
+
+try:
+    print("✓ LineNumberArea 导入成功")
+except ImportError as e:
+    print(f"✗ LineNumberArea 导入失败: {e}")
+
+try:
+    print("✓ Minimap 导入成功")
+except ImportError as e:
+    print(f"✗ Minimap 导入失败: {e}")
+
+
+class TestWindow(QWidget):
+    """测试窗口"""
+    
+    def __init__(self):
+        super().__init__()
+        
+        self.setWindowTitle("PyEdit 功能测试")
+        self.setMinimumSize(1000, 700)
+        
+        layout = QVBoxLayout(self)
+        
+        # 说明标签
+        info = QLabel("PyEdit 功能测试 - 测试上下文代码功能")
+        info.setStyleSheet("font-size: 14px; font-weight: bold; padding: 10px;")
+        layout.addWidget(info)
+        
+        # 创建编辑器
+        self.editor = PythonEditor()
+        
+        # 设置上下文代码（模拟节点编辑器的场景）
+        code_a = '''class DialogueNode(BaseNode):
+    """对话节点"""
+    
+    NODE_TYPE = "dialogue"
+    NAME = "Dialogue"
+    
+    speaker = ""
+    text = ""
+    choices = []
+    
+    def __init__(self):
+        super().__init__()
+        self.connections = []
+        self.variables = {}
+    
+    def on_enter(self):
+        """进入节点"""
+        print(f"Entering: {self.text}")
+    
+    def on_exit(self):
+        """离开节点"""
+        print(f"Exiting: {self.text}")
+'''
+        
+        code_b = '''def handle_connection(self, context: GameContext, player: Player, *args):
+    """处理连接"""
+    result = None
+    selected_choice = None
+'''
+        
+        prefix = "        "  # 8空格缩进
+        
+        self.editor.set_context_code(code_a, code_b, prefix)
+        
+        # 设置初始代码
+        initial_code = '''# 在这里编辑代码
+# 可用的变量: context, player, result, selected_choice
+# 可用的self属性: speaker, text, choices, connections, variables
+
+# 示例：显示对话选项
+for i, choice in enumerate(self.choices):
+    print(f"{i+1}. {choice}")
+
+# 获取玩家选择
+selected_choice = player.get_input()
+
+# 根据选择设置结果
+if selected_choice:
+    result = f"choice_{selected_choice}"
+else:
+    result = "default"
+
+return result
+'''
+        
+        self.editor.set_code(initial_code)
+        layout.addWidget(self.editor)
+        
+        # 测试按钮
+        btn_layout = QVBoxLayout()
+        
+        btn_get_code = QPushButton("获取代码")
+        btn_get_code.clicked.connect(self._get_code)
+        btn_layout.addWidget(btn_get_code)
+        
+        btn_fold = QPushButton("折叠全部 (Ctrl+Shift+-)")
+        btn_fold.clicked.connect(self.editor.fold_all)
+        btn_layout.addWidget(btn_fold)
+        
+        btn_unfold = QPushButton("展开全部 (Ctrl+Shift++)")
+        btn_unfold.clicked.connect(self.editor.unfold_all)
+        btn_layout.addWidget(btn_unfold)
+        
+        layout.addLayout(btn_layout)
+        
+        # 结果显示
+        self.result_label = QLabel("结果将显示在这里")
+        self.result_label.setStyleSheet("""
+            QLabel {
+                background-color: #f0f0f0;
+                border: 1px solid #cccccc;
+                padding: 10px;
+                font-family: Consolas, monospace;
+            }
+        """)
+        layout.addWidget(self.result_label)
+    
+    def _get_code(self):
+        """获取代码"""
+        code = self.editor.get_code()
+        # 只显示前200个字符
+        display = code[:200] + "..." if len(code) > 200 else code
+        self.result_label.setText(f"代码:\n{display}")
+
+
+def main():
+    """主函数"""
+    app = QApplication(sys.argv)
+    
+    print("\n启动测试窗口...")
+    
+    window = TestWindow()
+    window.show()
+    
+    sys.exit(app.exec())
+
+
+if __name__ == '__main__':
+    main()

chartflow/flow/__init__.py

@@ -0,0 +1,31 @@
+"""
+flow - Flow chart module for chartflow
+
+Provides IChart interface and FlowChart implementation for converting
+between node JSON format and fsm Stage/Logic instances.
+"""
+
+from chartflow.flow.chart import IChart, Chart, ChartNode, ChartEdge, ChartLogic
+from chartflow.flow.qchart import QFlowChart
+
+# Editor components
+from chartflow.flow.editor import (
+    StageEditor,
+    StageEditorDialog,
+    StageEditorFactory,
+    edit_state,
+)
+
+__all__ = [
+    "IChart",
+    "Chart",
+    "ChartNode",
+    "ChartEdge",
+    "ChartLogic",
+    "QFlowChart",
+    # Editor exports
+    "StageEditor",
+    "StageEditorDialog",
+    "StageEditorFactory",
+    "edit_state",
+]

chartflow/flow/_demo.py

@@ -0,0 +1,98 @@
+from __future__ import annotations
+"""qflow._demo - Minimal demo for QFlowChart"""
+
+
+import random
+import time
+import sys
+
+from PyQt6.QtWidgets import QApplication, QMainWindow
+from PyQt6.QtCore import Qt
+
+from chartflow.flow.qchart import QFlowChart
+from chartflow.flow.chart import ChartNode, ChartEdge
+from chartflow.fsm import *
+
+# Module-level QApplication reference to prevent garbage collection
+_qapp: QApplication | None = None
+
+
+class DemoLogic(Logic):
+    """entry -> running -> exit"""
+    NAME = "DemoLogic"
+
+    def entry(self, inst):
+        return "root"
+
+    # =============== Tree ================
+    @parallel("a", "b")
+    def root(self, it):
+        return True
+
+    def a(self, it):
+        return random.random() < 0.5
+    
+    @sequence("c", "d")
+    def b(self, it):
+        time.sleep(2)
+        # print("b被执行")
+        return True
+
+    def c(self, it):                                                  
+        return random.random() < 0.6
+    
+    def d(self, it):
+        return 'record'
+    # =====================================
+
+    @recursive
+    def record(self, inst):
+        return "exit"
+
+    def exit(self, inst):
+        self._alive = False
+
+
+class DemoChart(QFlowChart):
+    """A demo chart with sample nodes and edges for testing."""
+
+    def __init__(self):
+        # Ensure QApplication exists for headless/test environments
+        global _qapp
+        if QApplication.instance() is None:
+            _qapp = QApplication(sys.argv)
+        super().__init__()
+        # Create demo nodes
+        self.addNode(ChartNode(nid="entry", name="Entry", ntype="entry", shape="ellipse"))
+        self.addNode(ChartNode(nid="state1", name="State1", ntype="state", shape="rectangle"))
+        self.addNode(ChartNode(nid="state2", name="State2", ntype="state", shape="rectangle"))
+        self.addNode(ChartNode(nid="exit", name="Exit", ntype="exit", shape="ellipse"))
+
+        # Create demo edges
+        self.addEdge(ChartEdge(eid="e1", src="entry", dst="state1", etype="flow"))
+        self.addEdge(ChartEdge(eid="e2", src="state1", dst="state2", etype="flow"))
+        self.addEdge(ChartEdge(eid="e3", src="state2", dst="exit", etype="flow"))
+        self.addEdge(ChartEdge(eid="e4", src="state2", dst="state1", etype="flow", bidirectional=True))
+
+        # Set start node
+        self.startnode = "entry"
+
+
+class DemoWindow(QMainWindow):
+    """Demo window for running QFlowChart GUI."""
+
+    def __init__(self):
+        super().__init__()
+        self.setWindowTitle("QFlowChart Demo")
+        self.setGeometry(100, 100, 1000, 700)
+
+        self.setCentralWidget(QFlowChart(DemoLogic(), parent=self))
+
+
+if __name__ == "__main__":
+    import sys
+
+    app = QApplication(sys.argv)
+    window = DemoWindow()
+    window.show()
+    sys.exit(app.exec())