# -*- 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):
    """For handling importing of employee attendance"""
    _name = 'import.attendance'
    _description = 'A model for handling importing of attendance'

    name = fields.Char(string="Name", help="Name", default="Import Attendance")
    file_type = fields.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="For uploading files")

    def action_attendance_import(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
        error_msg = ""
        imported = 0
        row = 0
        for item in datas:
            vals = {}
            row_not_import_msg = "\n❌Row {rn} not imported.".format(rn=row)
            if item.get('Employee'):
                employee = hr_employee.search(
                    [('name', '=', item.get('Employee'))])
                if employee:
                    vals['employee_id'] = employee.id
                else:
                    error_msg += row_not_import_msg + (
                            "\n\t⚠ There is no employee with that name."
                            "found!"
                            % item.get('Employee'))
            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)
            imported += 1
        if error_msg:
            error_msg = "\n\n🏮 WARNING 🏮" + error_msg
        msg = (("Imported %d records."
                % imported) + error_msg)
        message = self.env['import.message'].create(
            {'message': msg})
        if message:
            return {
                'effect': {
                    'fadeout': 'slow',
                    'message': 'Imported Successfully',
                    'type': 'rainbow_man',
                }
            }