整合Excel两个表的数据 - 2 | xlrd | xlwt | Python

# coding=utf-8

import os
import datetime
import xlrd
from xlwt import Workbook
from xlrd import xldate_as_tuple

agent_num = {
    "14": 1,
    "13": 1,
    "12": 1,
    "11": 1,
    "10": 1,
    "9": 1,
    "8": 1,
    "7": 1,
    "6": 1,
    "4": 1,
    "3": 1,
    "0": 1
}

def creat_agent_sheets(book):
    agent_data = {
        "14": "厦门吉事多",
        "13": "中财支付",
        "12": "内蒙古懒人网络科技有限公司",
        "11": "快汇支付",
        "10": "创富通002",
        "9": "星译付信息科技南通有限公司",
        "8": "云付泰",
        "7": "莘丽支付",
        "6": "一键联支付",
        "4": "创富通",
        "3": "闪银掌柜",
        "0": "首瀚"
    }
    agent_sheets = {}
    for a in agent_data:
        agent_sheets[a] = book.add_sheet(agent_data[a])

    return agent_sheets

def create_book():
    w = Workbook()
    w = Workbook(encoding=‘utf-8‘)
    return w

def open_book(path):
    data = xlrd.open_workbook(path)
    return data

def save_book(book):
    book.save("11fee.xlsx")
    print "-----ok-----"

if __name__ == "__main__":
    # 创建空表格
    book1 = create_book()
    # 创建工作组
    agent_sheets_dict = creat_agent_sheets(book1)

    # 打开账单表格
    path_bill = "11m.xlsx"
    data_bill = open_book(path_bill)

    # 费用
    table_fee = data_bill.sheets()[1]
    nrows_fee = table_fee.nrows

    # 打开用户表格
    path_reg = "reg.xlsx"
    data_reg = open_book(path_reg)
    table_reg = data_reg.sheets()[0]
    reg_nrows = table_reg.nrows
    # 分类
    m = 0
    for d in xrange(nrows_fee):
        if d == 0:
            continue
        row_fee = table_fee.row_values(d)
        for r in xrange(reg_nrows):
            if r == 0:
                continue
            row_reg = table_reg.row_values(r)
            if str(row_reg[3]) == str(row_fee[8]):
                reg_n = str(int(row_reg[6] if row_reg[6] else 0))
                if reg_n:
                    w_sheet = agent_sheets_dict[reg_n]
                    No = agent_num[reg_n]
                    w_sheet.write(No, 0, agent_num[reg_n])
                    w_sheet.write(No, 1, str(datetime.datetime(*xldate_as_tuple((row_fee[2]), 0))))
                    w_sheet.write(No, 2, str(row_fee[3]))
                    w_sheet.write(No, 3, str(row_fee[4]))
                    w_sheet.write(No, 4, str(row_fee[5]))
                    w_sheet.write(No, 5, str(row_fee[6]))
                    w_sheet.write(No, 6, str(row_fee[8]))
                    agent_num[reg_n] += 1
    save_book(book1)