# -*- coding: utf-8 -*-

##############################################################################
#
#    Clear Groups for Odoo
#    Copyright (C) 2016 Bytebrand GmbH (<http://www.bytebrand.net>).
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Lesser General Public License as
#    published by the Free Software Foundation, either version 3 of the
#    License, or (at your option) any later version.
#
#    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 Affero General Public License for more details.
#
#    You should have received a copy of the GNU Lesser General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################
from datetime import datetime, timedelta
import pytz

from openerp.osv import fields, osv
from openerp.tools import DEFAULT_SERVER_DATETIME_FORMAT
from openerp.tools.translate import _


class import_leave_requests(osv.osv_memory):
    _name = 'import.leave.requests'
    _description = 'Import Leave Requests With Employee Tag'
    _columns = {
        'leave_dates': fields.binary('Select *.csv',
                                     required=True,
                                     help="Select csv file having "
                                          "holiday dates."),
        'leave_type_id': fields.many2one('hr.holidays.status',
                                         'Leave Type',
                                         required=True),
        'employee_tag_id': fields.many2one('hr.employee.category',
                                           "Employee Tag", required=True),
    }

    def convert_to_user_timezone(self, user_tz, dt):
        input_tz = pytz.timezone(user_tz)
        converted_date = input_tz.localize(dt, is_dst=False)
        converted_date = converted_date.astimezone(pytz.UTC).strftime(
            DEFAULT_SERVER_DATETIME_FORMAT)
        return converted_date

    def import_leave_data(self, cr, uid, ids, context=None):
        holiday_obj = self.pool.get('hr.holidays')
        employee_obj = self.pool.get('hr.employee')
        timesheet_obj = self.pool.get('hr_timesheet_sheet.sheet')
        converter = self.pool.get('ir.fields.converter')
        for data in self.browse(cr, uid, ids, context):
            leaves = (data.leave_dates.decode('base64')).split('\n')
            category_id = data.employee_tag_id.id
            employee_ids = employee_obj.search(cr, uid, [
                ('category_ids', 'in', [category_id])], context=context)
            for employee_id in employee_ids:
                employee = employee_obj.browse(cr, uid, employee_id, context)
                for leave in leaves[:-1]:
                    dt_fmt, tm_fmt = \
                        (timesheet_obj._get_user_datetime_format(
                            cr,
                            uid,
                            context=context))
                    try:
                        datetime.strptime(leave, dt_fmt)
                    except ValueError:
                        raise osv.except_osv(_('Data Error!'), _(
                            "Date format in your .csv file does not "
                            "match with database date format."))
                    dt1 = datetime.strptime(leave, dt_fmt)
                    dt2 = (datetime.strptime(leave, dt_fmt) +
                           timedelta(hours=23,
                                     minutes=59,
                                     seconds=59))
                    user_tz = employee.user_id and employee.user_id.tz or 'utc'
                    leave_date = self.convert_to_user_timezone(user_tz, dt1)
                    leave_date_to = self.convert_to_user_timezone(user_tz, dt2)
                    holiday_id = holiday_obj.create(cr, uid, {
                        'name': data.leave_type_id.name,
                        'date_from': leave_date,
                        'date_to': leave_date_to,
                        'holiday_status_id': data.leave_type_id.id,
                        'employee_id': employee_id,
                        'number_of_days_temp': 1.0,
                        'type': 'remove'
                    })
                    holiday_obj.holidays_validate(cr, uid, [holiday_id],
                                                  context=context)
        return True

        # END