# -*- coding: utf-8 -*-

import xlsxwriter

def get_format(wd, option={}):

    return wd.add_format(option)

# 设置居中

def get_format_center(wd,num=1):

    return wd.add_format({'align': 'center','valign': 'vcenter','border':num})

def set_border_(wd, num=1):

    return wd.add_format({}).set_border(num)

# 写数据

def _write_center(worksheet, cl, data, wd):

    return worksheet.write(cl, data, get_format_center(wd))

workbook = xlsxwriter.Workbook('report.xlsx')

worksheet = workbook.add_worksheet("测试总况")

worksheet2 = workbook.add_worksheet("测试详情")

def init(worksheet):

    # 设置列行的宽高

    worksheet.set_column("A:A", 15)

    worksheet.set_column("B:B", 20)

    worksheet.set_column("C:C", 20)

    worksheet.set_column("D:D", 20)

    worksheet.set_column("E:E", 20)

    worksheet.set_column("F:F", 20)

    worksheet.set_row(1, 30)

    worksheet.set_row(2, 30)

    worksheet.set_row(3, 30)

    worksheet.set_row(4, 30)

    worksheet.set_row(5, 30)

    # worksheet.set_row(0, 200)

    define_format_H1 = get_format(workbook, {'bold': True, 'font_size': 18})

    define_format_H2 = get_format(workbook, {'bold': True, 'font_size': 14})

    define_format_H1.set_border(1)

    define_format_H2.set_border(1)

    define_format_H1.set_align("center")

    define_format_H2.set_align("center")

    define_format_H2.set_bg_color("blue")

    define_format_H2.set_color("#ffffff")

    # Create a new Chart object.

    worksheet.merge_range('A1:F1', '测试报告总概况', define_format_H1)

    worksheet.merge_range('A2:F2', '测试概括', define_format_H2)

    worksheet.merge_range('A3:A6', '这里放图片', get_format_center(workbook))

    _write_center(worksheet, "B3", '项目名称', workbook)

    _write_center(worksheet, "B4", '接口版本', workbook)

    _write_center(worksheet, "B5", '脚本语言', workbook)

    _write_center(worksheet, "B6", '测试网络', workbook)

    data = {"test_name": "智商", "test_version": "v2.0.8", "test_pl": "android", "test_net": "wifi"}

    _write_center(worksheet, "C3", data['test_name'], workbook)

    _write_center(worksheet, "C4", data['test_version'], workbook)

    _write_center(worksheet, "C5", data['test_pl'], workbook)

    _write_center(worksheet, "C6", data['test_net'], workbook)

    _write_center(worksheet, "D3", "接口总数", workbook)

    _write_center(worksheet, "D4", "通过总数", workbook)

    _write_center(worksheet, "D5", "失败总数", workbook)

    _write_center(worksheet, "D6", "测试日期", workbook)

    data1 = {"test_sum": 100, "test_success": 80, "test_failed": 20, "test_date": "2018-10-10 12:10"}

    _write_center(worksheet, "E3", data1['test_sum'], workbook)

    _write_center(worksheet, "E4", data1['test_success'], workbook)

    _write_center(worksheet, "E5", data1['test_failed'], workbook)

    _write_center(worksheet, "E6", data1['test_date'], workbook)

    _write_center(worksheet, "F3", "分数", workbook)

    worksheet.merge_range('F4:F6', '60', get_format_center(workbook))

    pie(workbook, worksheet)

 # 生成饼形图

def pie(workbook, worksheet):

    chart1 = workbook.add_chart({'type': 'pie'})

    chart1.add_series({

    'name':       '接口测试统计',

    'categories':'=测试总况!$D$4:$D$5',

   'values':    '=测试总况!$E$4:$E$5',

    })

    chart1.set_title({'name': '接口测试统计'})

    chart1.set_style(10)

    worksheet.insert_chart('A9', chart1, {'x_offset': 25, 'y_offset': 10})

def test_detail(worksheet):

    # 设置列行的宽高

    worksheet.set_column("A:A", 30)

    worksheet.set_column("B:B", 20)

    worksheet.set_column("C:C", 20)

    worksheet.set_column("D:D", 20)

    worksheet.set_column("E:E", 20)

    worksheet.set_column("F:F", 20)

    worksheet.set_column("G:G", 20)

    worksheet.set_column("H:H", 20)

    worksheet.set_row(1, 30)

    worksheet.set_row(2, 30)

    worksheet.set_row(3, 30)

    worksheet.set_row(4, 30)

    worksheet.set_row(5, 30)

    worksheet.set_row(6, 30)

    worksheet.set_row(7, 30)

    worksheet.merge_range('A1:H1', '测试详情', get_format(workbook, {'bold': True, 'font_size': 18 ,'align': 'center','valign': 'vcenter','bg_color': 'blue', 'font_color': '#ffffff'}))

    _write_center(worksheet, "A2", '用例ID', workbook)

    _write_center(worksheet, "B2", '接口名称', workbook)

    _write_center(worksheet, "C2", '接口协议', workbook)

    _write_center(worksheet, "D2", 'URL', workbook)

    _write_center(worksheet, "E2", '参数', workbook)

    _write_center(worksheet, "F2", '预期值', workbook)

    _write_center(worksheet, "G2", '实际值', workbook)

    _write_center(worksheet, "H2", '测试结果', workbook)

    data = {"info": [{"t_id": "1001", "t_name": "登陆", "t_method": "post", "t_url": "http://XXX?login", "t_param": "{user_name:test,pwd:111111}",

                      "t_hope": "{code:1,msg:登陆成功}", "t_actual": "{code:0,msg:密码错误}", "t_result": "失败"}, {"t_id": "1002", "t_name": "商品列表", "t_method": "get", "t_url": "http://XXX?getFoodList", "t_param": "{}",

                      "t_hope": "{code:1,msg:成功,info:[{name:123,detal:dfadfa,img:product/1.png},{name:456,detal:dfadfa,img:product/1.png}]}", "t_actual": "{code:1,msg:成功,info:[{name:123,detal:dfadfa,img:product/1.png},{name:456,detal:dfadfa,img:product/1.png}]}", "t_result": "成功"}],

            "test_sum": 100,"test_success": 20, "test_failed": 80}

    temp = 4

    for item in data["info"]:

        _write_center(worksheet, "A"+str(temp), item["t_id"], workbook)

        _write_center(worksheet, "B"+str(temp), item["t_name"], workbook)

        _write_center(worksheet, "C"+str(temp), item["t_method"], workbook)

        _write_center(worksheet, "D"+str(temp), item["t_url"], workbook)

        _write_center(worksheet, "E"+str(temp), item["t_param"], workbook)

        _write_center(worksheet, "F"+str(temp), item["t_hope"], workbook)

        _write_center(worksheet, "G"+str(temp), item["t_actual"], workbook)

        _write_center(worksheet, "H"+str(temp), item["t_result"], workbook)

        temp = temp -1

init(worksheet)

test_detail(worksheet2)

workbook.close()