Git worktree؛ جداسازی تغییرات و حفظ کار نیمه‌تمام

نام فایل: worktree_fixture.py؛ این فایل را کامل و بدون حذف سطرها ذخیره کنید.

from pathlib import Path
import subprocess


def git(directory, *arguments):
    return subprocess.run(['git', '-C', str(directory), *arguments], check=True,
                          text=True, capture_output=True, encoding='utf-8').stdout.strip()


def create_fixture(root):
    root = Path(root).resolve()
    root.mkdir(parents=True, exist_ok=True)
    repository, feature = root / 'repository', root / 'feature'
    repository.mkdir()  # Refuse to reuse an existing checkout.
    git(repository, 'init', '-b', 'main')
    git(repository, 'config', 'user.name', 'Liyan Example')
    git(repository, 'config', 'user.email', 'example@example.invalid')
    git(repository, 'config', 'commit.gpgsign', 'false')
    git(repository, 'config', 'core.hooksPath', str(root / 'unused-hooks'))
    (repository / 'app.py').write_text('VALUE = "main"\n', encoding='utf-8')
    git(repository, 'add', 'app.py')
    git(repository, 'commit', '-m', 'Initial fixture')
    git(repository, 'worktree', 'add', '-b', 'feature/csv', str(feature))
    return repository, feature

create_fixture فقط مخزن تازه می‌سازد. تنظیم‌های کاربر، امضا و hook در همان fixture قرار می‌گیرند و هیچ remote فراخوانی نمی‌شود.

نام فایل: worktree_check.py؛ این فایل را کامل و بدون حذف سطرها ذخیره کنید.

from pathlib import Path
from worktree_fixture import git


def remove_clean_worktree(repository, feature, root):
    root = Path(root).resolve()
    repository, feature = Path(repository).resolve(), Path(feature).resolve()
    if repository.parent != root or feature.parent != root or feature == repository:
        raise ValueError('The worktree must be a separate child of this fixture root')
    status = git(feature, 'status', '--porcelain=v1', '--untracked-files=all')
    if status:
        raise RuntimeError('Worktree contains changes; commit or preserve them first')
    # No --force and no recursive filesystem deletion.
    git(repository, 'worktree', 'remove', str(feature))
    return git(repository, 'worktree', 'list', '--porcelain')

مسیرها ابتدا resolve و به ریشهٔ آزمایش محدود می‌شوند. وجود هر تغییر یا فایل untracked، RuntimeError می‌دهد و محتوای پوشه را حفظ می‌کند.

نام فایل: test_worktree.py؛ این فایل را کامل و بدون حذف سطرها ذخیره کنید.

from pathlib import Path
import tempfile
import unittest
from worktree_fixture import create_fixture, git
from worktree_check import remove_clean_worktree


class WorktreeTests(unittest.TestCase):
    def test_isolation_and_non_destructive_cleanup(self):
        with tempfile.TemporaryDirectory(prefix='liyan-worktree-test-') as folder:
            repository, feature = create_fixture(folder)
            (feature / 'app.py').write_text('VALUE = "feature"\n', encoding='utf-8')
            self.assertEqual((repository / 'app.py').read_text(encoding='utf-8'), 'VALUE = "main"\n')
            with self.assertRaisesRegex(RuntimeError, 'contains changes'):
                remove_clean_worktree(repository, feature, folder)
            self.assertTrue(feature.exists())
            self.assertIn('feature', (feature / 'app.py').read_text(encoding='utf-8'))
            git(feature, 'add', 'app.py')
            git(feature, 'commit', '-m', 'Complete feature example')
            remaining = remove_clean_worktree(repository, feature, folder)
            self.assertFalse(feature.exists())
            self.assertIn('refs/heads/main', remaining)

    def test_outside_path_is_refused(self):
        with self.assertRaises(ValueError):
            remove_clean_worktree(Path('root/repository'), Path('outside/feature'), Path('root'))


if __name__ == '__main__':
    unittest.main()

python -m unittest test_worktree.py را اجرا کنید. فایل main پیش و پس از ویرایش feature مقایسه می‌شود و مسیر خارج از ریشه نیز رد می‌شود.

نام فایل: run_worktree_demo.py؛ این فایل را کامل و بدون حذف سطرها ذخیره کنید.

import tempfile
from worktree_fixture import create_fixture, git
from worktree_check import remove_clean_worktree


with tempfile.TemporaryDirectory(prefix='liyan-worktree-demo-') as folder:
    repository, feature = create_fixture(folder)
    (feature / 'app.py').write_text('VALUE = "feature"\n', encoding='utf-8')
    assert git(repository, 'branch', '--show-current') == 'main'
    assert git(feature, 'branch', '--show-current') == 'feature/csv'
    print('main isolated')
    try:
        remove_clean_worktree(repository, feature, folder)
    except RuntimeError:
        print('dirty worktree preserved')
    git(feature, 'add', 'app.py')
    git(feature, 'commit', '-m', 'Save the feature')
    remove_clean_worktree(repository, feature, folder)
    print('clean worktree removed')

اجرای python run_worktree_demo.py باید سه پیام main isolated، dirty worktree preserved و clean worktree removed بدهد. همهٔ فایل‌های آزمایش در پوشهٔ موقت اختصاصی هستند.