import xlrd
import xlwt
import os
import json

def get_sheet(file_name,sheet_name):
    """
    获取一个excel文件中的sheet对象
    file_name:excel文件名
    sheet_name:sheet名称
    """
    workbook = xlrd.open_workbook(file_name)   #获取excel文件操作对象
    return workbook.sheet_by_name(sheet_name)

def parser_name(sheet,index):
    """
    解析sheet数据，返回一个数据列表
    sheet:sheet对象
    """
    name_list = sheet.col_values(index) #获取公司名称列表
    del name_list[0]  #第一个为表头，去除
    return name_list

def get_enterprise_list(sheet,index):
    """
    获取sheet中公司名称,返回一个公司名称列表
    sheet:sheet对象
    index:公司名称列索引
    """
    return parser_name(sheet,index)

def get_tilte_list(sheet):
    sb_list = list()
    sb_url_list = get_enterprise_list(sheet,0)#商标号--徐展商标中文名称
    sb_title_list = get_enterprise_list(sheet, 1)  #//续展商标号
    #sb_address_list = get_enterprise_list(sheet, 0)  #
    for url,title in zip(sb_url_list,sb_title_list):
        sb_list.append({"url":url,"title":title})

    return sb_list

def get_key_list(sheet):
    sb_list = list()
    sb_type_list = get_enterprise_list(sheet, 1)  #

    #sb_address_list = get_enterprise_list(sheet, 0)  #
    for key in zip(sb_type_list):
        sb_list.append({"key":key})

    return sb_list

def create_excel(file_name,data):
    """
    创建一个excel文件
    file_name:要创建的文件名
    data:excel数据
    """
    f = xlwt.Workbook()  #创建工作簿
    key_list = data.keys()  #获取数据中的key值列表
    for key in key_list:
        sheet = f.add_sheet(key)
        data_list = data.get(key)  #通过key获取对应的sheet的数据
        excel_rows = len(data_list)  #获取excel数据总行数
        for i in range(excel_rows):
            current_cols = len(data_list[i])  #获取当前行的列数
            for j in range(current_cols):
                sheet.write(i,j,data_list[i][j])  #写数据到当前sheet
    f.save(file_name)  #保存文件