flash-sim 0.0.1__py3-none-any.whl

flash/__init__.py

@@ -0,0 +1,157 @@
+"""FLASH plasma simulation package.
+
+This package provides a Python interface for the FLASH high-energy-density
+physics simulation code. It supports both standalone mode (without PhySimX)
+and plugin mode (as part of PhySimX).
+
+Standalone mode (without PhySimX installed):
+    >>> from flash import FlashSimulator
+    >>> sim = FlashSimulator(mock=True)
+    >>> result = sim.simulate(request)
+
+Plugin mode (with PhySimX installed):
+    >>> from flash import FlashSimulator  # Same API
+    >>> sim = FlashSimulator(mock=True)
+
+The package automatically detects whether physimx_core is installed:
+- If yes: uses physimx_core (plugin mode, full pydantic validation)
+- If no: uses vendored _core/ (standalone mode, dataclass-based)
+
+AI Agent Notes:
+    This package includes CLAUDE.md, _MODULE_DESCRIPTIONS, and .workbuddy/
+    for automatic discovery by AI coding agents.
+"""
+
+# =======================================================================
+# Smart Import Layer
+# =======================================================================
+# Priority 1: physimx_core installed (plugin mode, full features)
+# Priority 2: standalone (fallback to vendored _core/)
+
+_STANDALONE = False
+
+try:
+    # Plugin mode: use physimx_core (full features, pydantic validation)
+    from physimx_core.interface import BaseSimulator
+    from physimx_core.schema import (
+        CapabilityCard,
+        InputVar,
+        OutputVar,
+        PhysicsDomain,
+        SimulationRequest,
+        SimulationResult,
+        SimulationStatus,
+        SimulatorType,
+    )
+
+    _STANDALONE = False
+except ImportError:
+    # Standalone mode: use vendored _core/ (dataclass-based)
+    from ._core.interface import BaseSimulator
+    from ._core.schema import (
+        CapabilityCard,
+        InputVar,
+        OutputVar,
+        PhysicsDomain,
+        SimulationRequest,
+        SimulationResult,
+        SimulationStatus,
+        SimulatorType,
+    )
+
+    _STANDALONE = True
+
+
+# =======================================================================
+# Core Exports
+# =======================================================================
+
+from .interface import FlashSimulator
+from .config import FlashConfig, get_default_config
+
+# Optional exports (may not exist in all versions)
+try:
+    from .flash_run.env.env_manager import (
+        FlashEnvironment,
+        FlashEnvManager,
+        get_env_manager,
+    )
+except ImportError:
+    FlashEnvironment = None
+    FlashEnvManager = None
+    get_env_manager = None
+
+try:
+    from .flash_run.remote.remote_deploy import FlashRemoteDeploy
+except ImportError:
+    FlashRemoteDeploy = None
+
+try:
+    from .test.math_test import LaserSlabCa1D, FlashMathTest
+except ImportError:
+    LaserSlabCa1D = None
+    FlashMathTest = None
+
+
+# =======================================================================
+# __all__ and Metadata
+# =======================================================================
+
+__all__ = [
+    # Core
+    "FlashSimulator",
+    "FlashConfig",
+    "get_default_config",
+    # Environment management
+    "FlashEnvironment",
+    "FlashEnvManager",
+    "get_env_manager",
+    # Remote deployment
+    "FlashRemoteDeploy",
+    # Physics tests
+    "LaserSlabCa1D",
+    "FlashMathTest",
+    # Base classes (for extension)
+    "BaseSimulator",
+    # Schema (for type hints)
+    "CapabilityCard",
+    "InputVar",
+    "OutputVar",
+    "PhysicsDomain",
+    "SimulationRequest",
+    "SimulationResult",
+    "SimulationStatus",
+    "SimulatorType",
+]
+
+__version__ = "0.0.1"
+__standalone__ = _STANDALONE
+
+
+# =======================================================================
+# Module-level docstring for AI agents
+# =======================================================================
+
+
+def __dir__():
+    """Return list of public attributes (for AI agents)."""
+    return sorted(__all__)
+
+
+# Warm-up: pre-import common submodules (improves tab completion)
+import sys as _sys
+
+if "flash" in _sys.modules:
+    # Pre-import common submodules for better UX
+    _common_submodules = [
+        "flash.config",
+        "flash._core.credentials",
+        "flash.flash_run",
+        "flash.input_gen",
+        "flash.output_processors",
+    ]
+    for _mod in _common_submodules:
+        try:
+            __import__(_mod)
+        except ImportError:
+            pass

flash/_bootstrap.py

@@ -0,0 +1,35 @@
+"""
+_bootstrap.py — flash 包导入路径引导标记文件
+
+此文件作为路径标记, 用于脚本通过向上搜索找到项目根目录。
+
+关键发现 (2026-08-06 目录重组后): flash 包位于项目根的 flash/ 子目录
+(标准包布局)。项目根含 pyproject.toml, 导入 flash 需将项目根加入 sys.path。
+
+标准引导代码模板:
+
+```python
+import sys
+from pathlib import Path
+
+# Bootstrap: find flash package root by searching upward for marker
+_ROOT = Path(__file__).resolve().parent
+for _ in range(12):
+    if (_ROOT / "pyproject.toml").exists():
+        break
+    _ROOT = _ROOT.parent
+else:
+    _ROOT = None  # 已安装环境 (site-packages): 静默跳过
+_PARENT = _ROOT
+if _PARENT is not None and str(_PARENT) not in sys.path:
+    sys.path.insert(0, str(_PARENT))
+```
+"""
+
+if __name__ == "__main__":
+    import sys
+    from pathlib import Path
+
+    _ROOT = Path(__file__).resolve().parent
+    print(f"flash package root: {_ROOT}")
+    print(f"flash/__init__.py exists: {(_ROOT / '__init__.py').exists()}")

flash/_core/__init__.py

@@ -0,0 +1,39 @@
+"""Vendored mini-core for standalone operation.
+
+These are local copies of physimx_core base classes.
+They are ONLY used when physimx_core is not installed.
+
+When physimx_core IS installed, flash/__init__.py will import from
+physimx_core instead (see the smart import layer in flash/__init__.py).
+"""
+
+from .interface import BaseSimulator
+from .schema import (
+    CapabilityCard, InputVar, OutputVar, PhysicsDomain,
+    SimulationRequest, SimulationResult, SimulationStatus, SimulatorType,
+)
+# 从新的模块化 credentials 包导入
+from .credentials import (
+    get_credential_manager,
+    get_user_name,
+    set_user_name,
+    mask_secret,
+    collect_ssh_accounts,
+    get_primary_ssh,
+    set_primary_ssh,
+    interactive_menu as credentials_menu,
+)
+
+__all__ = [
+    "BaseSimulator",
+    "CapabilityCard", "InputVar", "OutputVar", "PhysicsDomain",
+    "SimulationRequest", "SimulationResult", "SimulationStatus", "SimulatorType",
+    "get_credential_manager",
+    "get_user_name",
+    "set_user_name",
+    "mask_secret",
+    "collect_ssh_accounts",
+    "get_primary_ssh",
+    "set_primary_ssh",
+    "credentials_menu",
+]

flash/_core/credentials/__init__.py

@@ -0,0 +1,44 @@
+"""
+Flash 凭据管理 -- 包导出
+============================
+导出核心函数，方便其他模块导入。
+"""
+
+from ._core import (
+    get_credential_manager,
+    get_user_name,
+    set_user_name,
+    mask_secret,
+    # SSH 辅助
+    collect_ssh_accounts,
+    get_primary_ssh,
+    set_primary_ssh,
+    next_ssh_number,
+    ssh_account_name,
+    # 旧接口桥接
+    load_ssh_credentials,
+    load_all_ssh_credentials,
+    get_best_route,
+)
+# 导入统一管理菜单
+from .manage import main as interactive_menu
+
+# 版本
+__version__ = "1.0.0"
+
+# 公开 API
+__all__ = [
+    "get_credential_manager",
+    "get_user_name",
+    "set_user_name",
+    "mask_secret",
+    "collect_ssh_accounts",
+    "get_primary_ssh",
+    "set_primary_ssh",
+    "next_ssh_number",
+    "ssh_account_name",
+    "load_ssh_credentials",
+    "load_all_ssh_credentials",
+    "get_best_route",
+    "interactive_menu",
+]

flash/_core/credentials/_config.py

@@ -0,0 +1,191 @@
+"""
+Flash 凭据管理 -- 配置定义
+============================
+定义所有凭据模板及其默认值。
+"""
+
+from typing import Any, Dict, List, Tuple
+
+# 字段类型: (key, label, default)
+FieldDef = Tuple[str, str, Any]
+
+# 凭据条目类型
+EntryDef = Dict[str, Any]
+
+# 默认值
+DEFAULT_USER_NAME = "hello"
+DEFAULT_PASSWORD = "123"
+
+# ── 凭据条目定义 ──────────────────────────────────
+
+# 所有 ParaCloud 可用线路 (host, port)
+ROUTES_ALL: List[Dict[str, Any]] = [
+    {"host": "ssh.cn-zhongwei-1.paracloud.com",      "port": 8443, "label": "中卫-1 :8443"},
+    {"host": "ssh.cn-hongkong-1.paracloud.com",       "port": 22,   "label": "香港-1 :22"},
+    {"host": "ssh.cn-zhongwei-1.paracloud.com",       "port": 22,   "label": "中卫-1 :22"},
+    {"host": "ssh.cn-zhongwei-1-v6.paracloud.com",    "port": 22,   "label": "中卫-1 v6 :22"},
+    {"host": "ssh.cn-zhongwei-1.paracloud.com",       "port": 2222, "label": "中卫-1 :2222"},
+    {"host": "ssh.cn-zhongwei-1-v6.paracloud.com",    "port": 2222, "label": "中卫-1 v6 :2222"},
+    {"host": "ssh.cn-zhongwei-cstnet.paracloud.com",  "port": 22,   "label": "中卫 cstnet :22"},
+    {"host": "ssh.cn-zhongwei-cstnet-v6.paracloud.com","port": 22,   "label": "中卫 cstnet v6 :22"},
+    {"host": "ssh.paracloud.com",                     "port": 2222, "label": "paracloud :2222"},
+]
+
+ENTRIES: List[EntryDef] = [
+    # FLASH SSH 账户（支持多个）
+    {
+        "name": "flash_ssh",
+        "title": "FLASH 超算 SSH #1 (ParaCloud 中卫 NC-E)",
+        "fields": [
+            ("connection_mode", "连接模式 [auto/manual]", "auto"),
+            ("password", "密码", "123"),
+        ],
+        "route_key": "scfa2696",
+        "manual_fields": [
+            ("host",     "SSH 主机", "ssh.cn-zhongwei-1.paracloud.com"),
+            ("port",     "SSH 端口", 22),
+            ("username", "用户名",   "scfa2696@NC-E"),
+        ],
+        "routes": ROUTES_ALL,
+    },
+    {
+        "name": "flash_ssh_2",
+        "title": "FLASH 超算 SSH #2 (ParaCloud 中卫 BSCC-T6)",
+        "fields": [
+            ("connection_mode", "连接模式 [auto/manual]", "auto"),
+            ("password", "密码", "123"),
+        ],
+        "route_key": "sch0348",
+        "manual_fields": [
+            ("host",     "SSH 主机", "ssh.cn-zhongwei-1.paracloud.com"),
+            ("port",     "SSH 端口", 22),
+            ("username", "用户名",   "sch0348@BSCC-T6"),
+        ],
+        "routes": ROUTES_ALL,
+    },
+
+    # Gitee 凭据
+    {
+        "name": "gitee",
+        "title": "Gitee 访问令牌",
+        "fields": [
+            ("token",       "访问令牌 (Personal Access Token)", "123"),
+            ("username",    "用户名",                                 DEFAULT_USER_NAME),
+            ("repo_url",   "仓库 URL (可选)",                      "https://gitee.com/physimx/flash.git"),
+        ],
+    },
+
+    # PyPI 发布令牌 (pypi.org / test.pypi.org)
+    {
+        "name": "pypi",
+        "title": "PyPI 发布令牌 (pypi.org)",
+        "fields": [
+            ("token",    "PyPI API Token (pypi-xxx)", ""),
+            ("username", "用户名",                      "__token__"),
+        ],
+    },
+    {
+        "name": "testpypi",
+        "title": "TestPyPI 发布令牌 (test.pypi.org)",
+        "fields": [
+            ("token",    "TestPyPI API Token (pypi-xxx)", ""),
+            ("username", "用户名",                          "__token__"),
+        ],
+    },
+
+    # API 密钥 -- AI 服务 (可扩展)
+    {
+        "name": "deepseek_api",
+        "title": "DeepSeek API (太极网关)",
+        "fields": [
+            ("base_url", "API 地址", "https://gateway.taichuai.cn/modelhub/api/v1"),
+            ("api_key",  "API Key", "123"),
+        ],
+    },
+    # 可扩展：添加更多 AI API 凭据
+    # {
+    #     "name": "openai_api",
+    #     "title": "OpenAI API",
+    #     "fields": [
+    #         ("api_key", "API Key", "123"),
+    #         ("base_url", "API 地址 (可选)", ""),
+    #     ],
+    # },
+    # {
+    #     "name": "claude_api",
+    #     "title": "Claude API (Anthropic)",
+    #     "fields": [
+    #         ("api_key", "API Key", "123"),
+    #         ("base_url", "API 地址 (可选)", ""),
+    #     ],
+    # },
+    # {
+    #     "name": "gemini_api",
+    #     "title": "Gemini API (Google)",
+    #     "fields": [
+    #         ("api_key", "API Key", "123"),
+    #     ],
+    # },
+    # {
+    #     "name": "qwen_api",
+    #     "title": "通义千问 API (阿里云)",
+    #     "fields": [
+    #         ("api_key", "API Key", "123"),
+    #         ("base_url", "API 地址 (可选)", ""),
+    #     ],
+    # },
+]
+
+# 按名称索引
+ENTRIES_BY_NAME: Dict[str, EntryDef] = {e["name"]: e for e in ENTRIES}
+
+
+# ── SSH 预配置辅助 ──────────────────────────────────
+
+def _get_field(entry: EntryDef, key: str, default=None):
+    """从 entry 的 fields/manual_fields 里取默认值。"""
+    for k, _, v in entry.get("fields", []):
+        if k == key:
+            return v
+    for k, _, v in entry.get("manual_fields", []):
+        if k == key:
+            return v
+    return default
+
+
+def get_ssh_username(name: str) -> str:
+    """获取 SSH 账户的用户名 (从预配置读取, 未找到则返回 name 本身)。"""
+    entry = ENTRIES_BY_NAME.get(name)
+    if entry:
+        return _get_field(entry, "username", name)
+    return name
+
+
+def get_ssh_routes(name: str) -> List[Dict[str, Any]]:
+    """获取 SSH 账户的线路列表 (从预配置读取)。"""
+    entry = ENTRIES_BY_NAME.get(name)
+    if not entry:
+        return []
+    # 优先使用 entry 的 routes 字段（多线路）
+    routes = entry.get("routes")
+    if routes:
+        return list(routes)
+    # fallback: 从 manual_fields 构建单条线路
+    host = _get_field(entry, "host")
+    port = _get_field(entry, "port", 22)
+    if not host:
+        return []
+    label = entry.get("title", f"{host}:{port}")
+    return [{"host": host, "port": int(port), "label": label}]
+
+
+# PRECONFIGURED_SSH: 从 ENTRES 自动构建
+PRECONFIGURED_SSH: List[Dict[str, Any]] = [
+    {
+        "name": e["name"],
+        "title": e["title"],
+        "ssh_username": get_ssh_username(e["name"]),
+        "routes": get_ssh_routes(e["name"]),
+    }
+    for e in ENTRIES if e["name"].startswith("flash_ssh")
+]