#!/usr/bin/python
# -*- coding: UTF-8 -*-
#-------------------------------------------------------------------------------
# Name:        createExcel.py
# Author:      ranyuan
# Email:       [email protected]	
# Created:     23/02/2016
# requirement: python >=2.6
#-------------------------------------------------------------------------------
import xlsxwriter  #pip install xlsxwriter
#import mysqldml   # this is for the example at end of page

class excelpro:
    def __init__(self,excelname):
        self.excelname = excelname

    # create a excel file
    def createExcel(self):
        workbook = xlsxwriter.Workbook(self.excelname)
        return  workbook

    # create Sheet if have many sheet in one excel file
    def createSheet(self,workbook,sheetName,queryColumnResult,queryResult):
        worksheet = workbook.add_worksheet(sheetName)
        # process column name to excel
        colres = queryColumnResult
        bold = workbook.add_format({'bold':True})
        row ,col = 0, 0
        for info in colres:
            worksheet.write(row,col,str(info[0]),bold)
            col = col +1

        # process values  to excel
        result = queryResult
        row , col = 1, 0
        for info in result:
            for object in info:
                worksheet.write(row,col,str(object))
                col =col + 1
            col = 0
            row = row + 1

    # close file
    def closeExcel(self,workbook):
        workbook.close()

    # this is a example 
#if __name__=="__main__":

#    sql1="select apply_no,id ,person_name,person_ident,person_id_unique  from cont_loan_apply limit 3"
#    sql2="select apply_no,id ,person_ident,person_id_unique  from cont_loan_apply limit 8"
#    dbins = mysqldml.opdb(charset='utf8') #初始化資料庫例項
#    excelproin = excelpro('/home/dbaops/demo2.xlsx') #初始化excel例項
#    workbook = excelproin.createExcel() #建立excel表
#    queryColumnResult = dbins.queryColumn(sql1) #獲取查詢的欄位名結果是個列表
#    queryResult = dbins.query(sql1) #獲取查詢的具體行的資料結果也是個列表
#    excelproin.createSheet(workbook,"sheet1",queryColumnResult,queryResult) #將sql1執行的結果放到sheet1中
#    queryColumnResult = dbins.queryColumn(sql2)
#    queryResult = dbins.query(sql2)
#    excelproin.createSheet(workbook,"sheet2",queryColumnResult,queryResult)
#    excelproin.closeExcel(workbook) #最後關閉excel例項