簡單使用openpyxl讀取合並單元格輸出json

#!/usr/bin/env python
# encoding: utf-8



import openpyxl
import collections
import json
import commands

#顏色顯示
def blue(string):
    print ("\033[0;34m%s\033[0m" % string)

def cyan(string):
    print ("\033[0;36m%s\033[0m" % string)

def green(string):
    print ("\033[0;32m%s\033[0m" % string)

def red(string):
    print ("\033[0;46m%s\033[0m" % string)

def write_Conf(path,data):

    with open(path, 'a+') as f:
        f.write(data + "\n")

def cmd(args):
    status, output = commands.getstatusoutput(args)
    if status == 0:
        return 0
    else:
        return "Command error or something"


def merge():
    #查詢合並的單元格
    m_list = worksheet.merged_cells
    #判斷單元格生成坐標輸出到list
    merge_all_list = []
    for m_area in m_list:
        # 合並單元格的起始行坐標、終止行坐標
        r1, r2, c1, c2 = m_area.min_row, m_area.max_row, m_area.min_col, m_area.max_col
        if (r1 != r2 and c1 != c2):
            row_col = [(x, y) for x in xrange(r1, r2 + 1) for y in xrange(c1, c2 + 1)]
            merge_all_list.append(row_col)
        elif (r1 == r2 and c1 != c2):  # or (r1 != r2 and c1 == c2):
            col = [(r1, n) for n in xrange(c1, c2 + 1)]
            merge_all_list.append(col)
        elif (r1 != r2 and c1 == c2):
            row = [(m, c1) for m in xrange(r1, r2 + 1)]
            merge_all_list.append(row)
    return merge_all_list,col   #返回一個元組(索引0是合並之后的單元格列表，索引1是第一行合並的單元格坐標)
result = cmd("> json.conf")
if result == 0:
    print "Have to empty"
else:
    print result

workbook = openpyxl.load_workbook("moban.xlsx")  #加載已經存在的excel
name_list = workbook.sheetnames
# worksheet = workbook.get_sheet_by_name(name_list[0])  #最新版本已經不能使用這種方法
worksheet = workbook[name_list[0]]
# print name_list
# print workbook
# print worksheet

rows = worksheet.rows
columns = worksheet.columns

#如下是最大行數和最大列數
row_number = worksheet.max_row
col_number = worksheet.max_column

#定義添加順序的字典
temp = collections.OrderedDict()

for i in xrange(2, worksheet.max_row+1):
    install_list = []
    col_merger = []
    for j in xrange(1, worksheet.max_column+1):

        merge_list = merge()

        '''
        merge_list結果就類似如下坐標
        [(1, 5), (1, 6), (1, 7), (1, 8), (1, 9), (1, 10), (1, 11), (1, 12), (1, 13), (1, 14)]
        [(3, 1), (4, 1)]
        [(2, 2), (2, 3), (3, 2), (3, 3)]
        [(4, 1), (5, 1), (6, 1)]
        [(8, 2), (8, 3), (8, 4), (9, 2), (9, 3), (9, 4), (10, 2), (10, 3), (10, 4)]
        '''
        #循環合並的單元格的坐標和字典的橫縱坐標對比，
        #如果存在單元格的list就獲取合並單元格中的左上角單元格值，然后跳出
        #如果不存在單元格的list就再繼續判斷是不是大於第五列，大於第五列就是添加list中然后跳出，不然添加的是重復的list
        for a, b in enumerate(merge_list[0]):
            if (i, j) in b:
                    cell_value_header = worksheet.cell(row=1, column=j).value
                    cell_value = worksheet.cell(row=merge_list[0][a][0][0], column=merge_list[0][a][0][1]).value
                    temp[cell_value_header] = cell_value
                    break
            else:
                if j >= merge_list[-1][0][-1]:
                    cell_value_header = worksheet.cell(row=1, column=merge_list[-1][0][-1]).value
                    cell_value = worksheet.cell(row=i, column=j).value
                    if not cell_value is None:
                        col_merger.append(cell_value)
                    temp[cell_value_header] = col_merger
                    break
                else:
                    cell_value_header = worksheet.cell(row=1, column=j).value
                    cell_value = worksheet.cell(row=i, column=j).value
                    temp[cell_value_header] = cell_value

    # print json.dumps(temp)
    write_Conf("json.txt", json.dumps(temp))