基于Python开发一个本地项目一键启动管理工具

py_folder_launcher.py

CONFIG_FILE = Path(__file__).with_name("config.json")
DATABASE_FILE = Path(__file__).with_name("launcher_data.db")

def load_config(config_file: Path = CONFIG_FILE) -> Path | None:
    config_file = Path(config_file)
    if not config_file.exists():
        return None
    data = json.loads(config_file.read_text(encoding="utf-8"))
    selected_folder = data.get("selected_folder")
    return Path(selected_folder) if selected_folder else None


def save_config(config_file: Path, selected_folder: Path) -> None:
    config_file = Path(config_file)
    config_file.write_text(
        json.dumps({"selected_folder": str(Path(selected_folder))}, ensure_ascii=False, indent=2),
        encoding="utf-8",
    )

def find_python_project_dirs(root: Path) -> list[Path]:
    root = Path(root)
    result = []
    for child in sorted((path for path in root.iterdir() if path.is_dir()), key=lambda path: path.name.lower()):
        if any(path.is_file() and path.suffix.lower() == ".py" for path in child.rglob("*.py")):
            result.append(child)
    return result

child.rglob("*.py")

def list_launchable_files(folder: Path) -> list[Path]:
    folder = Path(folder)
    return sorted(
        (path for path in folder.iterdir() if path.is_file() and path.suffix.lower() in {".py", ".bat"}),
        key=lambda path: path.name.lower(),
    )

def build_bat_content(py_file: Path) -> str:
    py_file = Path(py_file)
    return (
        "@echo off\r\n"
        "cd /d \"%~dp0\"\r\n"
        f"if not exist \"{py_file.name}\" (\r\n"
        f"    echo 找不到要启动的文件：{py_file.name}\r\n"
        "    pause\r\n"
        "    exit /b 1\r\n"
        ")\r\n"
        f"python \"{py_file.name}\"\r\n"
        "pause\r\n"
    )


def create_launcher_bat(py_file: Path) -> Path:
    py_file = Path(py_file)
    bat_file = py_file.with_name(f"{py_file.stem}_一键启动.bat")
    bat_file.write_text(build_bat_content(py_file), encoding="utf-8")
    return bat_file

@echo off
cd /d "%~dp0"
if not exist "main.py" (
    echo 找不到要启动的文件：main.py
    pause
    exit /b 1
)
python "main.py"
pause

if not exist "main.py" (...)

def format_folder_label(folder: Path) -> str:
    folder = Path(folder)
    if any(folder.glob("*_一键启动.bat")):
        return f"[已生成] {folder.name}"
    return folder.name

def run_bat_file(bat_file: Path) -> None:
    subprocess.Popen(["cmd", "/c", "start", "", str(Path(bat_file))], shell=False)

cmd /c start

def read_bat_preview(bat_file: Path) -> str:
    return Path(bat_file).read_text(encoding="utf-8")

def delete_bat_file(bat_file: Path) -> None:
    bat_file = Path(bat_file)
    if bat_file.suffix.lower() != ".bat":
        raise ValueError("只能删除 bat 文件")
    bat_file.unlink()

def ensure_database(db_path: Path = DATABASE_FILE) -> None:
    conn = sqlite3.connect(db_path)
    try:
        conn.execute(
            """
            CREATE TABLE IF NOT EXISTS folder_records (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                folder_path TEXT NOT NULL UNIQUE,
                folder_name TEXT NOT NULL,
                screenshot BLOB,
                notes TEXT NOT NULL DEFAULT '',
                is_invalid INTEGER NOT NULL DEFAULT 0
            )
            """
        )
        conn.commit()
    finally:
        conn.close()

with sqlite3.connect(db_path) as conn:
    ...

conn = sqlite3.connect(db_path)
try:
    ...
finally:
    conn.close()

def get_or_create_folder_record(db_path: Path, folder: Path) -> dict:
    folder = Path(folder)
    ensure_database(db_path)
    conn = sqlite3.connect(db_path)
    try:
        conn.row_factory = sqlite3.Row
        row = conn.execute("SELECT * FROM folder_records WHERE folder_path = ?", (str(folder),)).fetchone()
        if row is None:
            cursor = conn.execute(
                "INSERT INTO folder_records (folder_path, folder_name) VALUES (?, ?)",
                (str(folder), folder.name),
            )
            conn.commit()
            row = conn.execute("SELECT * FROM folder_records WHERE id = ?", (cursor.lastrowid,)).fetchone()
        return row_to_record(row)
    finally:
        conn.close()

WHERE folder_path = ?

def update_folder_record(db_path: Path, record_id: int, screenshot: bytes | None, notes: str, is_invalid: bool) -> None:
    ensure_database(db_path)
    conn = sqlite3.connect(db_path)
    try:
        conn.execute(
            "UPDATE folder_records SET screenshot = ?, notes = ?, is_invalid = ? WHERE id = ?",
            (screenshot, notes, int(is_invalid), record_id),
        )
        conn.commit()
    finally:
        conn.close()

def is_supported_image_bytes(data: bytes | None) -> bool:
    if not data:
        return False
    return data.startswith(b"\x89PNG\r\n\x1a\n") or data.startswith(b"\xff\xd8\xff") or data.startswith(b"BM")

wx.MemoryOutputStream
wx.MemoryInputStream

image.SaveFile(str(temp_path), wx.BITMAP_TYPE_PNG)
self.current_screenshot = temp_path.read_bytes()

wx.BoxSizer(wx.VERTICAL)
wx.BoxSizer(wx.HORIZONTAL)
wx.StaticBoxSizer(wx.VERTICAL, panel, "文件夹信息")

def test_sqlite_folder_record_saves_editable_fields(self):
    with tempfile.TemporaryDirectory() as temp_dir:
        db_path = Path(temp_dir) / "launcher_data.db"
        folder = Path(temp_dir) / "project_alpha"
        screenshot = b"fake image bytes"
        folder.mkdir()

        ensure_database(db_path)
        record = get_or_create_folder_record(db_path, folder)
        update_folder_record(db_path, record["id"], screenshot, "hello", True)
        result = get_or_create_folder_record(db_path, folder)

        self.assertEqual(record["folder_name"], "project_alpha")
        self.assertEqual(record["folder_path"], str(folder))
        self.assertEqual(result["screenshot"], screenshot)
        self.assertEqual(result["notes"], "hello")
        self.assertTrue(result["is_invalid"])

def test_wx_stream_compatibility_does_not_use_unavailable_memory_streams(self):
    source = Path(py_folder_launcher.__file__).read_text(encoding="utf-8")

    self.assertNotIn("wx.MemoryOutputStream", source)
    self.assertNotIn("wx.MemoryInputStream", source)

with sqlite3.connect(db_path) as conn:
    ...

wx.MemoryOutputStream
wx.MemoryInputStream

if bat_file.suffix.lower() != ".bat":
    raise ValueError("只能删除 bat 文件")

python -m pip install wxPython

python py_folder_launcher.py

python -m unittest test_py_launcher_helpers.py -v

python -m py_compile py_folder_launcher.py