Python辦公自動化從Excel中計算整理資料並寫入Word

from openpyxl import load_workbook
import os
# 獲取桌面的路徑
def GetDesktopPath():
    return os.path.join(os.path.expanduser("~"), 'Desktop')
path = GetDesktopPath() + '/資料/' # 形成資料夾的路徑便後續重複使用
workbook = load_workbook(filename=path + '資料.xlsx')
sheet = workbook.active # 獲取當前頁
# 可以用程式碼獲取資料範圍，如果要批次處理迴圈迭代也方便
# 獲取有資料範圍
print(sheet.dimensions)
# A1:W10

cells = sheet['A1:A4']  # 返回A1-A4的4個單元格
cells = sheet['A'] # 獲取A列
cells = sheet['A:C'] # 獲取A-C列
cells = sheet[5] # 獲取第5行
# 注意如果是上述用cells獲取返回的是巢狀元祖
for cell in cells:
    print(cell[0].value) # 遍歷cells依然需要取出元祖中元素才可以獲取值
# 獲取一個範圍的所有cell
# 也可以用iter_col返回列
for row in sheet.iter_rows(min_row=1, max_row=3,min_col=2, max_col=4):
    for cell in row:
        print(cell.value)

# SQE
SQE = sheet['Q2'].value
# 供應商&製造商
supplier = sheet['G2'].value
# 採購單號
C2_10 = sheet['C2:C10'] # 返回cell.tuple物件
# 利用列表推導式後面同理
vC2_10 = [str(cell[0].value) for cell in C2_10]
# 用set簡易去重後用,連線，填word表用
order_num = ','.join(set(vC2_10))
# 用set簡易去重後用&連線，word檔名命名使用
order_num_title = '&'.join(set(vC2_10))
# 產品型號
T2_10 = sheet['T2:T10']
vT2_10 = [str(cell[0].value) for cell in T2_10]
ptype = ','.join(set(vT2_10))
# 產品描述
P2_10 = sheet['P2:P10']
vP2_10 = [str(cell[0].value) for cell in P2_10]
info = ','.join(set(vP2_10))
info_title = '&'.join(set(vP2_10))
# 日期
# 用datetime庫獲取今日時間以及相應格式化
import datetime
today = datetime.datetime.today()
time = today.strftime('%Y年%m月%d日')
# 驗貨數量
V2_10 = sheet['V2:V10']
vV2_10 = [int(cell[0].value) for cell in V2_10]
total_num = sum(vV2_10) # 計算總數量
# 驗貨箱數
W2_10 = sheet['W2:W10']
vW2_10 = [int(cell[0].value) for cell in W2_10]
box_num = sum(vW2_10)
# 生成最終需要的word檔名
title = f'{order_num_title}-{supplier}-{total_num}-{info_title}-{time}-驗貨報告'
print(title)

# pip install pypiwin32
from win32com import client
docx_path = path + '模板.docx'
# doc轉docx的函數
def doc2docx(doc_path,docx_path):
    word = client.Dispatch("Word.Application")
    doc = word.Documents.Open(doc_path)
    doc.SaveAs(docx_path, 16)
    doc.Close()
    word.Quit()
    print('n doc檔案已轉換為docx n')
if not os.path.exists(docx_path):
    doc2docx(docx_path[:-1], docx_path)

docx_path = path + '模板.docx'
from docx import Document
# 範例化
document = Document(docx_path)
# 讀取word中的所有表格
tables = document.tables
# print(len(tables))
# 15

tables[0].cell(1, 1).text = SQE
tables[1].cell(1, 1).text = supplier
tables[1].cell(2, 1).text = supplier
tables[1].cell(3, 1).text = ptype
tables[1].cell(4, 1).text = info
tables[1].cell(5, 1).text = order_num
tables[1].cell(7, 1).text = time

for i in range(2, 11):
    tables[6].cell(i, 0).text = str(sheet[f'T{i}'].value)
    tables[6].cell(i, 1).text = str(sheet[f'P{i}'].value)
    tables[6].cell(i, 2).text = str(sheet[f'C{i}'].value)
    tables[6].cell(i, 4).text = str(sheet[f'V{i}'].value)
    tables[6].cell(i, 5).text = str(sheet[f'V{i}'].value)
    tables[6].cell(i, 6).text = '0'
    tables[6].cell(i, 7).text = str(sheet[f'W{i}'].value)
    tables[6].cell(i, 8).text = '0'
tables[6].cell(12, 4).text = str(total_num)
tables[6].cell(12, 5).text = str(total_num)
tables[6].cell(12, 7).text = str(box_num)

document.save(path + f'{title}.docx')
print('n檔案已生成')