athena-python-docx 0.21.5__tar.gz → 0.21.6__tar.gz

{athena_python_docx-0.21.5 → athena_python_docx-0.21.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: athena-python-docx
-Version: 0.21.5
+Version: 0.21.6
 Summary: Drop-in replacement for python-docx that connects to Athena's DOCX Studio collaborative document stack
 Project-URL: Homepage, https://athenaintelligence.ai
 Author-email: Athena Intelligence <engineering@athenaintelligence.ai>

{athena_python_docx-0.21.5 → athena_python_docx-0.21.6}/docx/__init__.py

@@ -6,7 +6,7 @@ See CLAUDE.md for the API parity contract.
 
 from __future__ import annotations
 
-__version__ = "0.21.5"
+__version__ = "0.21.6"
 
 from docx.api import Document
 from docx._buffer import discard_all, flush_all

{athena_python_docx-0.21.5 → athena_python_docx-0.21.6}/docx/text/run.py

@@ -668,12 +668,27 @@ class Run:
         h_px: float = _emu_to_px(h_emu)
         w_pt: float = w_emu / 12700.0
         h_pt: float = h_emu / 12700.0
+        # Anchor inside this run's paragraph so the picture lands inline and
+        # inherits the paragraph's alignment/spacing — python-docx semantics,
+        # where ``run.add_picture`` places the image in that run's paragraph.
+        # Cell-inner paragraphs keep the legacy document-end block: their
+        # blockIds are not addressable by the applier's selection resolver.
+        at: dict = {"kind": "documentEnd"}
+        if not getattr(self, "_in_cell", False) and self._block_id:
+            at = {
+                "kind": "selection",
+                "start": {
+                    "kind": "text",
+                    "blockId": self._block_id,
+                    "offset": self._range[1],
+                },
+            }
         result: object = run_sync(
             self._session.doc.create.image(
                 {
                     "src": data_uri,
                     "size": {"width": w_px, "height": h_px},
-                    "at": {"kind": "documentEnd"},
+                    "at": at,
                 },
             ),
         )

{athena_python_docx-0.21.5 → athena_python_docx-0.21.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "athena-python-docx"
-version = "0.21.5"
+version = "0.21.6"
 description = "Drop-in replacement for python-docx that connects to Athena's DOCX Studio collaborative document stack"
 readme = "README.md"
 license = "MIT"

athena_python_docx-0.21.6/tests/test_run_add_picture_anchor.py

@@ -0,0 +1,69 @@
+"""``Run.add_picture`` anchors inside its own paragraph (python-docx parity).
+
+The picture must land inline in the run's paragraph so it inherits that
+paragraph's alignment and spacing. An unanchored ``documentEnd`` create makes
+the applier wrap the image in a fresh left-aligned paragraph — so e.g. a
+centered logo paragraph stays empty while the logo renders left-aligned in its
+own block. Cell-inner runs and ``Document.add_picture`` keep the document-end
+block form.
+"""
+
+from __future__ import annotations
+
+import base64
+from typing import Any
+
+# Minimal valid 1x1 PNG.
+_PNG = bytes.fromhex(
+    "89504e470d0a1a0a0000000d49484452000000010000000108060000001f15c489"
+    "0000000a49444154789c6360000002000154a24f5e0000000049454e44ae426082"
+)
+_PNG_URI = f"data:image/png;base64,{base64.b64encode(_PNG).decode()}"
+
+
+def _image_call(fake_sd_methods: Any) -> dict:
+    return next(c for c in fake_sd_methods.calls if c[0] == "create.image")[1]
+
+
+def test_run_add_picture_anchors_in_its_paragraph(
+    mock_session: Any, fake_sd_methods: Any
+) -> None:
+    from docx.text.run import Run
+
+    run = Run(session=mock_session, block_id="p-42", range_=(0, 11))
+    run.add_picture(_PNG_URI)
+    params = _image_call(fake_sd_methods)
+    assert params["at"] == {
+        "kind": "selection",
+        "start": {"kind": "text", "blockId": "p-42", "offset": 11},
+    }
+
+
+def test_cell_inner_run_add_picture_keeps_document_end(
+    mock_session: Any, fake_sd_methods: Any
+) -> None:
+    # Cell-inner paragraphs are not addressable by the applier's selection
+    # resolver — a blockId anchor would abort the whole batch.
+    from docx.text.run import Run
+
+    run = Run(session=mock_session, block_id="tc:0:0:0", range_=(0, 0), in_cell=True)
+    run.add_picture(_PNG_URI)
+    assert _image_call(fake_sd_methods)["at"] == {"kind": "documentEnd"}
+
+
+def test_document_add_picture_keeps_document_end(
+    mock_session: Any, fake_sd_methods: Any
+) -> None:
+    # python-docx parity: Document.add_picture appends a new paragraph holding
+    # the picture, so the block-creating document-end form is correct there.
+    from docx.document import Document as DocClass
+
+    doc = DocClass.__new__(DocClass)
+    doc._session = mock_session
+    doc._saved = False
+    doc._closed = False
+    doc._pending_user_name = None
+    doc._pending_user_email = None
+    doc._pending_track_revisions = False
+    doc.add_picture(_PNG_URI)
+    assert _image_call(fake_sd_methods)["at"] == {"kind": "documentEnd"}