import os
import sys
import platform

libpath = os.path.dirname(os.path.abspath(__file__))
libpath += '/lib'
#sysstr = platform.system()
#if(sysstr =="Windows"):
#    libpath += '/win'
#else:
#    libpath += '/linux'

if (os.path.exists(libpath)):
    sys.path.append(libpath)

# 导入库操作
from docx import Document
# 导入英寸单位操作（可用于指定图片大小、表格宽高等）
import docx.shared 
from docx.shared import Cm, Emu, Pt,Mm,Inches

#style
import re
from docx.shared import Pt,RGBColor
from docx.enum.text import WD_ALIGN_PARAGRAPH
from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
from docx.oxml import OxmlElement
from docx.oxml.ns import qn

class Report:
    def __init__(self, path):
        self.create(path)
        self.path = path
        
    def create(self, path:str):
        if (os.path.exists(path)):
            os.remove(path)
        document = Document()
        self.doc = document
        
    def addTable(self, row, column, dataset):
        table = self.doc.add_table(row, column)  #新建表格    
        for  rowdata in dataset:     #自动填充到表格
            row_cells = table.row_cells(dataset.index(rowdata))
            for i in range(column):
                row_cells[i].text = rowdata[i]
        
    def addPicture(self, picList, layout, width, height):
        document = self.doc
        #document.add_picture(picPath, width, height)
        run = document.add_paragraph().add_run()
        totalW = document.sections[-1].page_width.mm - document.sections[-1].left_margin.mm - document.sections[-1].right_margin.mm

        w = totalW / layout
        h = height
        if (w < width): 
            h = w / width * height
        print(w)
        print(picList)
        for pic in picList:
            run.add_picture(pic, Mm(w), Mm(h))
        
    def addTitle(self, str, level):
        self.doc.add_heading(str, level)
        
    def addParagraph(self, str):
        self.doc.add_paragraph(str)
        
    def setTOC(self):
        paragraph = self.doc.add_paragraph()
        run = paragraph.add_run()
        
        fldChar = OxmlElement('w:fldChar')  # creates a new element
        fldChar.set(qn('w:fldCharType'), 'begin')  # sets attribute on element
        instrText = OxmlElement('w:instrText')
        instrText.set(qn('xml:space'), 'preserve')  # sets attribute on element
        instrText.text = 'TOC \\o "1-3" \\h \\z \\u'  # change 1-3 depending on heading levels you need

        fldChar2 = OxmlElement('w:fldChar')
        fldChar2.set(qn('w:fldCharType'), 'separate')

        # fldChar3 = OxmlElement('w:t')
        fldChar3 = OxmlElement('w:updateFields')
        fldChar3.set(qn('w:val'), 'true')
        fldChar3.text = "Right-click to update field."
        fldChar2.append(fldChar3)

        fldChar4 = OxmlElement('w:fldChar')
        fldChar4.set(qn('w:fldCharType'), 'end')
        
        r_element = run._r
        r_element.append(fldChar)
        r_element.append(instrText)
        r_element.append(fldChar2)
        r_element.append(fldChar4)
        p_element = paragraph._p

        # 添加分页符
        paragraph.add_run().add_break(docx.enum.text.WD_BREAK.PAGE)

    def writeDoc(self):
        self.doc.save(self.path)

    def set_updatefields_true(self, docx_path):
        namespace = "{http://schemas.openxmlformats.org/wordprocessingml/2006/main}"
        f = open(docx_path, 'rb')
        doc = Document(f)
        # add child to doc.settings element
        element_updatefields = lxml.etree.SubElement(
            doc.settings.element, f"{namespace}updateFields"
        )
        element_updatefields.set(f"{namespace}val", "true")
        doc.save(docx_path)  ## Heading ##
    
class docmStyle:
    def __init__(self, path):
        self.path = path
        self.doc  = Document(path)
    
    def zhengwen(self):
        for paragraph in self.doc.paragraphs:
            if re.match('^Heading 1',paragraph.style.name): 
                paragraph.paragraph_format.line_spacing = 1.5
                paragraph.paragraph_format.space_before = Pt(5)
                paragraph.paragraph_format.space_after = Pt(5)
                paragraph.paragraph_format.first_line_indent = Cm(0)
                paragraph.paragraph_format.left_indent = Cm(0)
                for run in paragraph.runs:
                    run.font.size = Pt(16)
                    run.font.name = u'黑体'
                    run.font.color.rgb = RGBColor(0,0,0)
                    run._element.rPr.rFonts.set(qn('w:eastAsia'), u'黑体')
                    run.font.bold = True
            elif re.match('^Heading 2',paragraph.style.name):
                paragraph.paragraph_format.line_spacing = 1.5
                paragraph.paragraph_format.space_before = Pt(5)
                paragraph.paragraph_format.space_after = Pt(5)
                paragraph.paragraph_format.first_line_indent = Cm(0)
                paragraph.paragraph_format.left_indent = Cm(0)
                for run in paragraph.runs:
                    run.font.size = Pt(14)
                    run.font.name = u'黑体'
                    run.font.color.rgb = RGBColor(0,0,0)
                    run._element.rPr.rFonts.set(qn('w:eastAsia'), u'黑体')
                    run.font.bold = True
            elif re.match('^Heading 3',paragraph.style.name):
                paragraph.paragraph_format.line_spacing = 1.5
                paragraph.paragraph_format.space_before = Pt(5)
                paragraph.paragraph_format.space_after = Pt(5)
                paragraph.paragraph_format.first_line_indent = Cm(0)
                paragraph.paragraph_format.left_indent = Cm(0)
                for run in paragraph.runs:
                    run.font.size = Pt(12)
                    run.font.name = u'黑体'
                    run.font.color.rgb = RGBColor(0,0,0)
                    run._element.rPr.rFonts.set(qn('w:eastAsia'), u'黑体')
                    run.font.bold = True
            elif re.match('^Heading 4',paragraph.style.name):
                paragraph.paragraph_format.line_spacing = 1.5
                paragraph.paragraph_format.space_before = Pt(5)
                paragraph.paragraph_format.space_after = Pt(5)
                paragraph.paragraph_format.first_line_indent = Cm(0)
                paragraph.paragraph_format.left_indent = Cm(0)
                for run in paragraph.runs:
                    run.font.size = Pt(12)
                    run.font.name = u'黑体'
                    run.font.color.rgb = RGBColor(0,0,0)
                    run._element.rPr.rFonts.set(qn('w:eastAsia'), u'黑体')
                    run.font.bold = True
            elif re.match('^Heading 5',paragraph.style.name):
                paragraph.paragraph_format.line_spacing = 1.5
                paragraph.paragraph_format.space_before = Pt(5)
                paragraph.paragraph_format.space_after = Pt(5)
                paragraph.paragraph_format.first_line_indent = Cm(0)
                paragraph.paragraph_format.left_indent = Cm(0)
                for run in paragraph.runs:
                    run.font.size = Pt(12)
                    run.font.name = u'黑体'
                    run.font.color.rgb = RGBColor(0,0,0)
                    run._element.rPr.rFonts.set(qn('w:eastAsia'), u'黑体')
                    run.font.bold = True
            elif re.match('^Heading 6',paragraph.style.name):
                paragraph.paragraph_format.line_spacing = 1.5
                paragraph.paragraph_format.space_before = Pt(5)
                paragraph.paragraph_format.space_after = Pt(5)
                paragraph.paragraph_format.first_line_indent = Cm(0)
                paragraph.paragraph_format.left_indent = Cm(0)
                for run in paragraph.runs:
                    run.font.size = Pt(12)
                    run.font.name = u'黑体'
                    run.font.color.rgb = RGBColor(0,0,0)
                    run._element.rPr.rFonts.set(qn('w:eastAsia'), u'黑体')
                    run.font.bold = True
            elif re.match('^Heading 7',paragraph.style.name):
                paragraph.paragraph_format.line_spacing = 1.5
                paragraph.paragraph_format.space_before = Pt(5)
                paragraph.paragraph_format.space_after = Pt(5)
                paragraph.paragraph_format.first_line_indent = Cm(0)
                paragraph.paragraph_format.left_indent = Cm(0)
                for run in paragraph.runs:
                    run.font.size = Pt(12)
                    run.font.name = u'黑体'
                    run.font.color.rgb = RGBColor(0,0,0)
                    run._element.rPr.rFonts.set(qn('w:eastAsia'), u'黑体')
                    run.font.bold = True
            elif re.match('^Title',paragraph.style.name):
                paragraph.paragraph_format.line_spacing = 1.5
                paragraph.paragraph_format.space_before = Pt(5)
                paragraph.paragraph_format.space_after = Pt(5)
                paragraph.paragraph_format.first_line_indent = Cm(0)
                paragraph.paragraph_format.left_indent = Cm(0) 
                paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER
                for run in paragraph.runs:
                    run.font.size = Pt(16)
                    run.font.name = u'黑体'
                    run.font.color.rgb = RGBColor(0,0,0)
                    run._element.rPr.rFonts.set(qn('w:eastAsia'), u'黑体')
                    run.font.bold = True
    #目录
            elif re.match('^toc',paragraph.style.name):
                for run in paragraph.runs:
                    run.font.size = Pt(10.5)
                    run.font.name = u'宋体'
                    run._element.rPr.rFonts.set(qn('w:eastAsia'), u'宋体')
                    run.font.italic = False
                    run.font.bold = False
    #图 表
            else:
                if re.match("(表)",paragraph.text):
                    print(paragraph.text)
                    paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER
                    for run in paragraph.runs:                
                        run.font.size = Pt(9)
                        run.font.name = u'黑体'
                        run._element.rPr.rFonts.set(qn('w:eastAsia'), u'黑体')
                    
                elif re.match("(图)",paragraph.text):
                    print(paragraph.text)
                    paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER
                    for run in paragraph.runs:                
                        run.font.size = Pt(9)
                        run.font.name = u'黑体'
                        run._element.rPr.rFonts.set(qn('w:eastAsia'), u'黑体')
    #正文
                else:
                    paragraph.paragraph_format.line_spacing = 1.5           
                    for run in paragraph.runs:
                        run.font.size = Pt(10.5)
                        run.font.name = u'宋体'
                        run._element.rPr.rFonts.set(qn('w:eastAsia'), u'宋体')
                        run.font.italic = False
                        run.font.bold = False
            
        
    def biaoge(self):
        document = self.doc
        tables=document.tables
        for tb in document.tables:
            for row in tb.rows:            
                for cell in row.cells:
                    for par in cell.paragraphs:
                        for run in par.runs:                    
                            run.font.size = Pt(9)
                            run.font.name ='宋体'  
                            run._element.rPr.rFonts.set(qn('w:eastAsia'), '宋体')


    def modifyStyle(self):
        self.zhengwen()
        self.biaoge()
        self.doc.save(self.path)


#def test():
#    report = Report('text.docx')
#    report.setTOC()
#    report.addTitle('Title",0)
#    report.addTitle("Head1",0)
#    report.addTitle("Head2",2)
#    report.addParagraph("paragraph")
#    dataSet = [['a','b','c','d'],['1','2','3','4'],['5','6','7','8'],['9','10','11','12']]
#    report.addTable(4,4,dataSet)
#  #  listPic = ['test.jpg', 'test.jpg', 'test.jpg', 'test.jpg']
#   # report.addPicture(listPic, 2, 1626,696)
#    report.writeDoc()

#    docstyle = docmStyle('text.docx')
#    docstyle.modifyStyle()