mihailkudravcev
94 lines
2.6 KiB
Python
94 lines
2.6 KiB
Python
"""Push generated wiki tree into a Gitea wiki git repository."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import shutil
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
|
|
class WikiPushError(RuntimeError):
|
|
pass
|
|
|
|
|
|
def _run(cmd: list[str], *, cwd: Path) -> str:
|
|
proc = subprocess.run(
|
|
cmd,
|
|
cwd=cwd,
|
|
capture_output=True,
|
|
text=True,
|
|
check=False,
|
|
)
|
|
if proc.returncode != 0:
|
|
raise WikiPushError(
|
|
f"$ {' '.join(cmd)}\n{proc.stdout}\n{proc.stderr}".strip()
|
|
)
|
|
return proc.stdout
|
|
|
|
|
|
def push_wiki(
|
|
content_dir: Path,
|
|
wiki_url: str,
|
|
*,
|
|
work_dir: Path,
|
|
message: str,
|
|
branch: str = "main",
|
|
keep_unmanaged: bool = False,
|
|
) -> None:
|
|
"""
|
|
Clone/fetch wiki repo into work_dir, replace managed pages, commit, push.
|
|
|
|
Managed pages = all *.md from content_dir plus images and manifest.json.
|
|
If keep_unmanaged=False, removes other *.md (except maybe custom) before copy.
|
|
"""
|
|
content_dir = content_dir.resolve()
|
|
work_dir = work_dir.resolve()
|
|
|
|
if work_dir.exists() and (work_dir / ".git").exists():
|
|
_run(["git", "fetch", "origin"], cwd=work_dir)
|
|
_run(["git", "checkout", branch], cwd=work_dir)
|
|
_run(["git", "reset", "--hard", f"origin/{branch}"], cwd=work_dir)
|
|
else:
|
|
if work_dir.exists():
|
|
shutil.rmtree(work_dir)
|
|
work_dir.parent.mkdir(parents=True, exist_ok=True)
|
|
_run(
|
|
["git", "clone", "--branch", branch, "--single-branch", wiki_url, str(work_dir)],
|
|
cwd=work_dir.parent,
|
|
)
|
|
|
|
if not keep_unmanaged:
|
|
for path in work_dir.iterdir():
|
|
if path.name == ".git":
|
|
continue
|
|
if path.is_file() and path.suffix.lower() in {".md", ".json"}:
|
|
path.unlink()
|
|
elif path.is_file() and path.suffix.lower() in {
|
|
".png",
|
|
".jpg",
|
|
".jpeg",
|
|
".gif",
|
|
".bmp",
|
|
".webp",
|
|
}:
|
|
path.unlink()
|
|
|
|
for src in content_dir.iterdir():
|
|
if src.name == ".git":
|
|
continue
|
|
dst = work_dir / src.name
|
|
if src.is_dir():
|
|
if dst.exists():
|
|
shutil.rmtree(dst)
|
|
shutil.copytree(src, dst)
|
|
else:
|
|
shutil.copy2(src, dst)
|
|
|
|
_run(["git", "add", "-A"], cwd=work_dir)
|
|
status = _run(["git", "status", "--porcelain"], cwd=work_dir)
|
|
if not status.strip():
|
|
return
|
|
|
|
_run(["git", "commit", "-m", message], cwd=work_dir)
|
|
_run(["git", "push", "origin", f"HEAD:{branch}"], cwd=work_dir)
|