@heylemon/lemonade 0.0.3 → 0.0.5

package/skills/docx/scripts/office/helpers/simplify_redlines.py

@@ -20,14 +20,6 @@ WORD_NS = "http://schemas.openxmlformats.org/wordprocessingml/2006/main"
 
 
 def simplify_redlines(input_dir: str) -> tuple[int, str]:
-    """Merge adjacent tracked changes from the same author in document.xml.
-
-    Args:
-        input_dir: Path to unpacked DOCX directory
-
-    Returns:
-        (merge_count, message)
-    """
     doc_xml = Path(input_dir) / "word" / "document.xml"
 
     if not doc_xml.exists():
@@ -39,7 +31,6 @@ def simplify_redlines(input_dir: str) -> tuple[int, str]:
 
         merge_count = 0
 
-        # Find all paragraphs and table cells (containers for content)
         containers = _find_elements(root, "p") + _find_elements(root, "tc")
 
         for container in containers:
@@ -54,10 +45,8 @@ def simplify_redlines(input_dir: str) -> tuple[int, str]:
 
 
 def _merge_tracked_changes_in(container, tag: str) -> int:
-    """Merge adjacent w:ins or w:del elements from the same author."""
     merge_count = 0
 
-    # Get direct children that are tracked changes of this type
     tracked = [
         child
         for child in container.childNodes
@@ -67,7 +56,6 @@ def _merge_tracked_changes_in(container, tag: str) -> int:
     if len(tracked) < 2:
         return 0
 
-    # Process from front: merge next into current when possible
     i = 0
     while i < len(tracked) - 1:
         curr = tracked[i]
@@ -78,7 +66,6 @@ def _merge_tracked_changes_in(container, tag: str) -> int:
             container.removeChild(next_elem)
             tracked.pop(i + 1)
             merge_count += 1
-            # Don't increment i - try to merge more into curr
         else:
             i += 1
 
@@ -86,13 +73,11 @@ def _merge_tracked_changes_in(container, tag: str) -> int:
 
 
 def _is_element(node, tag: str) -> bool:
-    """Check if node matches the given tag name."""
     name = node.localName or node.tagName
     return name == tag or name.endswith(f":{tag}")
 
 
 def _get_author(elem) -> str:
-    """Get the author attribute from a tracked change element."""
     author = elem.getAttribute("w:author")
     if not author:
         for attr in elem.attributes.values():
@@ -102,12 +87,9 @@ def _get_author(elem) -> str:
 
 
 def _can_merge_tracked(elem1, elem2) -> bool:
-    """Check if two tracked change elements can be merged."""
-    # Must be same author
     if _get_author(elem1) != _get_author(elem2):
         return False
 
-    # Must be truly adjacent (only whitespace between them)
     node = elem1.nextSibling
     while node and node != elem2:
         if node.nodeType == node.ELEMENT_NODE:
@@ -120,7 +102,6 @@ def _can_merge_tracked(elem1, elem2) -> bool:
 
 
 def _merge_tracked_content(target, source):
-    """Move all children from source tracked change to target."""
     while source.firstChild:
         child = source.firstChild
         source.removeChild(child)
@@ -128,7 +109,6 @@ def _merge_tracked_content(target, source):
 
 
 def _find_elements(root, tag: str) -> list:
-    """Find all elements matching tag name (with or without namespace)."""
     results = []
 
     def traverse(node):
@@ -144,11 +124,6 @@ def _find_elements(root, tag: str) -> list:
 
 
 def get_tracked_change_authors(doc_xml_path: Path) -> dict[str, int]:
-    """Get authors and their tracked change counts from a document.xml file.
-
-    Returns:
-        Dict mapping author name to count of tracked changes (w:ins + w:del)
-    """
     if not doc_xml_path.exists():
         return {}
 
@@ -172,7 +147,6 @@ def get_tracked_change_authors(doc_xml_path: Path) -> dict[str, int]:
 
 
 def _get_authors_from_docx(docx_path: Path) -> dict[str, int]:
-    """Get authors and counts from a packed DOCX file."""
     try:
         with zipfile.ZipFile(docx_path, "r") as zf:
             if "word/document.xml" not in zf.namelist():
@@ -196,22 +170,6 @@ def _get_authors_from_docx(docx_path: Path) -> dict[str, int]:
 
 
 def infer_author(modified_dir: Path, original_docx: Path, default: str = "Claude") -> str:
-    """Infer the author to validate by finding who added tracked changes.
-
-    Compares tracked change counts between modified and original documents.
-    Returns the author who added new tracked changes.
-
-    Args:
-        modified_dir: Path to unpacked DOCX directory
-        original_docx: Path to original DOCX file
-        default: Default author if no new changes found
-
-    Returns:
-        Author name to use for validation
-
-    Raises:
-        ValueError: If multiple authors added new changes (ambiguous)
-    """
     modified_xml = modified_dir / "word" / "document.xml"
     modified_authors = get_tracked_change_authors(modified_xml)
 
@@ -220,7 +178,6 @@ def infer_author(modified_dir: Path, original_docx: Path, default: str = "Claude
 
     original_authors = _get_authors_from_docx(original_docx)
 
-    # Calculate new changes per author (modified count - original count)
     new_changes: dict[str, int] = {}
     for author, count in modified_authors.items():
         original_count = original_authors.get(author, 0)

package/skills/docx/scripts/office/pack.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python3
 """Pack a directory into a DOCX, PPTX, or XLSX file.
 
 Validates with auto-repair, condenses XML formatting, and creates the Office file.
@@ -29,18 +28,6 @@ def pack(
     validate: bool = True,
     infer_author_func=None,
 ) -> tuple[None, str]:
-    """Pack a directory into an Office file (DOCX, PPTX, or XLSX).
-
-    Args:
-        input_directory: Path to unpacked Office document directory
-        output_file: Path to output Office file
-        original_file: Path to original file for validation comparison
-        validate: If True, run validation with auto-repair before packing
-        infer_author_func: Optional function to infer author for redlining validation
-
-    Returns:
-        (None, message) - message indicates success or failure
-    """
     input_dir = Path(input_directory)
     output_path = Path(output_file)
     suffix = output_path.suffix.lower()
@@ -51,7 +38,6 @@ def pack(
     if suffix not in {".docx", ".pptx", ".xlsx"}:
         return None, f"Error: {output_file} must be a .docx, .pptx, or .xlsx file"
 
-    # Validate with auto-repair if requested and original file provided
     if validate and original_file:
         original_path = Path(original_file)
         if original_path.exists():
@@ -63,17 +49,14 @@ def pack(
             if not success:
                 return None, f"Error: Validation failed for {input_dir}"
 
-    # Work in temporary directory to avoid modifying original
     with tempfile.TemporaryDirectory() as temp_dir:
         temp_content_dir = Path(temp_dir) / "content"
         shutil.copytree(input_dir, temp_content_dir)
 
-        # Process XML files to remove pretty-printing whitespace
         for pattern in ["*.xml", "*.rels"]:
             for xml_file in temp_content_dir.rglob(pattern):
                 _condense_xml(xml_file)
 
-        # Create final Office file as zip archive
         output_path.parent.mkdir(parents=True, exist_ok=True)
         with zipfile.ZipFile(output_path, "w", zipfile.ZIP_DEFLATED) as zf:
             for f in temp_content_dir.rglob("*"):
@@ -89,16 +72,10 @@ def _run_validation(
     suffix: str,
     infer_author_func=None,
 ) -> tuple[bool, str | None]:
-    """Run validation with auto-repair.
-
-    Returns:
-        (success, output) - success is True if all validations pass
-    """
     output_lines = []
     validators = []
 
     if suffix == ".docx":
-        # Infer author for redlining validation
         author = "Claude"
         if infer_author_func:
             try:
@@ -112,17 +89,14 @@ def _run_validation(
         ]
     elif suffix == ".pptx":
         validators = [PPTXSchemaValidator(unpacked_dir, original_file)]
-    # xlsx has no schema validator yet
 
     if not validators:
         return True, None
 
-    # Run auto-repair
     total_repairs = sum(v.repair() for v in validators)
     if total_repairs:
         output_lines.append(f"Auto-repaired {total_repairs} issue(s)")
 
-    # Run validation
     success = all(v.validate() for v in validators)
 
     if success:
@@ -132,18 +106,14 @@ def _run_validation(
 
 
 def _condense_xml(xml_file: Path) -> None:
-    """Strip unnecessary whitespace and remove comments from XML."""
     try:
         with open(xml_file, encoding="utf-8") as f:
             dom = defusedxml.minidom.parse(f)
 
-        # Process each element to remove whitespace and comments
         for element in dom.getElementsByTagName("*"):
-            # Skip text elements (w:t, a:t, etc.) - preserve their content
             if element.tagName.endswith(":t"):
                 continue
 
-            # Remove whitespace-only text nodes and comment nodes
             for child in list(element.childNodes):
                 if (
                     child.nodeType == child.TEXT_NODE