Add multi-company support hr_holidays_usability

Add modules hr_holidays_lunch_voucher and hr_holidays_lunch_voucher_natixis
2017-05-11 00:29:22 +02:00
parent 00592100f7
commit 193e5e6ecc
30 changed files with 792 additions and 4 deletions
--- a/hr_holidays_lunch_voucher/wizard/hr_holidays_post.py
+++ b/hr_holidays_lunch_voucher/wizard/hr_holidays_post.py
@@ -0,0 +1,74 @@
+# -*- coding: utf-8 -*-
+# © 2017 Akretion (http://www.akretion.com)
+# @author Alexis de Lattre <alexis.delattre@akretion.com>
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from openerp import models, fields, api
+from dateutil.relativedelta import relativedelta
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+class HrHolidaysPost(models.TransientModel):
+    _inherit = 'hr.holidays.post'
+
+    lunch_voucher_po = fields.Boolean(
+        string='Generate Lunch Voucher Purchase Order')
+    work_days = fields.Integer(string='Work Days')
+
+    @api.model
+    def default_get(self, fields_list):
+        res = super(HrHolidaysPost, self).default_get(fields_list)
+        hhpo = self.env['hr.holidays.public']
+        company = self.env.user.company_id
+        if company.lunch_voucher_product_id:
+            res['lunch_voucher_po'] = True
+        today = fields.Date.context_today(self)
+        today_dt = fields.Date.from_string(today)
+        cur_month = today_dt.month
+        # last day of month
+        date_dt = today_dt + relativedelta(day=31)
+        work_days = date_dt.day
+        logger.info('Number of days in the month: %d', work_days)
+        # from last day of month to the first
+        while date_dt.month == cur_month:
+            if hhpo.is_public_holiday(date_dt):
+                work_days -= 1
+                logger.info(
+                    "%s is a bank holiday, don't count", date_dt)
+            # if it's a saturday/sunday
+            elif date_dt.weekday() in (5, 6):
+                work_days -= 1
+                logger.info(
+                    "%s is a saturday/sunday, don't count", date_dt)
+            date_dt += relativedelta(days=-1)
+        logger.info('Number of work days in the month: %d', work_days)
+        res['work_days'] = work_days
+        return res
+
+    @api.multi
+    def run(self):
+        self.ensure_one()
+        lvao = self.env['lunch.voucher.attribution']
+        today = fields.Date.context_today(self)
+        employees = self.env['hr.employee'].search([
+            ('lunch_voucher', '=', True),
+            ('company_id', '=', self.env.user.company_id.id),
+            ])
+        lv_dict = {}
+        for employee in employees:
+            lv_dict[employee.id] = []
+        for hol in self.holidays_to_post_ids:
+            if not hol.lunch_voucher_id and hol.employee_id.id in lv_dict:
+                lv_dict[hol.employee_id.id].append(hol.id)
+        for employee_id, hol_ids in lv_dict.iteritems():
+            vals = {
+                'employee_id': employee_id,
+                'date': today,
+                'month_work_days': self.work_days,
+                }
+            if hol_ids:
+                vals['holiday_ids'] = [(6, 0, hol_ids)]
+            lvao.create(vals)
+        return super(HrHolidaysPost, self).run()