# -*- coding: utf-8 -*-

"""

Created on Sat Oct 24 17:21:31 2020

pip install -i https://mirrors.aliyun.com/pypi/simple/ docx

参考文档 https://python-docx.readthedocs.io/en/latest/

@author: Administrator

"""

from docx import Document

from docx.shared import Inches

from docx.oxml import OxmlElement

from docx.oxml.ns import qn

#设置表格的边框

def set_cell_border(cell, **kwargs):

"""

Set cell`s border

Usage:

set_cell_border(

cell,

top={"sz": 12, "val": "single", "color": "#FF0000", "space": "0"},

bottom={"sz": 12, "color": "#00FF00", "val": "single"},

left={"sz": 24, "val": "dashed", "shadow": "true"},

right={"sz": 12, "val": "dashed"},

)

"""

tc = cell._tc

tcPr = tc.get_or_add_tcPr()

# check for tag existnace, if none found, then create one

tcBorders = tcPr.first_child_found_in("w:tcBorders")

if tcBorders is None:

tcBorders = OxmlElement('w:tcBorders')

tcPr.append(tcBorders)

# list over all available tags

for edge in ('left', 'top', 'right', 'bottom', 'insideH', 'insideV'):

edge_data = kwargs.get(edge)

if edge_data:

tag = 'w:{}'.format(edge)

# check for tag existnace, if none found, then create one

element = tcBorders.find(qn(tag))

if element is None:

element = OxmlElement(tag)

tcBorders.append(element)

# looks like order of attributes is important

for key in ["sz", "val", "color", "space", "shadow"]:

if key in edge_data:

element.set(qn('w:{}'.format(key)), str(edge_data[key]))

document = Document()

document.add_heading('Document Title', 0)

p = document.add_paragraph('A plain paragraph having some ')

p.add_run('bold').bold = True

p.add_run(' and some ')

p.add_run('italic.').italic = True

document.add_heading('Heading, level 1', level=1)

document.add_paragraph('Intense quote', style='Intense Quote')

document.add_paragraph(

'first item in unordered list', style='List Bullet'

)

document.add_paragraph(

'first item in ordered list', style='List Number'

)

#document.add_picture('monty-truth.png', width=Inches(1.25))

records = (

(3, '101', 'Spam'),

(7, '422', 'Eggs'),

(4, '631', 'Spam, spam, eggs, and spam')

)

table = document.add_table(rows=1, cols=3)

hdr_cells = table.rows[0].cells

hdr_cells[0].text = 'Qty'

set_cell_border(

hdr_cells[0],

top={"sz": 12, "val": "single", "color": "#000000", "space": "0"},

bottom={"sz": 12, "val": "single", "color": "#000000", "space": "0"},

left={"sz": 12, "val": "single", "color": "#000000", "space": "0"},

right={"sz": 12, "val": "single", "color": "#000000", "space": "0"},

)

hdr_cells[1].text = 'Id'

hdr_cells[2].text = 'Desc'

for qty, id, desc in records:

row_cells = table.add_row().cells

row_cells[0].text = str(qty)

row_cells[1].text = id

row_cells[2].text = desc

document.add_page_break()

document.save('demo.docx')

print('finish !!!')