mofox-plugin-dev-toolkit 0.3.3__py3-none-any.whl

mpdt/utils/plugin_parser.py

@@ -0,0 +1,195 @@
+"""
+插件名称解析器
+使用 AST 解析插件文件，提取运行时插件名称
+"""
+
+import ast
+from pathlib import Path
+
+
+def extract_plugin_name(plugin_path: Path) -> str | None:
+    """从插件目录提取运行时插件名称
+
+    Args:
+        plugin_path: 插件目录路径
+
+    Returns:
+        插件名称，如果解析失败返回 None
+
+    Example:
+        >>> extract_plugin_name(Path("my_awesome_plugin"))
+        "awesome_plugin"  # 从 plugin.py 中的 plugin_name 属性读取
+    """
+    plugin_file = plugin_path / "plugin.py"
+
+    if not plugin_file.exists():
+        return None
+
+    try:
+        with open(plugin_file, encoding="utf-8") as f:
+            source = f.read()
+
+        tree = ast.parse(source)
+
+        # 查找 BasePlugin 的子类
+        for node in ast.walk(tree):
+            if isinstance(node, ast.ClassDef):
+                # 检查是否继承自 BasePlugin
+                is_base_plugin = False
+                for base in node.bases:
+                    if isinstance(base, ast.Name) and base.id == "BasePlugin":
+                        is_base_plugin = True
+                        break
+
+                if not is_base_plugin:
+                    continue
+
+                # 查找 plugin_name 属性
+                for item in node.body:
+                    # 处理普通赋值: plugin_name = "xxx"
+                    if isinstance(item, ast.Assign):
+                        for target in item.targets:
+                            if isinstance(target, ast.Name) and target.id == "plugin_name":
+                                if isinstance(item.value, ast.Constant):
+                                    return item.value.value
+
+                    # 处理带类型注解的赋值: plugin_name: str = "xxx"
+                    elif isinstance(item, ast.AnnAssign):
+                        if isinstance(item.target, ast.Name) and item.target.id == "plugin_name":
+                            if item.value and isinstance(item.value, ast.Constant):
+                                return item.value.value
+
+        return None
+
+    except Exception:
+        return None
+
+
+def get_plugin_info(plugin_path: Path) -> dict:
+    """获取插件详细信息
+
+    Args:
+        plugin_path: 插件目录路径
+
+    Returns:
+        包含插件信息的字典:
+        {
+            "dir_name": "my_awesome_plugin",
+            "plugin_name": "awesome_plugin",
+            "class_name": "MyAwesomePlugin",
+            "path": "/path/to/plugin",
+            "has_plugin_file": True,
+            "parse_success": True
+        }
+    """
+    info = {
+        "dir_name": plugin_path.name,
+        "plugin_name": None,
+        "class_name": None,
+        "path": str(plugin_path.absolute()),
+        "has_plugin_file": False,
+        "parse_success": False,
+    }
+
+    plugin_file = plugin_path / "plugin.py"
+
+    if not plugin_file.exists():
+        return info
+
+    info["has_plugin_file"] = True
+
+    try:
+        with open(plugin_file, encoding="utf-8") as f:
+            source = f.read()
+
+        tree = ast.parse(source)
+
+        # 查找 BasePlugin 的子类
+        for node in ast.walk(tree):
+            if isinstance(node, ast.ClassDef):
+                # 检查是否继承自 BasePlugin
+                is_base_plugin = False
+                for base in node.bases:
+                    if isinstance(base, ast.Name) and base.id == "BasePlugin":
+                        is_base_plugin = True
+                        break
+
+                if not is_base_plugin:
+                    continue
+
+                info["class_name"] = node.name
+
+                # 查找 plugin_name 属性
+                for item in node.body:
+                    # 处理普通赋值: plugin_name = "xxx"
+                    if isinstance(item, ast.Assign):
+                        for target in item.targets:
+                            if isinstance(target, ast.Name) and target.id == "plugin_name":
+                                if isinstance(item.value, ast.Constant):
+                                    info["plugin_name"] = item.value.value
+                                    info["parse_success"] = True
+                                    break
+
+                    # 处理带类型注解的赋值: plugin_name: str = "xxx"
+                    elif isinstance(item, ast.AnnAssign):
+                        if isinstance(item.target, ast.Name) and item.target.id == "plugin_name":
+                            if item.value and isinstance(item.value, ast.Constant):
+                                info["plugin_name"] = item.value.value
+                                info["parse_success"] = True
+                                break
+
+                # 找到 BasePlugin 子类后跳出
+                if info["class_name"]:
+                    break
+
+        return info
+
+    except Exception as e:
+        info["error"] = str(e)
+        return info
+
+
+def validate_plugin_structure(plugin_path: Path) -> tuple[bool, list[str]]:
+    """验证插件目录结构
+
+    Args:
+        plugin_path: 插件目录路径
+
+    Returns:
+        (是否有效, 错误/警告消息列表)
+    """
+    messages = []
+
+    if not plugin_path.is_dir():
+        return False, ["路径不是一个目录"]
+
+    # 检查必需文件
+    required_files = {"plugin.py": "插件主文件", "__init__.py": "包初始化文件"}
+
+    for filename, description in required_files.items():
+        file_path = plugin_path / filename
+        if not file_path.exists():
+            messages.append(f"缺少 {description}: {filename}")
+
+    # 如果缺少必需文件，直接返回
+    if messages:
+        return False, messages
+
+    # 检查 plugin.py 是否可以解析
+    plugin_name = extract_plugin_name(plugin_path)
+    if not plugin_name:
+        messages.append("无法从 plugin.py 中提取 plugin_name")
+        messages.append("请确保有一个继承自 BasePlugin 的类，并定义了 plugin_name 属性")
+        return False, messages
+
+    # 检查推荐文件
+    recommended_files = {"config.toml": "配置文件", "README.md": "说明文档"}
+
+    for filename, description in recommended_files.items():
+        file_path = plugin_path / filename
+        if not file_path.exists():
+            messages.append(f"建议添加 {description}: {filename}")
+
+    # 如果只有建议性消息，仍然返回有效
+    has_errors = any("缺少" in msg or "无法" in msg for msg in messages)
+    return not has_errors, messages

mpdt/utils/template_engine.py

@@ -0,0 +1,112 @@
+"""
+模板引擎
+"""
+
+from pathlib import Path
+from typing import Any
+
+from jinja2 import Environment, FileSystemLoader, Template
+
+
+class TemplateEngine:
+    """模板引擎类"""
+
+    def __init__(self, template_dir: Path | str | None = None):
+        """
+        初始化模板引擎
+
+        Args:
+            template_dir: 模板目录路径
+        """
+        if template_dir:
+            self.template_dir = Path(template_dir)
+            self.env = Environment(
+                loader=FileSystemLoader(str(self.template_dir)),
+                trim_blocks=True,
+                lstrip_blocks=True,
+            )
+        else:
+            # 使用默认模板目录
+            default_template_dir = Path(__file__).parent.parent / "templates"
+            self.template_dir = default_template_dir
+
+            if default_template_dir.exists():
+                self.env = Environment(
+                    loader=FileSystemLoader(str(default_template_dir)),
+                    trim_blocks=True,
+                    lstrip_blocks=True,
+                )
+            else:
+                self.env = None
+
+    def render_string(self, template_string: str, context: dict[str, Any]) -> str:
+        """
+        渲染字符串模板
+
+        Args:
+            template_string: 模板字符串
+            context: 模板变量
+
+        Returns:
+            渲染后的字符串
+        """
+        template = Template(template_string, trim_blocks=True, lstrip_blocks=True)
+        return template.render(**context)
+
+    def render_file(self, template_name: str, context: dict[str, Any]) -> str:
+        """
+        渲染文件模板
+
+        Args:
+            template_name: 模板文件名
+            context: 模板变量
+
+        Returns:
+            渲染后的字符串
+        """
+        if self.env is None:
+            raise ValueError("模板目录不存在，无法加载文件模板")
+
+        template = self.env.get_template(template_name)
+        return template.render(**context)
+
+    def render_to_file(self, template_name: str, context: dict[str, Any], output_path: Path | str) -> None:
+        """
+        渲染模板并写入文件
+
+        Args:
+            template_name: 模板文件名
+            context: 模板变量
+            output_path: 输出文件路径
+        """
+        content = self.render_file(template_name, context)
+        output_path = Path(output_path)
+        output_path.parent.mkdir(parents=True, exist_ok=True)
+        output_path.write_text(content, encoding="utf-8")
+
+
+def prepare_common_context(**kwargs: Any) -> dict[str, Any]:
+    """
+    准备通用模板变量
+
+    Args:
+        **kwargs: 额外的变量
+
+    Returns:
+        模板变量字典
+    """
+    from datetime import datetime
+
+    from mpdt.utils.file_ops import get_git_user_info
+
+    git_info = get_git_user_info()
+
+    context = {
+        "timestamp": datetime.now().isoformat(),
+        "year": datetime.now().year,
+        "author": git_info.get("name", ""),
+        "author_email": git_info.get("email", ""),
+    }
+
+    context.update(kwargs)
+    return context

mpdt/validators/__init__.py

@@ -0,0 +1,26 @@
+"""
+验证器模块
+"""
+
+from .auto_fix_validator import AutoFixValidator
+from .base import BaseValidator, ValidationIssue, ValidationLevel, ValidationResult
+from .component_validator import ComponentValidator
+from .config_validator import ConfigValidator
+from .metadata_validator import MetadataValidator
+from .structure_validator import StructureValidator
+from .style_validator import StyleValidator
+from .type_validator import TypeValidator
+
+__all__ = [
+    "BaseValidator",
+    "ValidationResult",
+    "ValidationIssue",
+    "ValidationLevel",
+    "StructureValidator",
+    "MetadataValidator",
+    "ComponentValidator",
+    "ConfigValidator",
+    "StyleValidator",
+    "TypeValidator",
+    "AutoFixValidator",
+]