import re
import os
REPO = "fishcode2025/mcp-sqlite"
BRANCH = "main"
PACKAGE_NAME = "mcp-sqlite"
BUILD_BADGE = f"[](https://github.com/{REPO}/actions/workflows/rust.yml)"
VERSION_BADGE = f"[](https://github.com/{REPO}/releases/latest)"
CRATES_BADGE = f"[](https://crates.io/crates/{PACKAGE_NAME})"
DOCS_BADGE = f"[](https://docs.rs/{PACKAGE_NAME})"
LICENSE_BADGE = f"[](https://github.com/{REPO}/blob/main/LICENSE)"
BADGES = f"{BUILD_BADGE} {VERSION_BADGE} {CRATES_BADGE} {DOCS_BADGE} {LICENSE_BADGE}"
with open("README.md", "r", encoding="utf-8") as f:
content = f.read()
pattern = r"(# SQLite MCP服务器\n\n)(\[!\[.*?\n\n)"
replacement = f"\\1{BADGES}\n\n"
new_content = re.sub(pattern, replacement, content, flags=re.DOTALL)
with open("README.md", "w", encoding="utf-8") as f:
f.write(new_content)
print("README.md徽章已更新")