Spaces:
Runtime error
Runtime error
File size: 1,705 Bytes
1748405 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
from typing import Dict, List
import pandas as pd
from config import CONFIG
from .entities import Employee, WorkingTime
def get_data(data: pd.DataFrame) -> pd.DataFrame:
data.columns = data.iloc[1, :]
data = data.iloc[2:, ]
data = data[['Mã N.Viên', "Tên nhân viên", "Ngày", "Vào", "Ra"]]
data = data.groupby(by=['Mã N.Viên', 'Ngày']).aggregate({
"Tên nhân viên": max,
"Vào": sum,
"Ra": sum
}).reset_index()
return data
def parse(data: pd.DataFrame) -> List[Employee]:
employees: List[Employee] = []
for (id, name), v in data.groupby(by=['Mã N.Viên', 'Tên nhân viên']):
employee = Employee(id=id, name=name, workdate={})
for val in v.values:
working_time = WorkingTime(date=val[1],
checkin=val[3],
checkout=val[4],
config=CONFIG)
date, work_time = working_time.to_tuple()
employee.workdate[date] = work_time
employees.append(employee)
return employees
def make_table(employees: List[Employee]) -> pd.DataFrame:
id_list: List[str] = []
name_list: List[str] = []
work_date: List[Dict] = []
for employee in employees:
id_list.append(employee.id)
name_list.append(employee.name)
work_date.append(employee.workdate)
work_df: pd.DataFrame = pd.DataFrame(work_date, dtype=int)
info_df: pd.DataFrame = pd.DataFrame({
'Mã nhân viên': id_list,
'Tên nhân viên': name_list
})
working_table: pd.DataFrame = pd.concat(objs=[info_df, work_df], axis=1)
return working_table
|