python:excel统计操作用类封装

 

# encoding: utf-8
# 版权所有 2023 涂聚文有限公司
# 许可信息查看：
# 描述：
# Author    : geovindu,Geovin Du 涂聚文.
# IDE       : PyCharm 2023.1 python 311
# Datetime  : 2023/9/17 5:40
# User      : geovindu
# Product   : PyCharm
# Project   : LukfookLeaveCalculation
# File      : ExportExcel.py
# explain   : 学习


import openpyxl as openws
import pandas as pd
import numpy as np
import pandasql
import os
import sys
import Common.Utils
import BLL.EmpLoyeeHolidaysGet
import Model.Employee
import Model.HolidayList
import Model.ExcelSheet
from openpyxl import load_workbook

class ExportExcelFile(object):
    """
    输出处理好的EXCEL文件

    """
    def __init__(self):
        self.sourcefile=""
        self.newfile=""


    def readSheet(self,sourcefile:str):
        """
        讀取EXCEL的工作表名稱
        :param sourcefile:
        :return:
        """
        sheenames = []
        try:
            if os.path.exists(sourcefile):
                dfsheet = pd.ExcelFile(sourcefile)
                if(len(dfsheet.sheet_names)>0):
                    for i in range(0,len(dfsheet.sheet_names)):
                        sheet=Model.ExcelSheet.Sheet()
                        sheet.SheetId=i+1
                        sheet.SheetName=dfsheet.sheet_names[i]
                        sheenames.append(sheet)
        except Exception as ex:
            print(ex)
            pass
        finally:
            print("ok")
            pass

        return sheenames



    def exportFile(self,sourcefile:str,newfile:str):
        """
        文件处理，
        :param sourcefile:源文件
        :param newfile:新文件
        :return:None
        """

        unstr = Common.Utils.Utils()
        thbll = BLL.EmpLoyeeHolidaysGet.EmpLoyeeHolidaysGet()
        try:
            if os.path.exists(sourcefile):  #判断文件是否存在
                dataframe1 = pd.read_excel(sourcefile)
                datarowcol = dataframe1.shape
                excelRows = datarowcol[0]  #得玻行
                excelColumns = datarowcol[1] #得到列
                titls = dataframe1.columns.to_list()
                year = thbll.getYear(titls[1])
                dmps = []
                # 姓名，工号，序号，部门 0行至3行为员工资料  第一行为标题，不算行的内容 可以切片方式GET数据
                # 序号
                idlist = []
                dataId = dataframe1.loc[0:0]
                for idx, datadd in dataId.iterrows():
                    #print("[{}]: {}".format(idx, datadd))
                    idlist = datadd.to_list()
                    for idd in range(1, len(idlist)):
                        # employvee=Model.Employee.employee()
                        # employvee.EmployeeId=idd
                        # dmps.append(employvee)
                        print(idd)
                # 工号
                nolist = []
                dataNo = dataframe1.loc[1:1]
                for idx, dataoo in dataNo.iterrows():
                    #print("[{}]: {}".format(idx, dataoo))
                    nolist = dataoo.to_list()

                #print("*******工号******")
                # 部门
                deplist = []
                dataDep = dataframe1.loc[2:2]
                for idx, datapp in dataDep.iterrows():
                    #print("[{}]: {}".format(idx, datapp))
                    deplist = datapp.to_list()
                #print("*******部门******")

                # 姓名
                namelist = []
                dataName = dataframe1.loc[3:3]
                for idx, datann in dataName.iterrows():
                    #print("[{}]: {}".format(idx, datann))
                    namelist = datann.to_list()

                #print("*******姓名******")

                for i in range(1, len(namelist)):
                    employvee = Model.Employee.employee()
                    employvee.EmployeeId = idlist[i]
                    employvee.EmployeeNo = nolist[i]
                    employvee.EmployeeName = namelist[i]
                    employvee.EmployeeDep = deplist[i]
                    dmps.append(employvee)

                for ob in dmps:
                    print(ob.EmployeeId, ob.EmployeeNo, ob.EmployeeName, ob.EmployeeDep)
                AnnualLeave=0
                Overtime=0
                DeferredHoliday=0
                SpecialLeave=0
                PersonalLeave=0
                SickLeave=0
                ChildcareLeave=0
                LateTime=0
                RegularHoliday=0

                # 内容
                mon=0
                day=0
                getdate=[]
                getHolidays=[]
                readrows=(excelRows-1)-3  #索此值从零开始，所以总数减一，再减三行统计的
                data4 = dataframe1.loc[4:readrows]  #第四行开始内空
                getrow=1
                emplist=[]
                for idx, datavalue in data4.iterrows():
                    #strnum=Common.Utils.Utils.getAnnualLeave(data)

                    employs = Model.Employee.employee()
                    #print("[{}]: {}".format(idx, datavalue))
                    slist=datavalue.tolist()
                    hbll=BLL.EmpLoyeeHolidaysGet.EmpLoyeeHolidaysGet() #病假4小時_x000D_事假0.5小時 未处理


                    #print(slist) #这是读取第1列的日期数据
                    for i in range(0, 1):
                        strvalue = str(slist[i])
                        #print("get value:", strvalue.replace('nan', ''), type(strvalue))
                        mon=hbll.getMonth(strvalue)
                        day=hbll.getDay(strvalue)
                        #print("*******时间************",mon,day)
                        getdate.append(str(year)+'-'+str(mon)+'-'+str(day))


                    #print(slist)
                    hid=1
                    for i in range(1,len(slist)):

                        getdd = Model.HolidayList.HolidayList()
                        getdd.HolidayDate = str(year) + '-' + str(mon) + '-' + str(day)

                        #strvalue=str(slist[i]).replace(r'\s+|\\n', ' ', regex=True)  _x000D_
                        #isnan=np.isnan(slist[i])
                        #if(isnan):
                        strvalue = str(slist[i]).replace('_x000D_', ' ')  # 规换单元格的换行符，否则处理不了正确数据
                        #strvalue = str(slist[i]).replace('nan', ' ')
                        #print("get value:",strvalue.replace('nan',''),type(strvalue))

                        strnums = hbll.getHolidays(strvalue)  # float str
                        #print("hid:",hid)
                        getdd.HolidayEmpId = hid
                        if(len(strnums)>0):
                            for sn in range(len(strnums)):
                                #print("str:",strnums[sn].HolidayId,strnums[sn].HolidayName,strnums[sn].WorkTime)
                                if strnums[sn].HolidayId==6:
                                    AnnualLeave=AnnualLeave+strnums[sn].WorkTime
                                if strnums[sn].HolidayId == 4:
                                    Overtime=Overtime+strnums[sn].WorkTime
                                if strnums[sn].HolidayId == 3:
                                    DeferredHoliday = DeferredHoliday + strnums[sn].WorkTime
                                if strnums[sn].HolidayId == 2:
                                    SpecialLeave=SpecialLeave+strnums[sn].WorkTime
                                if strnums[sn].HolidayId == 5:
                                    PersonalLeave=PersonalLeave+strnums[sn].WorkTime
                                if strnums[sn].HolidayId == 7:
                                    SickLeave=SickLeave+strnums[sn].WorkTime
                                if strnums[sn].HolidayId == 8:
                                    ChildcareLeave=ChildcareLeave+strnums[sn].WorkTime
                                if strnums[sn].HolidayId == 9:
                                    LateTime=LateTime+strnums[sn].WorkTime
                                if strnums[sn].HolidayId == 1:
                                    RegularHoliday=RegularHoliday+strnums[sn].WorkTime

                            getdd.Holdays = strnums
                        else:
                            #print("0")
                            getdd.Holdays =[]
                        hid+=1
                        getHolidays.append((getdd))
                    employs.EmployeeHolidays=getHolidays
                    employs.EmployeeActiveDate=getdate
                    getrow+=1
                    # getHolidays.append(strnums)
                    #print("类型：", type(datavalue))
                    emplist.append((employs))


                    flist=[]
                    getIdAnnualLeave=[]
                    fdata=('ID','工号','部门','姓名','日期','假期')
                    flist.append(fdata)
                    #print("date:",getdate,len(getdate))
                    #print(len(getHolidays))
                    #for dddd in emplist:
                        #print(dddd.EmployeeId,len(dddd.EmployeeHolidays))  #emplist[0].EmployeeHolidays:
                    for sd in emplist[0].EmployeeHolidays:
                        fdata=(sd.HolidayEmpId,thbll.getEmplee(dmps,sd.HolidayEmpId)[0],thbll.getEmplee(dmps,sd.HolidayEmpId)[1],thbll.getEmplee(dmps,sd.HolidayEmpId)[2],sd.HolidayDate,thbll.getHoliday(sd.Holdays))
                        flist.append(fdata)
                        #ananuaint=thbll.getEmpleeIdAnnualLeave(sd.HolidayEmpId,emplist[0].EmployeeHolidays)
                        #getIdAnnualLeave.append(ananuaint)
                        #print(sd.HolidayEmpId,thbll.getEmplee(dmps,sd.HolidayEmpId)[0],thbll.getEmplee(dmps,sd.HolidayEmpId)[1],thbll.getEmplee(dmps,sd.HolidayEmpId)[2],sd.HolidayDate,thbll.getHoliday(sd.Holdays))
                    #print("**************")
                    #print(getIdAnnualLeave)
                    #print("*************",year,"年",mon,"月","考勤合计*************")#,day,"日"
                    #print("年假合计：",AnnualLeave,"小时，加班合计：",Overtime,"小时，补休合计：",DeferredHoliday,"小时，特假合计：",SpecialLeave,"小时")
                    #print("事假合计",PersonalLeave,"小时，病假合计：",SickLeave,"小时，迟到合计：",LateTime,"分钟，育儿假合计：",ChildcareLeave,"小时")
                    #print("例休：",RegularHoliday,"天")
                    #print("*****************************************************************")
                    #年假
                    for sd in emplist[0].EmployeeHolidays:
                        ananuaint=[sd.HolidayEmpId,thbll.getHolidayAnnualLeave(sd.Holdays)]
                        getIdAnnualLeave.append(ananuaint)

                    #print(getIdAnnualLeave)
                    dd=thbll.getEmpleeIdAnnualLeave(1,getIdAnnualLeave)
                    #print("年假 id=1 sum:",dd)
                    #加班
                    getIDOvertimm=[]
                    for sd in emplist[0].EmployeeHolidays:
                        overtimeint=[sd.HolidayEmpId,thbll.getHolidayOvertime(sd.Holdays)]
                        getIDOvertimm.append(overtimeint)
                    #print(getIDOvertimm)
                    dov=thbll.getEmpleeIdOvertime(1,getIDOvertimm)
                    #print("加班 id=1 sum:",dov)
                    #补休
                    getIdDeferred=[]
                    for sd in emplist[0].EmployeeHolidays:
                        defint=[sd.HolidayEmpId,thbll.getHolidayDeferred(sd.Holdays)]
                        getIdDeferred.append(defint)
                    #print(getIdDeferred)
                    doef=thbll.getEmpleeIdeferred(1,getIdDeferred)
                    #print("补休 id=1 sum:",doef)

                    #特假 SpecialLeave
                    getIdSpecialLeave=[]
                    for sd in emplist[0].EmployeeHolidays:
                        defint=[sd.HolidayEmpId,thbll.getSpecialLeave(sd.Holdays)]
                        getIdSpecialLeave.append(defint)

                    #事假
                    getIdPersonalLeave=[]
                    for sd in emplist[0].EmployeeHolidays:
                        defint=[sd.HolidayEmpId,thbll.getPersonalLeave(sd.Holdays)]
                        getIdPersonalLeave.append(defint)

                    #病假
                    getIdSickLeave=[]
                    for sd in emplist[0].EmployeeHolidays:
                        defint=[sd.HolidayEmpId,thbll.getSickLeave(sd.Holdays)]
                        getIdSickLeave.append(defint)

                    #迟到
                    getIdLateTime=[]
                    for sd in emplist[0].EmployeeHolidays:
                        defint=[sd.HolidayEmpId,thbll.getLateTime(sd.Holdays)]
                        getIdLateTime.append(defint)



                    #育儿假
                    getIdChildcareLeave=[]
                    for sd in emplist[0].EmployeeHolidays:
                        defint=[sd.HolidayEmpId,thbll.getChildcareLeave(sd.Holdays)]
                        getIdChildcareLeave.append(defint)


                    #例休
                    getIdRegularHoliday=[]
                    for sd in emplist[0].EmployeeHolidays:
                        defint=[sd.HolidayEmpId,thbll.getRegularHoliday(sd.Holdays)]
                        getIdRegularHoliday.append(defint)

                    taum= []
                    tva=("名称","合计")
                    taum.append(tva)
                    tva=("年假", str(AnnualLeave) + "小时")
                    taum.append(tva)
                    tva=("加班",str(Overtime)+"小时")
                    taum.append(tva)
                    tva=("补休",str(DeferredHoliday)+"小时")
                    taum.append(tva)
                    tva=("特假",str(SpecialLeave)+"小时")
                    taum.append(tva)
                    tva=("事假",str(PersonalLeave)+"小时")
                    taum.append(tva)
                    tva=("病假",str(SickLeave)+"小时")
                    taum.append(tva)
                    tva=("迟到",str(LateTime)+"分钟")
                    taum.append(tva)
                    tva=("育儿假",str(ChildcareLeave)+"小时")
                    taum.append(tva)
                    tva=("例休",str(RegularHoliday)+"天")
                    taum.append(tva)
                    sheetname1 = str(year) + "年" + str(mon) + "月假期"
                    writer = pd.ExcelWriter(sheetname1+"明細.xlsx")  # 这里是创建了可写入不同sheet的文件
                    text1 = pd.DataFrame(flist,columns=['ID','工号','部门','姓名','日期','假期'])
                    text1.to_excel(writer, sheet_name=sheetname1,header=0, index=False)  # sheet命名为
                    sheetname2=str(year)+"年"+str(mon)+"月合计"
                    text2 = pd.DataFrame(taum) #,columns=['名称','合计']
                    text2.to_excel(writer, sheet_name=sheetname2, header=0, index=False)  # sheet命名为
                    text3=dataframe1;
                    sheetname3=str(year) + "年" + str(mon) + "月明細"
                    text3.to_excel(writer,sheet_name=sheetname3, header=0, index=False)
                    #writer.sheets.update()
                    writer.close()


                    dataframe2 = openws.load_workbook(sourcefile)
                    # Define variable to read sheet
                    sheet = dataframe2.active
                    setrow=datarowcol[0] #总共行
                    setcol=datarowcol[1] #总共列
                    #每个人的合计
                    for idd in range(1, len(idlist)):
                        sheet.cell(row=setrow - 1, column=1+idd).value = str(thbll.getEmpleeIdAnnualLeave(idd,getIdAnnualLeave)) + "小时時"
                        sheet.cell(row=setrow - 0, column=1+idd).value = str(thbll.getEmpleeIdOvertime(idd,getIDOvertimm))+"小时"
                        sheet.cell(row=setrow + 1, column=1+idd).value = str(thbll.getEmpleeIdeferred(idd,getIdDeferred))+"小时"

                        sheet.cell(row=setrow + 2, column=1 + idd).value = str(thbll.getEmpleeIdeferred(idd, getIdSpecialLeave)) + "小时"
                        sheet.cell(row=setrow + 3, column=1 + idd).value = str(thbll.getEmpleeIdeferred(idd, getIdPersonalLeave)) + "小时"
                        sheet.cell(row=setrow + 4, column=1 + idd).value = str(thbll.getEmpleeIdeferred(idd, getIdSickLeave)) + "小时"
                        sheet.cell(row=setrow + 5, column=1 + idd).value = str(thbll.getEmpleeIdeferred(idd, getIdLateTime)) + "分鈡"
                        sheet.cell(row=setrow + 6, column=1 + idd).value = str(thbll.getEmpleeIdeferred(idd, getIdChildcareLeave)) + "小时"
                        sheet.cell(row=setrow + 7, column=1 + idd).value = str(thbll.getEmpleeIdeferred(idd, getIdRegularHoliday)) + "天"

                    sheet.cell(row=setrow + 2, column=1).value = "本月特假汇总時數"
                    sheet.cell(row=setrow + 3, column=1).value = "本月事假汇总時數"
                    sheet.cell(row=setrow + 4, column=1).value = "本月病假汇总時數"
                    sheet.cell(row=setrow + 5, column=1).value = "本月遲到汇总分鈡數"
                    sheet.cell(row=setrow + 6, column=1).value = "本月育兒假汇总時數"
                    sheet.cell(row=setrow + 7, column=1).value = "本月例休汇总天數"

                    #合部汇总
                    sheet.cell(row=setrow + 8, column=1).value = "年假汇总"
                    sheet.cell(row=setrow + 9, column=1).value = "加班汇总"
                    sheet.cell(row=setrow + 10, column=1).value = "补休汇总"
                    sheet.cell(row=setrow + 11, column=1).value = "特假汇总"
                    sheet.cell(row=setrow + 12, column=1).value = "事假汇总"
                    sheet.cell(row=setrow + 13, column=1).value = "病假汇总"
                    sheet.cell(row=setrow + 14, column=1).value = "遲到汇总"
                    sheet.cell(row=setrow + 15, column=1).value = "育兒假汇总"
                    sheet.cell(row=setrow + 16, column=1).value = "例休汇总"

                    sheet.cell(row=setrow + 8, column=2).value = str(AnnualLeave) + "小時"
                    sheet.cell(row=setrow + 9, column=2).value = str(Overtime)+"小時"
                    sheet.cell(row=setrow + 10, column=2).value = str(DeferredHoliday)+"小時"
                    sheet.cell(row=setrow + 11, column=2).value = str(SpecialLeave) + "小時"
                    sheet.cell(row=setrow + 12, column=2).value = str(PersonalLeave)+"小時"
                    sheet.cell(row=setrow + 13, column=2).value = str(SickLeave)+"小時"
                    sheet.cell(row=setrow + 14, column=2).value = str(LateTime) + "分鈡"
                    sheet.cell(row=setrow + 15, column=2).value = str(ChildcareLeave)+"小時"
                    sheet.cell(row=setrow + 16, column=2).value = str(RegularHoliday)+"天"

                    dataframe2.save(newfile) #创建新文件
        except Exception as ex:
            print(ex)
            pass

        finally:
            print("文件操作成功！")
            pass

　　

调用：

    sourcefile="2023年7月.xlsx"
    newfile="2023年7月new.xlsx"


    export=BLL.ExportExcel.ExportExcelFile()
    sheetnames =export.readSheet(sourcefile)
    for objsheet in sheetnames:
        print(objsheet.SheetName)
    export.exportFile(sourcefile,newfile)