Igor20264
@@ -0,0 +1,230 @@
|
||||
import logging
|
||||
from collections import defaultdict
|
||||
from io import BytesIO
|
||||
|
||||
from PIL import Image
|
||||
from PIL.ImageDraw import ImageDraw
|
||||
from docx.document import Document
|
||||
from docx.shared import Parented, Length, Pt, Cm
|
||||
|
||||
from docx.oxml import parse_xml, register_element_cls, CT_P
|
||||
from docx.oxml.ns import nsdecls
|
||||
from docx.oxml.shape import CT_Picture
|
||||
from docx.oxml.xmlchemy import BaseOxmlElement, OneAndOnlyOne
|
||||
from docx.text.paragraph import Paragraph
|
||||
|
||||
from md2gost.util import create_element
|
||||
from .renderer import BOTTOM_MARGIN
|
||||
|
||||
EMUS_PER_PX = Pt(1)
|
||||
|
||||
|
||||
# refer to docx.oxml.shape.CT_Inline
|
||||
class CT_Anchor(BaseOxmlElement):
|
||||
"""
|
||||
``<w:anchor>`` element, container for a floating image.
|
||||
"""
|
||||
extent = OneAndOnlyOne('wp:extent')
|
||||
docPr = OneAndOnlyOne('wp:docPr')
|
||||
graphic = OneAndOnlyOne('a:graphic')
|
||||
|
||||
@classmethod
|
||||
def new(cls, cx, cy, shape_id, pic, pos_x, pos_y):
|
||||
"""
|
||||
Return a new ``<wp:anchor>`` element populated with the values passed
|
||||
as parameters.
|
||||
"""
|
||||
anchor = parse_xml(cls._anchor_xml(pos_x, pos_y))
|
||||
anchor.extent.cx = cx
|
||||
anchor.extent.cy = cy
|
||||
anchor.docPr.id = shape_id
|
||||
anchor.docPr.name = 'Picture %d' % shape_id
|
||||
anchor.graphic.graphicData.uri = (
|
||||
'http://schemas.openxmlformats.org/drawingml/2006/picture'
|
||||
)
|
||||
anchor.graphic.graphicData._insert_pic(pic)
|
||||
return anchor
|
||||
|
||||
@classmethod
|
||||
def new_pic_anchor(cls, shape_id, rId, filename, cx, cy, pos_x, pos_y):
|
||||
"""
|
||||
Return a new `wp:anchor` element containing the `pic:pic` element
|
||||
specified by the argument values.
|
||||
"""
|
||||
pic_id = 0 # Word doesn't seem to use this, but does not omit it
|
||||
pic = CT_Picture.new(pic_id, filename, rId, cx, cy)
|
||||
|
||||
try:
|
||||
pic.nvPicPr.xpath("pic:cNvPicPr")[0].append(
|
||||
create_element("a:picLocks", {
|
||||
"noChangeAspect": "1",
|
||||
"noMove": "1",
|
||||
"noResize": "1",
|
||||
"noRot": "1",
|
||||
})
|
||||
)
|
||||
except IndexError:
|
||||
pass
|
||||
|
||||
anchor = cls.new(cx, cy, shape_id, pic, pos_x, pos_y)
|
||||
anchor.graphic.graphicData._insert_pic(pic)
|
||||
return anchor
|
||||
|
||||
@classmethod
|
||||
def _anchor_xml(cls, pos_x, pos_y):
|
||||
return (
|
||||
'<wp:anchor distT="0" distB="0" distL="0" distR="0" simplePos="0" relativeHeight="0" \n'
|
||||
' behindDoc="1" locked="0" layoutInCell="1" allowOverlap="1" \n'
|
||||
' %s>\n'
|
||||
' <wp:simplePos x="0" y="0"/>\n'
|
||||
' <wp:positionH relativeFrom="page">\n'
|
||||
' <wp:posOffset>%d</wp:posOffset>\n'
|
||||
' </wp:positionH>\n'
|
||||
' <wp:positionV relativeFrom="page">\n'
|
||||
' <wp:posOffset>%d</wp:posOffset>\n'
|
||||
' </wp:positionV>\n'
|
||||
' <wp:extent cx="914400" cy="914400"/>\n'
|
||||
' <wp:wrapNone/>\n'
|
||||
' <wp:docPr id="666" name="unnamed"/>\n'
|
||||
' <wp:cNvGraphicFramePr>\n'
|
||||
' <a:graphicFrameLocks noChangeAspect="1"/>\n'
|
||||
' </wp:cNvGraphicFramePr>\n'
|
||||
' <a:graphic>\n'
|
||||
' <a:graphicData uri="URI not set"/>\n'
|
||||
' </a:graphic>\n'
|
||||
'</wp:anchor>' % ( nsdecls('wp', 'a', 'pic', 'r'), int(pos_x), int(pos_y) )
|
||||
)
|
||||
|
||||
|
||||
# refer to docx.parts.story.BaseStoryPart.new_pic_inline
|
||||
def new_pic_anchor(part, image_descriptor, width, height, pos_x, pos_y):
|
||||
"""Return a newly-created `w:anchor` element.
|
||||
|
||||
The element contains the image specified by *image_descriptor* and is scaled
|
||||
based on the values of *width* and *height*.
|
||||
"""
|
||||
rId, image = part.get_or_add_image(image_descriptor)
|
||||
cx, cy = image.scaled_dimensions(width, height)
|
||||
shape_id, filename = part.next_id, image.filename
|
||||
return CT_Anchor.new_pic_anchor(shape_id, rId, filename, cx, cy, pos_x, pos_y)
|
||||
|
||||
|
||||
# refer to docx.text.run.add_picture
|
||||
def add_float_picture(p, image_path_or_stream, width=None, height=None, pos_x=0, pos_y=0):
|
||||
"""Add float picture at fixed position `pos_x` and `pos_y` to the top-left point of page.
|
||||
"""
|
||||
run = p.add_run()
|
||||
anchor = new_pic_anchor(run.part, image_path_or_stream, width, height, pos_x, pos_y)
|
||||
run._r.add_drawing(anchor)
|
||||
|
||||
|
||||
# refer to docx.oxml.shape.__init__.py
|
||||
register_element_cls('wp:anchor', CT_Anchor)
|
||||
|
||||
def to_px(length: Length) -> int:
|
||||
return round(length / EMUS_PER_PX)
|
||||
|
||||
|
||||
class _Page:
|
||||
def __init__(self, width: Length, height: Length,
|
||||
margin_left: Length, margin_top: Length,
|
||||
margin_right: Length, margin_bottom: Length, first_color: str = None):
|
||||
self._left_offset = margin_left
|
||||
self._right_offset = margin_right
|
||||
self._max_height = height - margin_bottom
|
||||
|
||||
self._width, self._height = width, height
|
||||
self._image = Image.new("RGBA", (to_px(width), to_px(height)))
|
||||
self._draw = ImageDraw(self._image)
|
||||
self._offset = Length(margin_top)
|
||||
|
||||
self._colors = [
|
||||
(255, 0, 0, 100), # red
|
||||
(0, 255, 0, 100), # green
|
||||
(0, 0, 255, 100), # blue
|
||||
]
|
||||
if first_color in self._colors:
|
||||
self._i = self._colors.index(first_color)
|
||||
else:
|
||||
self._i = 0
|
||||
|
||||
@classmethod
|
||||
def from_document(cls, document: Document, *args, **kwargs):
|
||||
return cls(
|
||||
document.sections[0].page_width,
|
||||
document.sections[0].page_height,
|
||||
document.sections[0].left_margin,
|
||||
document.sections[0].top_margin,
|
||||
document.sections[0].right_margin,
|
||||
BOTTOM_MARGIN, # todo: fix detection
|
||||
*args, **kwargs
|
||||
)
|
||||
|
||||
@property
|
||||
def last_color(self):
|
||||
return self._colors[(self._i-1) % len(self._colors)]
|
||||
|
||||
def add_height(self, height: Length) -> Length:
|
||||
"""Return remaining height, that didn't fit to the page"""
|
||||
fitting = min(height, self._max_height - self._offset)
|
||||
remaining = height - fitting
|
||||
|
||||
self._draw.rectangle(
|
||||
(
|
||||
(to_px(self._left_offset), to_px(self._offset)),
|
||||
(to_px(self._width-self._right_offset), to_px(self._offset + fitting))
|
||||
),
|
||||
self._get_color()
|
||||
)
|
||||
self._offset += height
|
||||
return remaining
|
||||
|
||||
@property
|
||||
def image(self) -> BytesIO:
|
||||
io = BytesIO()
|
||||
self._image.save(io, Image.registered_extensions()[".png"])
|
||||
return io
|
||||
|
||||
def _get_color(self):
|
||||
color = self._colors[self._i % len(self._colors)]
|
||||
self._i += 1
|
||||
return color
|
||||
|
||||
|
||||
class Debugger:
|
||||
def __init__(self, document: Document):
|
||||
self._document = document
|
||||
|
||||
self._pages: list[_Page] = [_Page.from_document(document)]
|
||||
self._paragraphs_by_page: defaultdict[int, Paragraph] = defaultdict(lambda: None)
|
||||
|
||||
def add(self, docx_element: Parented, height: Length):
|
||||
remaining_height = self._current_page.add_height(height)
|
||||
|
||||
while remaining_height:
|
||||
self._pages.append(_Page.from_document(self._document, self._current_page.last_color))
|
||||
remaining_height = self._current_page.add_height(remaining_height)
|
||||
|
||||
if not self._paragraphs_by_page[len(self._pages)-1] and isinstance(docx_element, Paragraph)\
|
||||
and docx_element.text and "\n" not in docx_element.text:
|
||||
self._paragraphs_by_page[len(self._pages)-1] = docx_element
|
||||
|
||||
def after_rendered(self):
|
||||
"""Must be called after rendering is finished"""
|
||||
if not self._document.paragraphs:
|
||||
logging.debug("No paragraphs found, can't add debug info")
|
||||
return
|
||||
|
||||
for i in range(len(self._pages)):
|
||||
if self._paragraphs_by_page[i] is None:
|
||||
logging.debug(f"Skipping page {i} as there are no paragraphs")
|
||||
continue
|
||||
add_float_picture(
|
||||
self._paragraphs_by_page[i],
|
||||
self._pages[i].image,
|
||||
# self._document.sections[0].page_width
|
||||
)
|
||||
|
||||
@property
|
||||
def _current_page(self):
|
||||
return self._pages[-1]
|
||||
Reference in New Issue
Block a user