# encoding: utf-8
# 版权所有 2023 涂聚文有限公司
# 许可信息查看：
# 描述：
# Author    : geovindu,Geovin Du 涂聚文.
# IDE       : PyCharm 2023.1 python 311
# Datetime  : 2023/9/3 7:04
# User      : geovindu
# Product   : PyCharm
# Project   : LukfookLeaveCalculation
# File      : EmpLoyeeHolidaysGet.py
# explain   : 学习
 
 
import sys
import os
import Common.Utils
import Model.Holiday
 
 
class EmpLoyeeHolidaysGet(object):
    """
    抽取需要的数据
    """
 
    def getYear(self,strsource):
        """
 
        :param strsource:
        :return:
        """
        refloat = 0
        if (strsource != None):
            if ("年" in strsource):
                refloat=Common.Utils.Utils.getYear(strsource)
        return refloat
 
    def getMonth(self, strsource):
        """
 
        :param strsource:
        :return:
        """
        refloat = 0
        if (strsource != None):
            if ("月" in strsource):
                refloat = Common.Utils.Utils.getMonth(strsource)
        return refloat
 
    def getDay(self, strsource):
        """
 
        :param strsource:
        :return:
        """
        refloat = 0
        if (strsource != None):
            if ("日" in strsource):
                refloat = Common.Utils.Utils.getDay(strsource)
        return refloat
 
    def getHolidays(self,strsource):
        """
        獲得假期限列表
        :param strsource:
        :return:
        """
        hday=[]
        if (strsource != None):
            if("星期" not in strsource):
                if("nan" not in strsource):
                    spstr = strsource.split(' ')  # 有两个，有换行符，两个假期，一定要这换行符  NaN
                    if (len(spstr) > 1):
                        for i in range(len(spstr)):
                            sk = spstr[i]
                            #print("for sk", sk)
                            hd = Model.Holiday.Holiday()
                            hd=EmpLoyeeHolidaysGet.checkHoliday(self,sk)
                            hday.append(hd)
 
                    elif (len(spstr) == 1):
                        sk = spstr[0]
                        #print("sk",sk)
                        hd = Model.Holiday.Holiday()
                        hd = EmpLoyeeHolidaysGet.checkHoliday(self,sk)
                        hday.append(hd)
 
                    else:
                        hday = []
 
        else:
            hday = []
 
        return hday
 
 
    def checkHoliday(self,strsource):
        """
        检测是什么假期
        :param strsource:
        :return:
        """
        hd=Model.Holiday.Holiday()
        #print("chek")
        if ("例休" in strsource):
            #print("cehck例休")
            hd.HolidayId=1
            hd.HolidayName="例休"
            hd.WorkTime=Common.Utils.Utils.getRegularHoliday(strsource)
 
        if ("特假" in strsource):
            hd.HolidayId=2
            hd.HolidayName="特假"
            hd.workTime=Common.Utils.Utils.getSpecialLeave(strsource)
 
        if ("補休" in strsource):
            hd.HolidayId=3
            hd.HolidayName="補休"
            hd.workTime = Common.Utils.Utils.getDeferredHoliday(strsource)
 
 
        if ("加班" in strsource):
            hd.HolidayId=4
            hd.HolidayName="加班"
            hd.workTime = Common.Utils.Utils.getOvertime(strsource)
 
        if ("事假" in strsource):
            hd.HolidayId=5
            hd.HolidayName="事假"
            hd.workTime = Common.Utils.Utils.getPersonalLeave(strsource)
 
        if ("年假" in strsource):
            hd.HolidayId = 6
            hd.HolidayName = "年假"
            hd.workTime = Common.Utils.Utils.getAnnualLeave(strsource)
 
 
        if ("病假" in strsource):
            hd.HolidayId = 7
            hd.HolidayName = "病假"
            hd.workTime = Common.Utils.Utils.getSickLeave(strsource)
 
 
        if ("育兒假" in strsource):
            hd.HolidayId = 8
            hd.HolidayName = "育兒假"
            hd.workTime = Common.Utils.Utils.getChildcareLeave(strsource)
 
        if ("遲到" in strsource):
            hd.HolidayId = 9
            hd.HolidayName = "遲到"
            hd.workTime = Common.Utils.Utils.getLateTime(strsource)
 
 
        return  hd
 
    def getEmplee(self,Empleelist,id:int):
        """
        從員工列表中找到對應ID員工的信息
        :param Empleelist:
        :param id:
        :return:
        """
        ee=[]
        for em in Empleelist:
            if(em.EmployeeId==id):
                ee.append(em.EmployeeNo)
                ee.append(em.EmployeeDep)
                ee.append(em.EmployeeName)
                return ee
 
 
    def getHoliday(self,holidays):
        """
        返回假期
        :param holidays:
        :return:
        """
        getname=""
        if(len(holidays)>0):
            for hd in holidays:
                if(hd.HolidayId==1):
                    getname=getname+hd.HolidayName+str(hd.WorkTime)+"天 "
                elif(hd.HolidayId==9):
                    getname = getname + hd.HolidayName + str(hd.WorkTime) + "分钟 "
                else:
                    getname = getname + hd.HolidayName + str(hd.WorkTime) + "小时 "
        return getname
 
    def getHolidayAnnualLeave(self,holidays):
        """
        返回年假数
        :param holidays:
        :return:
        """
        getsum=0
        if(len(holidays)>0):
            for hd in holidays:
                if(hd.HolidayId==6):
                    getsum=getsum+hd.WorkTime
                    #print("sum:",getsum,hd.HolidayName)
 
        return getsum
 
    def getHolidayOvertime(self,holidays):
        """
        返回加班数
        :param holidays:
        :return:
        """
        getsum=0
        if(len(holidays)>0):
            for hd in holidays:
                if(hd.HolidayId==4):
                    getsum=getsum+hd.WorkTime
                    #print("sum:",getsum,hd.HolidayName)
 
        return getsum
 
    def getHolidayDeferred(self,holidays):
        """
        返回補休数
        :param holidays:
        :return:
        """
        getsum=0
        if(len(holidays)>0):
            for hd in holidays:
                if(hd.HolidayId==3):
                    getsum=getsum+hd.WorkTime
                    #print("sum:",getsum,hd.HolidayName)
 
        return getsum
 
 
 
    def getEmpleeIdAnnualLeave(self, id:int,listEmp:list):
       """
       年假ID
       :param self:
       :param listEmp:
       :return:
       """
       annual=0
       #print(listEmp)
       for sd in listEmp:
           #print(sd[0],sd[1])
           if(sd[0]==id):
               #print(sd[1])
               annual=annual+sd[1]
       return annual
 
    def getEmpleeIdOvertime(self, id:int,listEmp:list):
       """
       加班时间ID
       :param self:
       :param listEmp:
       :return:
       """
       annual=0
       #print(listEmp)
       for sd in listEmp:
           #print(sd[0],sd[1])
           if(sd[0]==id):
               #print(sd[1])
               annual=annual+sd[1]
       return annual
 
    def getEmpleeIdeferred(self, id:int,listEmp:list):
       """
       调休时间ID
       :param self:
       :param listEmp:
       :return:
       """
       annual=0
       #print(listEmp)
       for sd in listEmp:
           #print(sd[0],sd[1])
           if(sd[0]==id):
               #print(sd[1])
               annual=annual+sd[1]
       return annual

# This is a sample Python script.
# encoding: utf-8
# 版权所有 2023 涂聚文有限公司
# 许可信息查看：
# 描述：
# Author    : geovindu,Geovin Du 涂聚文.
# IDE       : PyCharm 2023.1 python 311
# Datetime  : 2023/9/2 20:20
# User      : geovindu
# Product   : PyCharm
# Project   : LukfookLeaveCalculation
# File      : main.py
# explain   : 学习
# Press Shift+F10 to execute it or replace it with your code.
# Press Double Shift to search everywhere for classes, files, tool windows, actions, and settings.
 
import xlrd
import xlwt
import xlwings as xw
import xlsxwriter
 
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
from openpyxl import load_workbook
 
def print_hi(name):
    # Use a breakpoint in the code line below to debug your script.
    print(f'Hi, {name}')  # Press Ctrl+F8 to toggle the breakpoint.
 
 
# Press the green button in the gutter to run the script.
if __name__ == '__main__':
    print_hi('PyCharm')
    unstr=Common.Utils.Utils()
    strtest="年假2小時 事假0.5小時"  # 换行符
    #strtest-strtest.replace('_x000D_', ' ', regex=True)
    #strtest = "NaN"
    spstr = strtest.split(" ")
    thbll = BLL.EmpLoyeeHolidaysGet.EmpLoyeeHolidaysGet()
    tt=thbll.getHolidays(spstr[0])
    print("tt",tt[0].WorkTime,spstr)
 
    dataframe1 = pd.read_excel('2023年7月.xlsx')
 
    #print(dataframe1)
    #for idx, data in dataframe1.iterrows():
        #print("[{}]: {}".format(idx, data))
    # 第1列为日期，其他列是员工对应的列的考勤
    print("*************")
    print(dataframe1.index)
    print("*************")
    print(dataframe1.shape) #行列
    datarowcol=dataframe1.shape
    print("行：",datarowcol[0],"，列：",datarowcol[1])
    excelRows=datarowcol[0]
    excelColumns=datarowcol[1]
    print("*********列标题****")
    print(dataframe1.columns) #第一行标题   pandas.core.indexes.base.Index
    titls=dataframe1.columns.to_list()
    print(titls)
    print("标题：",titls[1])
    year=thbll.getYear(titls[1])
    print("*************",year)
    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)
    print("*******序号******")
    #工号
    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)
 
 
    #data1 = dataframe1.loc[0:3]
    #print(type(data1))
    #print(data1)
    #for idx, data in data1.iterrows():
        #print("[{}]: {}".format(idx, data))
    #print("*************")
 
    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)
 
    '''
     
 
    '''
    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("2023年7月.xlsx")
    # 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).value = "年假汇总"
    sheet.cell(row=setrow +3, column=1).value = "加班汇总"
    sheet.cell(row=setrow +4, column=1).value = "补休汇总"
    sheet.cell(row=setrow+2, column=2).value = str(AnnualLeave) + "小时"
    sheet.cell(row=setrow +3, column=2).value = str(Overtime)+"小时"
    sheet.cell(row=setrow +4, column=2).value = str(DeferredHoliday)+"小时"
 
    dataframe2.save(r'demo.xlsx')
    # Iterate the loop to read the cell values
    #for row in range(0, dataframe2.max_row):
        #for col in dataframe2.iter_cols(1, dataframe2.max_column):
           #print(col[row].value)
 
    # a single cell
    #v1 = xw.range("A1:A7").value
    #v2 = xw.range("F5").value
    #print("Result:", v1)
 
# https://www.geeksforgeeks.org/how-to-add-one-row-in-an-existing-pandas-dataframe/ 新添加一行
# See PyCharm help at https://www.jetbrains.com/help/pycharm/