# -*- coding: utf-8 -*-#
# -------------------------------------------------------------------------------
# Name:         test_excel
# Description:
# Author:       GLK
# Date:         2021/7/7
# -------------------------------------------------------------------------------
import xlrd
import os
import datetime
from xlutils.filter import process, XLRDReader, XLWTWriter
#修改表內(nèi)容
def update_sheet(sheet,sheet_copy,style_list):
    if sheet.name != '模板':
        # 得到這個(gè)表單自己寫的內(nèi)容的個(gè)數(shù)(因?yàn)槊總€(gè)人表格行數(shù)個(gè)數(shù)都不同，所以沒法定位自己寫了幾個(gè)）
        # rows = sheet.nrows - 11
        # 得到一個(gè)自己寫的列表
        print(f"表單名：{sheet.name}", end=" ")
        print(f"行數(shù)：{sheet.nrows}", end=" ")
        print(f"列數(shù)：{sheet.ncols}", end=" ")
        # print(f"自己寫的內(nèi)容個(gè)數(shù)：{rows}", end=" ")
        # 得到一個(gè)單元格的索引位置的格式
        style = style_list[sheet.cell_xf_index(4, 8)]
        print("工時(shí)內(nèi)容分別為：",end="")
        #根據(jù)觀察最多是寫了5個(gè)工時(shí)內(nèi)容
        for i in range(0,5):
            # 得到這一單元格的內(nèi)容
            value = sheet.cell_value(4 + i, 8)
            if value != "":
                print(value,end=",")
                # 按照上面的格式修改單元格
                sheet_copy.write(4 + i, 6, value, style)
        print("  工時(shí)日志修改成功!")
#通過地址定位到表
def select_sheet(addr,date):
    try:
        rb = xlrd.open_workbook(addr, formatting_info=True)
        # 參考xlutils.copy庫內(nèi)的用法 參考xlutils.filter內(nèi)的參數(shù)定義style_list(相當(dāng)于拆分copy方法)
        w = XLWTWriter()
        process(XLRDReader(rb, addr), w)
        wb = w.output[0][1]
        style_list = w.style_list
        try:
            # 原本的表   (通過名字查找)
            sheet = rb.sheet_by_name(date)
            # 復(fù)制出來的表 (通過名字查找)
            sheet_copy = wb.get_sheet(date)
        except:
            print("表名不存在,可能是周六日")
        #修改表
        update_sheet(sheet,sheet_copy,style_list)
        wb.save(addr)
    except:
        print("非xls格式的文件")
#讀取目錄下所有的文件名放到result中
def get_all_word(cwd):
    work_time = []
    get_dir = os.listdir(cwd)
    for i in get_dir:
        sub_dir = os.path.join(cwd,i)
        if os.path.isdir(sub_dir):
            get_all_word(sub_dir)
        else:
            work_time.append(i)
    return work_time
#得到當(dāng)前時(shí)間起一周前的時(shí)間表名
def get_date_week():
    #得到今天的時(shí)間
    today = datetime.date.today()
    #將當(dāng)前日期一周的時(shí)間放到list中
    week_time = [str(today - datetime.timedelta(days =7 - w)) for w in range(1, 8)]
    #因?yàn)楦袷絾栴} 需要替換掉 "-"
    day_time = []
    for day in week_time:
        replace = day.replace("-", "")
        day_time.append(replace)
    print("當(dāng)前一周的表單名為:",day_time)
    return day_time
if __name__ == '__main__':
    print("======請將所有.xls格式的工時(shí)日志文件放到'D:\工時(shí)日志'下=======")
    print()
    # 1、存放全部excel的目錄地址
    addr = r'D:\\工時(shí)日志\\'
    #2、得到addr目錄下所有的文件
    work_time = get_all_word(addr)
    print('"D:\工時(shí)日志"目錄下的文件有:',work_time)
    #3、得到當(dāng)前一周的表名
    week_time = get_date_week()
    print()
    #4、遍歷每個(gè)表
    for date in week_time:
        for work_t in work_time:
            # 拼接
            addr_work = addr + work_t
            print("正在修改的文件為：",work_t,end=" ")
            #傳入地址和要修改的表名
            select_sheet(addr_work, date)
        print("#############################")
    print("全部修改完成!")

# -*- coding: utf-8 -*-#
#-------------------------------------------------------------------------------
# Name:         test_excel
# Description:  
# Author:       GLK
# Date:         2021/7/6
#-------------------------------------------------------------------------------
import openpyxl
import os
#讀取并修改excel工時(shí)
def update(wb,time):
    try:
        sheet = wb[time]
        if sheet.title != '模板':
            #得到這個(gè)表單自己寫的內(nèi)容的個(gè)數(shù)(總長度減去固定的表格長度就是自己添加的個(gè)數(shù))
            rows = sheet.max_row - 11
            #得到一個(gè)自己寫的列表
            print(f"名：{sheet.title}",end=" ")
            print(f"行：{sheet.max_row}",end=" ")
            print(f"列：{sheet.max_column}",end=" ")
            for i in range(0,5): #通過遍歷個(gè)數(shù)得到索引到每個(gè)位置進(jìn)行讀取和修改
                xl_cell = sheet.cell(row=5+i, column=9).value
                sheet.cell(row=5+i, column=7,value=xl_cell)
                print(f"工時(shí)內(nèi)容：{xl_cell}",end=",")
            print()
    except:
        print("表不存在")
#加載excel文件并修改設(shè)置時(shí)間
def excel_work(addr_work,date):
    # 加載 excel 文件
    print(addr_work)
    try:
        wb = openpyxl.load_workbook(addr_work)
        update(wb,date)
        wb.save(addr_work)
    except:
        print("不是.xlsx格式")
#讀取目錄下所有的文件名放到work_time中
def get_all_word(cwd):
    work_time = []
    get_dir = os.listdir(cwd)
    for i in get_dir:
        sub_dir = os.path.join(cwd,i)
        if os.path.isdir(sub_dir):
            get_all_word(sub_dir)
        else:
            work_time.append(i)
    return work_time
if __name__ == '__main__':
    #存放全部excel的目錄地址
    addr = r'D:\\工時(shí)日志\\'
    #將當(dāng)前目錄下的文件名放到list中
    work_time = get_all_word(addr)
    print(work_time)
    date = input("請輸入需要修改的日期：")
    for work_t in work_time:
        #拼接
        addr_work = addr + work_t
        excel_work(addr_work,date)
    # active = wb.active
    # print(active)
    #
    # # 得到sheet對象
    # sheet = wb['20210701']
    #
    # sheet['A1'] = '修改'
    # sheet['C%d'%(5)] = 'aaaaaaaaaaa'
    #
    # ## 指定不同的文件名，可以另存為別的文件
    # wb.save(r'D:\工時(shí)日志\工作簿1.xlsx')

from xlrd import open_workbook
 r_xls = open_workbook(fileName,formatting_info=True)  # 讀取excel文件
 row = r_xls.sheets()[sheet].nrows  # 獲取已有的行數(shù)
 excel.save(fileName) # 保存

 def table_data_list(self):
        """獲取table數(shù)據(jù)，返回一個(gè)json"""
        par_ids_list, sec_ids_list, dep_ids_list, job_list, major_list = [], [], [], [], []
        en_master_qty_list, en_un_qty_list, ex_master_qty_list, ex_un_qty_list, notes_list = [], [], [], [], []
        count_num = 0
        if self:
            for line in self.line_ids:
                major_name = line.major_ids.mapped('name')
                par_ids_list.append(line.parent_company_id.name)
                sec_ids_list.append(line.secondary_company_id.name)
                dep_ids_list.append(line.department_name)
                job_list.append(line.job)
                major_list.append(','.join(major_name))
                en_master_qty_list.append(line.en_master_qty)
                en_un_qty_list.append(line.en_undergraduate_qty)
                ex_master_qty_list.append(line.ex_master_qty)
                ex_un_qty_list.append(line.ex_undergraduate_qty)
                notes_list.append(line.notes)
                count_num += 1
        data_array = {
            'id': self.id,
            'tab_name': self.name,
            'count_num': count_num,
            'sec_ids_list': sec_ids_list,
            'par_ids_list': par_ids_list,
            'dep_ids_list': dep_ids_list,
            'job_list': job_list,
            'major_list': major_list,
            'en_master_qty_list': en_master_qty_list,
            'en_un_qty_list': en_un_qty_list,
            'ex_master_qty_list': ex_master_qty_list,
            'ex_un_qty_list': ex_un_qty_list,
            'notes_list': notes_list,
        }
        return data_array
    def btn_excl_method(self):
        """導(dǎo)出excel入口函數(shù)"""
        data_array = self.table_data_list()
        context = dict(self._context or {})
        wiz_obj = self.env['hd.export.export.data.wizard']
        filename = '計(jì)劃編制-%s' % (datetime.datetime.today())
        wiz_id = wiz_obj.sudo().create({
            'file_data': self.file_data_excel(data_array)
        })
        value = dict(
            type='ir.actions.act_url',
            target='self',
            url='/web/content?model=%s&id=%s&field=file_data&download=true&filename=%s.xls' % (
                'hd.export.export.data.wizard', wiz_id.id, filename),
        )
        return value
    def file_data_excel(self, data_array):
        # 1、使用xlrd打開Excel
        workbook1 = open_workbook(
            "E:\\HD_Settled\\dtcloud360\\appstore\\dtcloud_hd_graduate\\static\\src\\download\\hd.graduate.application.line.xlsx")
        # 2、使用xlutils模塊的copy復(fù)制打開的文件，并保留原格式
        open_mb_file_cp = copy.copy(workbook1)
        # 3、使用下標(biāo)定位的方式定位到Excel工作簿里的工作表
        worksheet = open_mb_file_cp.get_sheet(0)
        count_num = data_array['count_num']
        if count_num > 0:
            for row in range(1, count_num + 1):
                rews = row - 1
                worksheet.write(row, 0, data_array['sec_ids_list'][rews] or '')
                worksheet.write(row, 1, data_array['par_ids_list'][rews] or '')
                worksheet.write(row, 2, data_array['dep_ids_list'][rews] or '')
                worksheet.write(row, 3, data_array['job_list'][rews] or '')
                worksheet.write(row, 4, data_array['major_list'][rews] or '')
                worksheet.write(row, 5, data_array['en_master_qty_list'][rews])
                worksheet.write(row, 6, data_array['en_un_qty_list'][rews])
                worksheet.write(row, 7, data_array['ex_master_qty_list'][rews])
                worksheet.write(row, 8, data_array['ex_un_qty_list'][rews])
                worksheet.write(row, 9, data_array['notes_list'][rews] or '')
        buffer = BytesIO()
        open_mb_file_cp.save(buffer)
        return base64.encodebytes(buffer.getvalue())