Files
md-vector-db/scripts/bump_version.py
T

48 lines
1.4 KiB
Python

"""版本号管理 — 更新 pyproject.toml 中的 version."""
import re
import sys
from pathlib import Path
def bump(part: str) -> str:
"""递增版本号.
Args:
part: "major" | "minor" | "patch"
Returns:
新版本号字符串
"""
pyproject = Path(__file__).parent.parent / "pyproject.toml"
content = pyproject.read_text(encoding="utf-8")
match = re.search(r'version\s*=\s*"(\d+)\.(\d+)\.(\d+)"', content)
if not match:
print("错误: 未找到 version 字段", file=sys.stderr)
sys.exit(1)
major, minor, patch = int(match[1]), int(match[2]), int(match[3])
if part == "major":
major += 1
minor = 0
patch = 0
elif part == "minor":
minor += 1
patch = 0
elif part == "patch":
patch += 1
else:
print(f"错误: 未知的版本部分 '{part}',可选: major/minor/patch", file=sys.stderr)
sys.exit(1)
new_version = f"{major}.{minor}.{patch}"
new_content = content.replace(match[0], f'version = "{new_version}"')
pyproject.write_text(new_content, encoding="utf-8")
print(f"版本: {match[1]}.{match[2]}.{match[3]}{new_version}")
return new_version
if __name__ == "__main__":
if len(sys.argv) != 2:
print("用法: python scripts/bump_version.py <major|minor|patch>")
sys.exit(1)
bump(sys.argv[1])