import xlrd
from docx import Document
from docx.enum.section import WD_ORIENTATION
from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
from docx.shared import Pt, Cm, RGBColor
from docx.oxml.ns import qn
from docx.enum.table import WD_CELL_VERTICAL_ALIGNMENT

def read_excel(self):
    """讀取Excel"""
    book = xlrd.open_workbook(self.excel_file)
    sheet = book.sheet_by_name(self.sheet_name)
    nrows = sheet.nrows # 行數(shù)
    ncols = sheet.ncols # 列數(shù)
    datas = [] # 存放數(shù)據(jù)
    # 第一列 標(biāo)題
    keys = sheet.row_values(0)
    for row in range(1, nrows):
      data = {} # 每一行數(shù)據(jù)
      for col in range(0, ncols):
        value = sheet.cell_value(row, col) # 取出每一個單元格的數(shù)據(jù)
        # 替換到特殊字符
        value = value.replace('<', '').replace('>', '').replace('$', '')
        data[keys[col]] = value
        # 截取第一列元素
        if col == 0:
          first = '' # 截取元素 第1
          second = '' # 截取元素 第2
          third = '' # 截取元素 第3
          arrs = value.lstrip('/').split('/') # 去掉第一個/ 然后再以/分組
          if len(arrs) > 0:
            if len(arrs) == 1:
              first = arrs[0]
              second = first
              third = second
            elif len(arrs) == 2:
              first = arrs[0]
              second = arrs[1]
              third = second
            elif len(arrs) == 3:
              first = arrs[0]
              second = arrs[1]
              third = arrs[2]
            else:
              first = arrs[0]
              second = arrs[1]
              third = arrs[2]
          else:
            first = value.ltrip('/')
            second = first
            third = second
          data['first'] = first
          data['second'] = second
          data['third'] = third
        # 截取第一列結(jié)束
      datas.append(data)
    return datas

def write_word(self, datas):
    """生成word文件"""
    if len(datas) < 1:
      print('Excel沒有內(nèi)容')
      return
    # 定義word文檔對象
    doc = Document()
    # 添加橫向
    section = doc.sections[0] # doc.add_section(start_type=WD_SECTION_START.CONTINUOUS) # 添加橫向頁的連續(xù)節(jié)
    section.orientation = WD_ORIENTATION.LANDSCAPE
    page_h, page_w = section.page_width, section.page_height
    section.page_width = page_w # 設(shè)置橫向紙的寬度
    section.page_height = page_h # 設(shè)置橫向紙的高度
    # 設(shè)置字體
    doc.styles['Normal'].font.name = u'宋體'
    doc.styles['Normal']._element.rPr.rFonts.set(qn('w:eastAsia'), u'宋體')
    # 獲取第3部分(部門) 并去重
    data_third = []
    for data in datas:
      third = data['third']
      if data_third.count(third) == 0:
        data_third.append(third)
    for third in data_third:
      h2 = doc.add_heading(third, level=2) # 寫入部門,二級標(biāo)題
      run = h2.runs[0] # 可以通過add_run來設(shè)置文字，也可以通過數(shù)組來獲取
      run.font.color.rgb = RGBColor(0, 0, 0)
      run.font.name = u'宋體'
      doc.add_paragraph(text='', style=None) # 增加空白行 換行
      # 開始獲取模板
      data_template = []
      for data in datas:
        if data['third'] == third:
          template = {'first': data['first'], '模板名稱': data['模板名稱']}
          if data_template.count(template) == 0:
            data_template.append(template)
      # 獲取模板完成
      # 遍歷模板
      for template in data_template:
        h3 = doc.add_heading(template['模板名稱'], level=3) # 插入模板名稱,三級標(biāo)題
        run = h3.runs[0] # 可以通過add_run來設(shè)置文字，也可以通過數(shù)組來獲取
        run.font.color.rgb = RGBColor(0, 0, 0)
        run.font.name = u'宋體'
        doc.add_paragraph(text='', style=None) # 換行
        data_table = filter(
          lambda data: data['third'] == third and data['模板名稱'] == template['模板名稱'] and data['first'] ==
                 template['first'], datas)
        data_table = list(data_table)
        # 新增表格 4行5列
        doc_table = doc.add_table(rows=4, cols=5)
        doc_table.style = "Table Grid"
        doc_table.style.font.size = Pt(9)
        doc_table.style.font.name = '宋體'

        # 合并單元格 賦值
        doc_table.rows[0].cells[1].merge(doc_table.rows[0].cells[4])
        doc_table.rows[1].cells[1].merge(doc_table.rows[1].cells[4])
        doc_table.rows[2].cells[1].merge(doc_table.rows[2].cells[4])
        doc_table.rows[0].cells[0].text = '流程名稱：'
        doc_table.rows[0].cells[1].text = data_table[0]['模板名稱']
        doc_table.rows[1].cells[0].text = '使用人：'
        doc_table.rows[1].cells[1].text = data_table[0]['first']
        doc_table.rows[2].cells[0].text = '流程說明：'
        doc_table.rows[2].cells[1].text = data_table[0]['流程說明']

        # 設(shè)置標(biāo)題
        head_cells = doc_table.rows[3].cells # 前面還有三行，特殊處理
        head_cells[0].text = '節(jié)點'
        head_cells[1].text = '節(jié)點名'
        head_cells[2].text = '處理人員'
        head_cells[3].text = '處理方式'
        head_cells[4].text = '跳轉(zhuǎn)信息'
        # 設(shè)置列寬
        head_cells[0].width = Cm(1.9)
        head_cells[1].width = Cm(4.83)
        head_cells[2].width = Cm(8.25)
        head_cells[3].width = Cm(2.54)
        head_cells[4].width = Cm(5.64)
        # 第1 列水平居中，并設(shè)置行高，所有單元格內(nèi)容垂直居中
        for i in range(0, 4):
          # 水平居中
          p = doc_table.rows[i].cells[0].paragraphs[0]
          p.alignment = WD_PARAGRAPH_ALIGNMENT.CENTER
          doc_table.rows[i].height = Cm(0.6) # 行高
          # 垂直居中
          for j in range(0, 5):
            doc_table.rows[i].cells[j].vertical_alignment = WD_CELL_VERTICAL_ALIGNMENT.CENTER

        # 生成表格并填充內(nèi)容
        row_num = 0
        for data in data_table:
          row = doc_table.add_row()
          row_cells = row.cells
          row_cells[0].text = str(row_num + 1) # 序號，需要轉(zhuǎn)換成字符串
          row_cells[1].text = data['節(jié)點名稱']
          row_cells[2].text = data['審批人員']
          row_cells[3].text = data['審批方式']
          row_cells[4].text = ''
          # 水平居中
          p = row_cells[0].paragraphs[0]
          p.alignment = WD_PARAGRAPH_ALIGNMENT.CENTER
          row.height = Cm(0.6) # 行高
          # 垂直居中
          for j in range(0, 5):
            row_cells[j].vertical_alignment = WD_CELL_VERTICAL_ALIGNMENT.CENTER
          row_num = row_num + 1

        doc.add_paragraph(text='', style=None) # 換行
    doc.save(self.word_file)