You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
4.3 KiB
109 lines
4.3 KiB
# -*- coding: utf-8 -*-
|
|
#############################################################################
|
|
#
|
|
# Cybrosys Technologies Pvt. Ltd.
|
|
#
|
|
# Copyright (C) 2024-TODAY Cybrosys Technologies(<https://www.cybrosys.com>)
|
|
# Author: Cybrosys Techno Solutions(<https://www.cybrosys.com>)
|
|
#
|
|
# You can modify it under the terms of the GNU LESSER
|
|
# GENERAL PUBLIC LICENSE (LGPL v3), Version 3.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU LESSER GENERAL PUBLIC LICENSE (LGPL v3) for more details.
|
|
#
|
|
# You should have received a copy of the GNU LESSER GENERAL PUBLIC LICENSE
|
|
# (LGPL v3) along with this program.
|
|
# If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
#############################################################################
|
|
import base64
|
|
import binascii
|
|
import csv
|
|
import io
|
|
import tempfile
|
|
import xlrd
|
|
from odoo import fields, models
|
|
from odoo.exceptions import ValidationError
|
|
|
|
|
|
class ImportAttendance(models.TransientModel):
|
|
""" Model for Attendance import wizard. """
|
|
_name = 'import.attendance'
|
|
_description = 'Attendance Import'
|
|
|
|
file_type = fields.Selection(
|
|
selection=[('csv', 'CSV File'), ('xls', 'XLS File')],
|
|
string='Select File Type', default='csv',
|
|
help="It helps to select File Type")
|
|
file_upload = fields.Binary(string="Upload File",
|
|
help="It helps to upload files")
|
|
|
|
def action_import_attendance(self):
|
|
"""Creating attendance record using uploaded xl/csv files"""
|
|
hr_employee = self.env['hr.employee']
|
|
hr_attendance = self.env['hr.attendance']
|
|
datas = {}
|
|
if self.file_type == 'csv':
|
|
try:
|
|
csv_data = base64.b64decode(self.file_upload)
|
|
data_file = io.StringIO(csv_data.decode("utf-8"))
|
|
data_file.seek(0)
|
|
datas = csv.DictReader(data_file, delimiter=',')
|
|
except:
|
|
raise ValidationError(
|
|
"File not Valid.\n\nPlease check the type and format "
|
|
"of the file and try again!")
|
|
if self.file_type == 'xls':
|
|
try:
|
|
fp = tempfile.NamedTemporaryFile(delete=False,
|
|
suffix=".xlsx")
|
|
fp.write(binascii.a2b_base64(self.file_upload))
|
|
fp.seek(0)
|
|
workbook = xlrd.open_workbook(fp.name)
|
|
sheet = workbook.sheet_by_index(0)
|
|
except:
|
|
raise ValidationError(
|
|
"""File not Valid.\n\nPlease check the """
|
|
"""type and format of the file and try again!""")
|
|
headers = sheet.row_values(0)
|
|
data = []
|
|
for row_index in range(1, sheet.nrows):
|
|
row = sheet.row_values(row_index)
|
|
data += [{k: v for k, v in zip(headers, row)}]
|
|
datas = data
|
|
for item in datas:
|
|
vals = {}
|
|
if item.get('Employee'):
|
|
employee = hr_employee.search(
|
|
[('name', '=', item.get('Employee'))])
|
|
if employee:
|
|
vals['employee_id'] = employee.id
|
|
else:
|
|
raise ValidationError(
|
|
"There is no employee with that name."
|
|
"\n\nPlease check and try again!")
|
|
if item.get('Check In'):
|
|
if self.file_type == 'csv':
|
|
vals['check_in'] = item.get('Check In')
|
|
else:
|
|
vals['check_in'] = xlrd.xldate_as_datetime(
|
|
item.get('Check In'), 0)
|
|
if item.get('Check Out'):
|
|
if self.file_type == 'csv':
|
|
vals['check_out'] = item.get('Check Out')
|
|
else:
|
|
vals['check_out'] = xlrd.xldate_as_datetime(
|
|
item.get('Check Out'), 0)
|
|
if item.get('Worked Hours'):
|
|
vals['worked_hours'] = item.get('Worked Hours')
|
|
hr_attendance.create(vals)
|
|
return {
|
|
'effect': {
|
|
'fadeout': 'slow',
|
|
'message': 'Imported Successfully',
|
|
'type': 'rainbow_man',
|
|
}
|
|
}
|
|
|