copilot-swe-agent[bot]andArtemFilin1990
fd20fce7fe
Co-authored-by: ArtemFilin1990 <221041513+ArtemFilin1990@users.noreply.github.com>
132 lines
4.1 KiB
Python
132 lines
4.1 KiB
Python
#!/usr/bin/env python3
|
|
"""Automatically fix articles structure validation errors."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[1]
|
|
sys.path.insert(0, str(REPO_ROOT))
|
|
|
|
from scripts.validate_articles_structure import EXPECTED, TEMPLATE_MARKERS, read_first_h1 # noqa: E402
|
|
|
|
|
|
def fix_h1_mismatch(file_path: Path, expected_h1: str) -> bool:
|
|
"""Fix H1 title mismatch."""
|
|
content = file_path.read_text(encoding="utf-8")
|
|
lines = content.splitlines(keepends=True)
|
|
|
|
if not lines:
|
|
return False
|
|
|
|
# Find first H1
|
|
for i, line in enumerate(lines):
|
|
if line.startswith("# "):
|
|
lines[i] = f"# {expected_h1}\n"
|
|
file_path.write_text("".join(lines), encoding="utf-8")
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
def add_missing_template_sections(file_path: Path) -> bool:
|
|
"""Add missing template sections to an article."""
|
|
content = file_path.read_text(encoding="utf-8")
|
|
|
|
# Check which markers are missing
|
|
missing = [marker for marker in TEMPLATE_MARKERS if marker not in content]
|
|
|
|
if not missing:
|
|
return False
|
|
|
|
# Find where to insert (after first H1)
|
|
lines = content.splitlines(keepends=True)
|
|
insert_pos = 0
|
|
|
|
for i, line in enumerate(lines):
|
|
if line.startswith("# "):
|
|
insert_pos = i + 1
|
|
# Skip any blank lines after H1
|
|
while insert_pos < len(lines) and lines[insert_pos].strip() == "":
|
|
insert_pos += 1
|
|
break
|
|
|
|
# Build sections to add
|
|
sections_to_add = []
|
|
for marker in TEMPLATE_MARKERS:
|
|
if marker not in content:
|
|
sections_to_add.append(marker)
|
|
sections_to_add.append("")
|
|
|
|
if marker == "## Цель":
|
|
sections_to_add.append("(Описание цели и назначения статьи)")
|
|
elif marker == "## Ключевые термины":
|
|
sections_to_add.append("- **Термин** — определение")
|
|
elif marker == "## Суть":
|
|
sections_to_add.append("(Основное содержание статьи)")
|
|
elif marker == "## Примеры":
|
|
sections_to_add.append("(Примеры использования)")
|
|
elif marker == "## Связанные данные":
|
|
sections_to_add.append("- (Ссылки на связанные статьи)")
|
|
elif marker == "## Источники":
|
|
sections_to_add.append("- (Источники информации)")
|
|
|
|
sections_to_add.append("")
|
|
|
|
# Insert after H1
|
|
if sections_to_add:
|
|
lines.insert(insert_pos, "\n".join(sections_to_add) + "\n")
|
|
file_path.write_text("".join(lines), encoding="utf-8")
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
def main() -> int:
|
|
"""Fix all validation errors."""
|
|
fixed_count = 0
|
|
error_count = 0
|
|
|
|
for rel, expected_h1 in EXPECTED:
|
|
file_path = REPO_ROOT / rel
|
|
|
|
if not file_path.exists():
|
|
print(f"⚠️ Missing file: {rel}")
|
|
error_count += 1
|
|
continue
|
|
|
|
# Check H1
|
|
actual_h1 = read_first_h1(file_path)
|
|
if actual_h1 != expected_h1:
|
|
print(f"🔧 Fixing H1 in {rel}")
|
|
if fix_h1_mismatch(file_path, expected_h1):
|
|
fixed_count += 1
|
|
else:
|
|
print(f"❌ Failed to fix H1 in {rel}")
|
|
error_count += 1
|
|
continue
|
|
|
|
# Check template markers
|
|
content = file_path.read_text(encoding="utf-8")
|
|
missing_markers = [m for m in TEMPLATE_MARKERS if m not in content]
|
|
|
|
if missing_markers:
|
|
print(f"🔧 Adding missing template sections to {rel}")
|
|
if add_missing_template_sections(file_path):
|
|
fixed_count += 1
|
|
else:
|
|
print(f"❌ Failed to add templates to {rel}")
|
|
error_count += 1
|
|
|
|
print(f"\n✓ Fixed {fixed_count} articles")
|
|
if error_count > 0:
|
|
print(f"❌ {error_count} errors remain")
|
|
return 1
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|