98 lines
4.5 KiB
Python
98 lines
4.5 KiB
Python
# -*- 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
|