sync-pilot 0.1.0__tar.gz

sync_pilot-0.1.0/.gitignore

@@ -0,0 +1,7 @@
+__pycache__/
+*.pyc
+.venv/
+*.egg-info/
+dist/
+build/
+.eggs/

sync_pilot-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Liang Yi
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

sync_pilot-0.1.0/PKG-INFO

@@ -0,0 +1,127 @@
+Metadata-Version: 2.4
+Name: sync-pilot
+Version: 0.1.0
+Summary: Interactive TUI tool for syncing projects to remote servers via rsync + SSH
+Project-URL: Homepage, https://github.com/cidxb/sync-pilot
+Project-URL: Repository, https://github.com/cidxb/sync-pilot
+Author: Liang Yi
+License-Expression: MIT
+License-File: LICENSE
+Keywords: remote,rsync,ssh,sync,training,tui
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: System :: Systems Administration
+Classifier: Topic :: Utilities
+Requires-Python: >=3.10
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: rich>=13.0
+Description-Content-Type: text/markdown
+
+# sync-pilot
+
+交互式 TUI 工具，通过 rsync + SSH 同步项目到远端服务器。专为 ML/DL 训练工作流设计。
+
+Interactive TUI tool for syncing projects to remote servers via rsync + SSH. Built for ML/DL training workflows.
+
+## Features / 功能
+
+- **Push/Pull** — rsync 同步代码和 checkpoints
+- **Remote Training** — 一键推送代码 + tmux 启动远端训练，自动加 `uv run` 前缀
+- **Monitor** — GPU 状态、tmux session 输出查看、实时 attach、session 管理
+- **Status** — 项目状态总览（远端目录大小、checkpoint 数量）
+- **Data Push** — 一次性推送大型数据集
+- **i18n** — 自动检测系统语言（中/英），支持 `--lang` 和 `SYNC_PILOT_LANG` 覆盖
+
+## Install / 安装
+
+```bash
+pip install sync-pilot
+# or
+uv add sync-pilot
+```
+
+## Quick Start / 快速开始
+
+```bash
+# 1. Initialize config in your project directory
+cd /path/to/your/project
+sync-pilot init
+
+# 2. Edit sync_config.yaml with your remote server info
+vim sync_config.yaml
+
+# 3. Launch interactive TUI
+sync-pilot
+```
+
+## Usage / 使用
+
+```bash
+sync-pilot              # Interactive TUI menu
+sync-pilot init         # Generate config files in current directory
+sync-pilot push         # Push code to remote
+sync-pilot pull         # Pull checkpoints from remote
+sync-pilot train        # Push + launch remote training
+sync-pilot push-data    # One-time data upload
+sync-pilot watch        # Monitor GPU & tmux sessions
+sync-pilot status       # Project status overview
+```
+
+### Options
+
+```
+-c, --config FILE    Config file path (default: auto-detect)
+-p, --project NAME   Target project name
+--lang zh|en         Force UI language
+-V, --version        Show version
+```
+
+## Configuration / 配置
+
+`sync_config.yaml`:
+
+```yaml
+remote_host: 1.2.3.4
+remote_user: root
+remote_port: 22          # Optional, default 22
+
+projects:
+  my_project:
+    description: "My ML project"
+    local_path: /home/user/workspace/project/
+    remote_path: /mnt/workspace/project/
+    exclude_file: rsync_exclude.txt   # Relative to this config file
+    use_uv: true                      # Auto-prefix python commands with uv run
+    pull_directories:
+      - checkpoints/
+    checkpoint_patterns:              # For status search (default: *.pth, *.pt)
+      - "*.pth"
+      - "*.safetensors"
+
+  training_data:
+    description: "Training dataset (32GB)"
+    local_path: /home/user/data/
+    remote_path: /mnt/data/
+    pull_only: true                   # Prevent accidental push
+```
+
+### Config file auto-detection
+
+sync-pilot looks for config files in this order:
+1. `sync_config.yaml`
+2. `sync-pilot.yaml`
+3. `.sync-pilot.yaml`
+
+## Requirements / 依赖
+
+- Python >= 3.10
+- `rsync` and `ssh` available on PATH
+- `tmux` on the remote server (for training/watch features)
+- SSH key-based auth configured (recommended)
+
+## License
+
+MIT

sync_pilot-0.1.0/README.md

@@ -0,0 +1,105 @@
+# sync-pilot
+
+交互式 TUI 工具，通过 rsync + SSH 同步项目到远端服务器。专为 ML/DL 训练工作流设计。
+
+Interactive TUI tool for syncing projects to remote servers via rsync + SSH. Built for ML/DL training workflows.
+
+## Features / 功能
+
+- **Push/Pull** — rsync 同步代码和 checkpoints
+- **Remote Training** — 一键推送代码 + tmux 启动远端训练，自动加 `uv run` 前缀
+- **Monitor** — GPU 状态、tmux session 输出查看、实时 attach、session 管理
+- **Status** — 项目状态总览（远端目录大小、checkpoint 数量）
+- **Data Push** — 一次性推送大型数据集
+- **i18n** — 自动检测系统语言（中/英），支持 `--lang` 和 `SYNC_PILOT_LANG` 覆盖
+
+## Install / 安装
+
+```bash
+pip install sync-pilot
+# or
+uv add sync-pilot
+```
+
+## Quick Start / 快速开始
+
+```bash
+# 1. Initialize config in your project directory
+cd /path/to/your/project
+sync-pilot init
+
+# 2. Edit sync_config.yaml with your remote server info
+vim sync_config.yaml
+
+# 3. Launch interactive TUI
+sync-pilot
+```
+
+## Usage / 使用
+
+```bash
+sync-pilot              # Interactive TUI menu
+sync-pilot init         # Generate config files in current directory
+sync-pilot push         # Push code to remote
+sync-pilot pull         # Pull checkpoints from remote
+sync-pilot train        # Push + launch remote training
+sync-pilot push-data    # One-time data upload
+sync-pilot watch        # Monitor GPU & tmux sessions
+sync-pilot status       # Project status overview
+```
+
+### Options
+
+```
+-c, --config FILE    Config file path (default: auto-detect)
+-p, --project NAME   Target project name
+--lang zh|en         Force UI language
+-V, --version        Show version
+```
+
+## Configuration / 配置
+
+`sync_config.yaml`:
+
+```yaml
+remote_host: 1.2.3.4
+remote_user: root
+remote_port: 22          # Optional, default 22
+
+projects:
+  my_project:
+    description: "My ML project"
+    local_path: /home/user/workspace/project/
+    remote_path: /mnt/workspace/project/
+    exclude_file: rsync_exclude.txt   # Relative to this config file
+    use_uv: true                      # Auto-prefix python commands with uv run
+    pull_directories:
+      - checkpoints/
+    checkpoint_patterns:              # For status search (default: *.pth, *.pt)
+      - "*.pth"
+      - "*.safetensors"
+
+  training_data:
+    description: "Training dataset (32GB)"
+    local_path: /home/user/data/
+    remote_path: /mnt/data/
+    pull_only: true                   # Prevent accidental push
+```
+
+### Config file auto-detection
+
+sync-pilot looks for config files in this order:
+1. `sync_config.yaml`
+2. `sync-pilot.yaml`
+3. `.sync-pilot.yaml`
+
+## Requirements / 依赖
+
+- Python >= 3.10
+- `rsync` and `ssh` available on PATH
+- `tmux` on the remote server (for training/watch features)
+- SSH key-based auth configured (recommended)
+
+## License
+
+MIT

sync_pilot-0.1.0/pyproject.toml

@@ -0,0 +1,33 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "sync-pilot"
+version = "0.1.0"
+description = "Interactive TUI tool for syncing projects to remote servers via rsync + SSH"
+readme = "README.md"
+license = "MIT"
+requires-python = ">=3.10"
+authors = [{ name = "Liang Yi" }]
+keywords = ["rsync", "ssh", "sync", "tui", "remote", "training"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Environment :: Console",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Topic :: System :: Systems Administration",
+    "Topic :: Utilities",
+]
+dependencies = [
+    "pyyaml>=6.0",
+    "rich>=13.0",
+]
+
+[project.scripts]
+sync-pilot = "sync_pilot.cli:main"
+
+[project.urls]
+Homepage = "https://github.com/cidxb/sync-pilot"
+Repository = "https://github.com/cidxb/sync-pilot"

sync_pilot-0.1.0/src/sync_pilot/__init__.py

@@ -0,0 +1,3 @@
+"""sync-pilot: Interactive TUI tool for syncing projects to remote servers."""
+
+__version__ = "0.1.0"

sync_pilot-0.1.0/src/sync_pilot/cli.py

@@ -0,0 +1,113 @@
+"""CLI entry point for sync-pilot."""
+
+import os
+import sys
+import shutil
+import argparse
+from importlib import resources
+
+from rich.console import Console
+
+from . import __version__
+from .i18n import t, set_locale
+from .core import (
+    find_config,
+    load_config,
+    interactive_menu,
+    action_push,
+    action_pull,
+    action_train,
+    action_push_data,
+    action_watch,
+    action_status,
+    console,
+)
+
+
+def action_init():
+    """Generate default config files in CWD."""
+    template_dir = resources.files("sync_pilot") / "templates"
+    files = ["sync_config.yaml", "rsync_exclude.txt"]
+
+    for fname in files:
+        dest = os.path.join(os.getcwd(), fname)
+        if os.path.exists(dest):
+            console.print(f"[yellow]{t('config_exists', path=fname)}[/yellow]")
+            continue
+
+        src = template_dir / fname
+        with resources.as_file(src) as src_path:
+            shutil.copy2(str(src_path), dest)
+        console.print(f"[green]{t('config_created', path=fname)}[/green]")
+
+    console.print(f"\n{t('init_done')}")
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        prog="sync-pilot",
+        description="Interactive TUI for syncing projects to remote servers via rsync + SSH",
+    )
+    parser.add_argument(
+        "-V", "--version",
+        action="version",
+        version=f"sync-pilot {__version__}",
+    )
+    parser.add_argument(
+        "action",
+        nargs="?",
+        default=None,
+        choices=["init", "push", "pull", "train", "push-data", "watch", "status"],
+        help="Action to execute (omit for interactive menu)",
+    )
+    parser.add_argument(
+        "command",
+        nargs="?",
+        default=None,
+        help="Training command (only for 'train' action)",
+    )
+    parser.add_argument("--project", "-p", default=None, help="Project name")
+    parser.add_argument("--config", "-c", default=None, help="Config file path")
+    parser.add_argument("--lang", default=None, choices=["zh", "en"], help="Force UI language")
+
+    args = parser.parse_args()
+
+    # Language override
+    if args.lang:
+        set_locale(args.lang)
+
+    # init doesn't need config
+    if args.action == "init":
+        action_init()
+        return
+
+    config_path = find_config(args.config)
+    config = load_config(config_path)
+
+    if args.action is None:
+        interactive_menu(config)
+    elif args.action == "push":
+        action_push(config, args.project)
+    elif args.action == "pull":
+        action_pull(config, args.project)
+    elif args.action == "train":
+        action_train(config, command=args.command, project_name=args.project)
+    elif args.action == "push-data":
+        project = args.project
+        if project is None:
+            for name, proj in config["projects"].items():
+                if proj.get("pull_only") and not proj.get("pull_directories") and not proj.get("pull_directory"):
+                    project = name
+                    break
+            if project is None:
+                console.print(f"[yellow]{t('specify_data_project')}[/yellow]")
+                sys.exit(1)
+        action_push_data(config, project)
+    elif args.action == "watch":
+        action_watch(config)
+    elif args.action == "status":
+        action_status(config)
+
+
+if __name__ == "__main__":
+    main()