# coding=utf-8

import xlrd

import xlwt

import datetime

import os

class excelProcess:

    def __init__(self,keywordExcelFile,mainExcelFile):

        self.keywordExcelFile = keywordExcelFile

        self.mainExcelFile = mainExcelFile

    def WriteLog(self, message,date):

        fileName = os.path.join(os.getcwd(),  date  +   '.txt')

        with open(fileName, 'a') as f:

            f.write(message)

    def WriteSheetRow(self,sheet, rowValueList, rowIndex, isBold):

        i = 0

        style = xlwt.easyxf('font: bold 1')

        # style = xlwt.easyxf('font: bold 0, color red;')#红色字体

        # style2 = xlwt.easyxf('pattern: pattern solid, fore_colour yellow; font: bold on;') # 设置Excel单元格的背景色为黄色，字体为粗体

        for svalue in rowValueList:

            if isBold:

                sheet.write(rowIndex, i, svalue, style)

            else:

                sheet.write(rowIndex, i, svalue)

            i = i + 1

    def save_Excel(self):

        wbk = xlwt.Workbook()

        sheet = wbk.add_sheet('sheet1', cell_overwrite_ok=True)

        headList = ['IR_SITENAME', 'IR_AUTHORS', 'SY_INFOTYPE', 'RID', 'IR_URLTITLE','SY_KEYWORDS',

                    'IR_URLNAME', 'IR_URLTIME',

                    'IR_GROUPNAME', 'IR_CHANNEL',

                    'SY_BB_COMMON', 'summary', 'keyword'

                    ]

        rowIndex = 0

        self.WriteSheetRow(sheet, headList, rowIndex, True)

        for i in range(1, 11):

            rowIndex = rowIndex + 1

            valueList = []

            for j in range(1, 14):

                valueList.append(j * i)

            self.WriteSheetRow(sheet, valueList, rowIndex, False)

        fileName = os.path.join(os.getcwd(),'test.xlsx')

        wbk.save(fileName)

    def readExel(self):

        filename = r'D:\work\Excel_txtProcesss\new-微博-合并\58.xlsx'

        inwb = openpyxl.load_workbook(filename)  # 读文件

        sheetnames = inwb.get_sheet_names()  # 获取读文件中所有的sheet，通过名字的方式

        ws = inwb.get_sheet_by_name(sheetnames[0])  # 获取第一个sheet内容

        # 获取sheet的最大行数和列数

        rows = ws.max_row

        cols = ws.max_column

        for r in range(1,rows):

            for c in range(1,cols):

                print(ws.cell(r,c).value)

            if r==10:

                break

    def writeExcel(self):

        outwb = openpyxl.Workbook()  # 打开一个将写的文件

        outws = outwb.create_sheet(index=0)  # 在将写的文件创建sheet

        for row in range(1,70000):

            for col in range(1,4):

                outws.cell(row, col).value = row*2  # 写文件

            print(row)

        saveExcel = "D:\\work\\Excel_txtProcesss\\test.xlsx"

        outwb.save(saveExcel)  # 一定要记得保存

from pyExcelerator import *

# excel 第一行数据

excel_headDatas = [u'发布时间', u'文章标题', u'文章链接', u'文章简介']

articles =[

    {u'发布时间':u'2017年5月9日',

     u'文章标题':u'Python项目实战教程：国内就能访问的google搜索引擎',

     u'文章链接':'http://mp.weixin.qq.com/s?timestamp=1494557315',

     u'文章简介':u'大家可以留言、想了解python那个方向的知识、不然我也不知道'},

    {u'发布时间':u'2017年5月4日',

     u'文章标题':u'对于学习Django的建议、你知道的有那些',

     u'文章链接':'http://mp.weixin.qq.com/s?timestamp=1494557323',

     u'文章简介':u'随着Django1.4第二个候选版的发布，虽然还不支持Python3，但Django团队已经在着手计划中，据官方博客所说，Django1.5将会试验性的支持python3'}

]

# 定义excel操作句柄

excle_Workbook = Workbook()

excel_sheet_name = time.strftime('%Y-%m-%d')

excel_sheet = excle_Workbook.add_sheet(excel_sheet_name)

index = 0

#标题

for data in excel_headDatas:

    excel_sheet.write(0, index, data)

    index += 1

index = 1

#内容

for article in articles:

    colIndex = 0

    for item in excel_headDatas:

        excel_sheet.write(index, colIndex, article[item])

        colIndex += 1

    index += 1

#保存test.xlsx到当前程序目录

excle_Workbook.save('test.xlsx')

# db = mongoDB.mongoDbBase()

# db.Get_information_stat()