小言_互联网的博客

用 python-docx 创建浮动图片

263人阅读  评论(0)

相信大家对python-docx这个常用的操作docx文档的库都不陌生,它支持以内联形状(Inline Shape)的形式插入图片,即图片和文本之间没有重叠,遵循流动版式(flow layout)。但是,截至最新的0.8.10版本,python-docx尚不支持插入浮动图片(floating picture)。这显然不能满足丰富多彩的文档样式的需要,因此本文探究基于python-docx插入浮动图片——剖析xml、追踪源码,最后得到完整代码。

问题提出

作者在尝试实现PDF文档转docx(pdf2docx:https://github.com/dothinking/pdf2docx,开发中)的过程中遇到一个需求:根据背景图片在PDF页面的具体位置(例如左上角坐标和图片区域的长宽),将其重现到docx页面的相应位置。考虑到背景图片与文本的重叠,这就需要实现精确定位的浮动图片,参考下图示例。

Word中的设置

我们先尝试在Office Word中,手动解决上述问题。具备基础的Word使用经验即可知,通过设置图片版式来控制图片的浮动和具体位置。

上图版式设置中的文本环绕样式,大体可以分为三类:

分类 文本重叠 自由定位 样式名称
嵌入型 In line with text
环绕型 Square, Tight, Through, Top and bottom
完全浮动 behind text, In front of text

例如最常见的嵌入型图片,它占据了整行区域,我们既不能将其与文字重叠,也不能自由放置它的位置,而是由页面排版自动确定。对于环绕型图片,文本可以进入图片所在行,但是无法与之重叠;并且,我们可以用鼠标自由拖动其位置。完全浮动型图片则可以浮于文本上方或者衬于文本下方,同时支持随意放置其位置。

如果需要精确定位,则可在图片版式的位置(Position)选项卡进行设置。它提供了多种定位方式,例如绝对定位——根据图片左上角点距离水平和竖直参考的坐标值来定位。至于参考对象,可以是页面(Page)本身,这样(0, 0)就是页面左上角;也可以是边距(Margin),此时(0, 0)即为正文区域的左上角。

综上,我们需要实现精确定位衬于文本下方的图片版式。

docx背后的xml

我们还知道,docx文档的背后是xml格式的数据,python-docx正是通过处理xml的方式来读写word文档。所以,接下来先手工创建word文档,然后查看图片部分的xml内容。

作为对比,首先分别创建一个普通嵌入型图片文件和一个衬于文本下方的浮动型图片文件。然后执行查看步骤:右键docx文件 | 7-zip打开压缩包 | word | document.xml,复制文件内容并格式化xml,得到如下的关于图片部分的片段。为了便于对比分析,删除了一些节点属性。

内联图片片段:


   
  1. <w:drawing>
  2.     <wp:inline>
  3.         <wp:extent cx= "3297600" cy= "2782800"/>
  4.         <wp:effectExtent l= "0" t= "0" r= "0" b= "0"/>
  5.         <wp:docPr id= "1" name= "Picture 1"/>
  6.         <wp:cNvGraphicFramePr>
  7.             <a:graphicFrameLocks/>
  8.         </wp:cNvGraphicFramePr>
  9.         <a:graphic>
  10.             <a:graphicData>
  11.                 <pic:pic>
  12.                     <!-- more pic content -->
  13.                 </pic:pic>
  14.             </a:graphicData>
  15.         </a:graphic>
  16.     </wp:inline>
  17. </w:drawing>

浮动图片片段:


   
  1. <w:drawing>
  2.     <wp:anchor behindDoc= "1" locked= "0" layoutInCell= "1" allowOverlap= "1">
  3.         <wp:simplePos x= "0" y= "0"/>
  4.         <wp:positionH relativeFrom= "page">
  5.             <wp:posOffset> 285750</wp:posOffset>
  6.         </wp:positionH>
  7.         <wp:positionV relativeFrom= "page">
  8.             <wp:posOffset> 457200</wp:posOffset>
  9.         </wp:positionV>
  10.         <wp:extent cx= "3297600" cy= "2782800"/>
  11.         <wp:effectExtent l= "0" t= "0" r= "0" b= "0"/>
  12.         <wp:wrapNone/>
  13.         <wp:docPr id= "1" name= "Picture 1"/>
  14.         <wp:cNvGraphicFramePr>
  15.             <a:graphicFrameLocks/>
  16.         </wp:cNvGraphicFramePr>
  17.         <a:graphic>
  18.             <a:graphicData>
  19.                 <pic:pic>
  20.                     <!-- more pic content -->
  21.                 </pic:pic>
  22.             </a:graphicData>
  23.         </a:graphic>
  24.     </wp:anchor>
  25. </w:drawing>

对比发现以下相同/相似点:

  • 两类图片都放在<w:drawing>节点下:内联图片<wp:inline>,浮动图片<wp:anchor>

  • 具备相同的内容节点:<wp:extent><wp:docPr><a:graphic>

除此之外,浮动图片还有一些独有特征,并且我们可以从命名上猜测和解读:

  • <wp:anchor>节点的behindDoc属性表明图片版式为衬于文本下方

  • <wp:positionH><wp:positionV>节点表明水平和竖直绝对定位方式,其中:

    • relativeFrom属性指定用于定位的参考对象

    • 子节点<wp:posOffset>指定具体坐标值

从内联图片开始

从xml的结构对比来看,我们完全可以根据python-docx对内联图片的实现来插入浮动图片。于是,从插入内联图片的代码入手:


   
  1. from docx  import Document
  2. from docx.shared  import Pt
  3. document = Document()
  4. document.add_picture( 'image.jpg', width=Pt( 200))
  5. document.save( 'output.docx')

python-docx安装文件夹site-packages/docx进行内容搜索add_picture,得到docx.text.run.add_picture原始定义处:


   
  1. def add_picture(self, image_path_or_stream, width=None, height=None):
  2.     inline = self.part.new_pic_inline(image_path_or_stream, width, height)
  3.     self._r.add_drawing(inline)
  4.      return InlineShape(inline)

继续搜索new_pic_inline得到docx.parts.story.BaseStoryPart.new_pic_inline。从注释可知这是利用CT_Inline类创建<wp:inline>元素,因此后续创建浮动图片的<wp:anchor>可以在此基础上修改。


   
  1. def new_pic_inline(self, image_descriptor, width, height):
  2.      "" "Return a newly-created `w:inline` element.
  3.     The element contains the image specified by *image_descriptor* and is scaled
  4.     based on the values of *width* and *height*.
  5.     " ""
  6.     rId, image = self.get_or_add_image(image_descriptor)
  7.     cx, cy = image.scaled_dimensions(width, height)
  8.     shape_id, filename = self.next_id, image.filename
  9.      return CT_Inline.new_pic_inline(shape_id, rId, filename, cx, cy)

于是进入CT_Inline类(限于篇幅,删除了前两个类方法newnew_pic_inline的具体代码)——终于见到了一开始探索的xml代码:


   
  1. class CT_Inline(BaseOxmlElement):
  2.      "" "
  3.     ``<w:inline>`` element, container for an inline shape.
  4.     " ""
  5.     @classmethod
  6.     def  new(cls, cx, cy, shape_id, pic):
  7.         pass
  8.     @classmethod
  9.     def new_pic_inline(cls, shape_id, rId, filename, cx, cy):
  10.         pass
  11.     @classmethod
  12.     def _inline_xml(cls):
  13.          return (
  14.              '<wp:inline %s>\n'
  15.              '  <wp:extent cx="914400" cy="914400"/>\n'
  16.              '  <wp:docPr id="666" name="unnamed"/>\n'
  17.              '  <wp:cNvGraphicFramePr>\n'
  18.              '    <a:graphicFrameLocks noChangeAspect="1"/>\n'
  19.              '  </wp:cNvGraphicFramePr>\n'
  20.              '  <a:graphic>\n'
  21.              '    <a:graphicData uri="URI not set"/>\n'
  22.              '  </a:graphic>\n'
  23.              '</wp:inline>' % nsdecls( 'wp''a''pic''r')
  24.         )

简单扫一下CT_Inline类的三个方法,即可将它们联系上:

  • _inline_xml()方法给出内联图片<wp:inline>的xml结构。

  • new()方法调用_inline_xml(),并为其中的子节点例如<wp:extent><wp:docPr>赋值。

  • new_pic_inline()调用new(),同时拼接CT_Picture类的结果(节点<pic:pic>,即图片的具体内容)到<a:graphicData>节点中去。

综上,实现了内联图片的完整xml结构。

插入浮动图片

从xml结构的对比及上述python-docx对内联图片的实现,得到创建浮动图片的思路:

  • 初始化<wp:anchor>结构,例如behindDoc="1"指定图片版式为衬于文本下方

  • 使用类似的代码填充<wp:anchor>元素,尤其是<wp:extent><wp:docPr><pic:pic>

  • 填充<wp:positionH><wp:positionV>精确定位图片

具体实践中发现还有关键的一步——注册xml标签名称到对应的类,例如<wp:inline>CT_Inline


   
  1. # docx.oxml.__init__.py
  2. register_element_cls( 'wp:inline', CT_Inline)

综上,利用python-docx插入浮动图片(衬于文本下方、页面定位)的完整代码如下:


   
  1. # -*- coding: utf -8 -*-
  2. # filename: add_float_picture.py
  3. '' '
  4. Implement floating image based on python-docx.
  5. - Text wrapping style: BEHIND TEXT <wp:anchor behindDoc="1">
  6. - Picture position: top-left corner of PAGE `<wp:positionH relativeFrom="page">`.
  7. Create a docx sample (Layout | Positions | More Layout Options) and explore the 
  8. source xml (Open as a zip | word | document.xml) to implement other text wrapping
  9. styles and position modes per `CT_Anchor._anchor_xml()`.
  10. ' ''
  11. from docx.oxml  import parse_xml, register_element_cls
  12. from docx.oxml.ns  import nsdecls
  13. from docx.oxml.shape  import CT_Picture
  14. from docx.oxml.xmlchemy  import BaseOxmlElement, OneAndOnlyOne
  15. # refer to docx.oxml.shape.CT_Inline
  16. class CT_Anchor(BaseOxmlElement):
  17.      "" "
  18.     ``<w:anchor>`` element, container for a floating image.
  19.     " ""
  20.     extent = OneAndOnlyOne( 'wp:extent')
  21.     docPr = OneAndOnlyOne( 'wp:docPr')
  22.     graphic = OneAndOnlyOne( 'a:graphic')
  23.     @classmethod
  24.     def  new(cls, cx, cy, shape_id, pic, pos_x, pos_y):
  25.          "" "
  26.         Return a new ``<wp:anchor>`` element populated with the values passed
  27.         as parameters.
  28.         " ""
  29.         anchor = parse_xml(cls._anchor_xml(pos_x, pos_y))
  30.         anchor.extent.cx = cx
  31.         anchor.extent.cy = cy
  32.         anchor.docPr.id = shape_id
  33.         anchor.docPr.name =  'Picture %d' % shape_id
  34.         anchor.graphic.graphicData.uri = (
  35.              'http://schemas.openxmlformats.org/drawingml/2006/picture'
  36.         )
  37.         anchor.graphic.graphicData._insert_pic(pic)
  38.          return anchor
  39.     @classmethod
  40.     def new_pic_anchor(cls, shape_id, rId, filename, cx, cy, pos_x, pos_y):
  41.          "" "
  42.         Return a new `wp:anchor` element containing the `pic:pic` element
  43.         specified by the argument values.
  44.         " ""
  45.         pic_id =  0  # Word doesn 't seem to use this, but does not omit it
  46.         pic = CT_Picture.new(pic_id, filename, rId, cx, cy)
  47.         anchor = cls.new(cx, cy, shape_id, pic, pos_x, pos_y)
  48.         anchor.graphic.graphicData._insert_pic(pic)
  49.         return anchor
  50.     @classmethod
  51.     def _anchor_xml(cls, pos_x, pos_y):
  52.         return (
  53.             '<wp:anchor distT= "0" distB= "0" distL= "0" distR= "0" simplePos= "0" relativeHeight= "0" \n '
  54.             '           behindDoc= "1" locked= "0" layoutInCell= "1" allowOverlap= "1" \n '
  55.             '           %s>\n '
  56.             '  <wp:simplePos x= "0" y= "0"/>\n '
  57.             '  <wp:positionH relativeFrom= "page">\n '
  58.             '    <wp:posOffset>%d</wp:posOffset>\n '
  59.             '  </wp:positionH>\n '
  60.             '  <wp:positionV relativeFrom= "page">\n '
  61.             '    <wp:posOffset>%d</wp:posOffset>\n '
  62.             '  </wp:positionV>\n '                    
  63.             '  <wp:extent cx= "914400" cy= "914400"/>\n '
  64.             '  <wp:wrapNone/>\n '
  65.             '  <wp:docPr id= "666" name= "unnamed"/>\n '
  66.             '  <wp:cNvGraphicFramePr>\n '
  67.             '    <a:graphicFrameLocks noChangeAspect= "1"/>\n '
  68.             '  </wp:cNvGraphicFramePr>\n '
  69.             '  <a:graphic>\n '
  70.             '    <a:graphicData uri= "URI not set"/>\n '
  71.             '  </a:graphic>\n '
  72.             '</wp:anchor> ' % ( nsdecls('wp ', 'a ', 'pic ', 'r '), int(pos_x), int(pos_y) )
  73.         )
  74. # refer to docx.parts.story.BaseStoryPart.new_pic_inline
  75. def new_pic_anchor(part, image_descriptor, width, height, pos_x, pos_y):
  76.     """Return a newly-created `w:anchor` element.
  77.     The element contains the image specified by *image_descriptor* and is scaled
  78.     based on the values of *width* and *height*.
  79.     """
  80.     rId, image = part.get_or_add_image(image_descriptor)
  81.     cx, cy = image.scaled_dimensions(width, height)
  82.     shape_id, filename = part.next_id, image.filename    
  83.     return CT_Anchor.new_pic_anchor(shape_id, rId, filename, cx, cy, pos_x, pos_y)
  84. # refer to docx.text.run.add_picture
  85. def add_float_picture(p, image_path_or_stream, width=None, height=None, pos_x=0, pos_y=0):
  86.     """Add float picture at fixed position `pos_x` and `pos_y` to the top-left point of page.
  87.     """
  88.     run = p.add_run()
  89.     anchor = new_pic_anchor(run.part, image_path_or_stream, width, height, pos_x, pos_y)
  90.     run._r.add_drawing(anchor)
  91. # refer to docx.oxml.__init__.py
  92. register_element_cls('wp:anchor ', CT_Anchor)

示例

最后,来一个例子看看结果吧:


   
  1. from docx  import Document
  2. from docx.shared  import Inches, Pt
  3. from add_float_picture  import add_float_picture
  4. if __name__ ==  '__main__':
  5.     document = Document()
  6.     # add a floating picture
  7.     p = document.add_paragraph()
  8.     add_float_picture(p,  'test.png', width=Inches( 5.0), pos_x=Pt( 20), pos_y=Pt( 30))
  9.     # add text
  10.     p.add_run( 'Hello World '* 50)
  11.     document.save( 'output.docx')

作者:crazyhat,Python及科学计算爱好者

赞 助 作 者

更多阅读

程序运行慢?你怕是写的假 Python

让 Pandas DataFrame 性能飞升 40 倍

用 PyQt 打造具有专业外观的GUI(上)

特别推荐

程序员摸鱼指南

为你精选的硅谷极客资讯,

来自FLAG巨头开发者、技术、创投一手消息


点击下方阅读原文加入社区会员


转载:https://blog.csdn.net/BF02jgtRS00XKtCx/article/details/111188806
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场