initial commit 2

4 files changed, 220 insertions, 0 deletions

diff --git a/addons/hr_presence/models/__init__.py b/addons/hr_presence/models/__init__.py
new file mode 100644
index 00000000..76aa0205
--- /dev/null
+++ b/addons/hr_presence/models/__init__.py
@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from . import hr_employee
+from . import res_company
+from . import res_users_log
diff --git a/addons/hr_presence/models/hr_employee.py b/addons/hr_presence/models/hr_employee.py
new file mode 100644
index 00000000..3da6c9f5
--- /dev/null
+++ b/addons/hr_presence/models/hr_employee.py
@@ -0,0 +1,194 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+import logging
+
+from ast import literal_eval
+from odoo import fields, models, _, api
+from odoo.exceptions import UserError
+from odoo.fields import Datetime
+
+_logger = logging.getLogger(__name__)
+
+
+class Employee(models.AbstractModel):
+    _inherit = 'hr.employee.base'
+
+    email_sent = fields.Boolean(default=False)
+    ip_connected = fields.Boolean(default=False)
+    manually_set_present = fields.Boolean(default=False)
+
+    # Stored field used in the presence kanban reporting view
+    # to allow group by state.
+    hr_presence_state_display = fields.Selection([
+        ('to_define', 'To Define'),
+        ('present', 'Present'),
+        ('absent', 'Absent'),
+        ])
+
+    def _compute_presence_state(self):
+        super()._compute_presence_state()
+        employees = self.filtered(lambda e: e.hr_presence_state != 'present' and not e.is_absent)
+        company = self.env.company
+        employee_to_check_working = employees.filtered(lambda e:
+                                                       not e.is_absent and
+                                                       (e.email_sent or e.ip_connected or e.manually_set_present))
+        working_now_list = employee_to_check_working._get_employee_working_now()
+        for employee in employees:
+            if not employee.is_absent and company.hr_presence_last_compute_date and employee.id in working_now_list and \
+                    company.hr_presence_last_compute_date.day == Datetime.now().day and \
+                    (employee.email_sent or employee.ip_connected or employee.manually_set_present):
+                employee.hr_presence_state = 'present'
+
+    @api.model
+    def _check_presence(self):
+        company = self.env.company
+        if not company.hr_presence_last_compute_date or \
+                company.hr_presence_last_compute_date.day != Datetime.now().day:
+            self.env['hr.employee'].search([
+                ('company_id', '=', company.id)
+            ]).write({
+                'email_sent': False,
+                'ip_connected': False,
+                'manually_set_present': False
+            })
+
+        employees = self.env['hr.employee'].search([('company_id', '=', company.id)])
+        all_employees = employees
+
+
+        # Check on IP
+        if literal_eval(self.env['ir.config_parameter'].sudo().get_param('hr.hr_presence_control_ip', 'False')):
+            ip_list = company.hr_presence_control_ip_list
+            ip_list = ip_list.split(',') if ip_list else []
+            ip_employees = self.env['hr.employee']
+            for employee in employees:
+                employee_ips = self.env['res.users.log'].search([
+                    ('create_uid', '=', employee.user_id.id),
+                    ('ip', '!=', False),
+                    ('create_date', '>=', Datetime.to_string(Datetime.now().replace(hour=0, minute=0, second=0, microsecond=0)))]
+                ).mapped('ip')
+                if any(ip in ip_list for ip in employee_ips):
+                    ip_employees |= employee
+            ip_employees.write({'ip_connected': True})
+            employees = employees - ip_employees
+
+        # Check on sent emails
+        if literal_eval(self.env['ir.config_parameter'].sudo().get_param('hr.hr_presence_control_email', 'False')):
+            email_employees = self.env['hr.employee']
+            threshold = company.hr_presence_control_email_amount
+            for employee in employees:
+                sent_emails = self.env['mail.message'].search_count([
+                    ('author_id', '=', employee.user_id.partner_id.id),
+                    ('date', '>=', Datetime.to_string(Datetime.now().replace(hour=0, minute=0, second=0, microsecond=0))),
+                    ('date', '<=', Datetime.to_string(Datetime.now()))])
+                if sent_emails >= threshold:
+                    email_employees |= employee
+            email_employees.write({'email_sent': True})
+            employees = employees - email_employees
+
+        company.sudo().hr_presence_last_compute_date = Datetime.now()
+
+        for employee in all_employees:
+            employee.hr_presence_state_display = employee.hr_presence_state
+
+    @api.model
+    def _action_open_presence_view(self):
+        # Compute the presence/absence for the employees on the same
+        # company than the HR/manager. Then opens the kanban view
+        # of the employees with an undefined presence/absence
+
+        _logger.info("Employees presence checked by: %s" % self.env.user.name)
+
+        self._check_presence()
+
+        return {
+            "type": "ir.actions.act_window",
+            "res_model": "hr.employee",
+            "views": [[self.env.ref('hr_presence.hr_employee_view_kanban').id, "kanban"], [False, "tree"], [False, "form"]],
+            'view_mode': 'kanban,tree,form',
+            "domain": [],
+            "name": "Employee's Presence to Define",
+            "search_view_id": [self.env.ref('hr_presence.hr_employee_view_presence_search').id, 'search'],
+            "context": {'search_default_group_hr_presence_state': 1,
+                        'searchpanel_default_hr_presence_state_display': 'to_define'},
+        }
+
+    def action_set_present(self):
+        if not self.env.user.has_group('hr.group_hr_manager'):
+            raise UserError(_("You don't have the right to do this. Please contact an Administrator."))
+        self.write({'manually_set_present': True})
+
+    def write(self, vals):
+        if vals.get('hr_presence_state_display') == 'present':
+            vals['manually_set_present'] = True
+        return super().write(vals)
+
+    def action_open_leave_request(self):
+        self.ensure_one()
+        return {
+            "type": "ir.actions.act_window",
+            "res_model": "hr.leave",
+            "views": [[False, "form"]],
+            "view_mode": 'form',
+            "context": {'default_employee_id': self.id},
+        }
+
+    # --------------------------------------------------
+    # Messaging
+    # --------------------------------------------------
+
+    def action_send_sms(self):
+        self.ensure_one()
+        if not self.env.user.has_group('hr.group_hr_manager'):
+            raise UserError(_("You don't have the right to do this. Please contact an Administrator."))
+        if not self.mobile_phone:
+            raise UserError(_("There is no professional mobile for this employee."))
+
+        context = dict(self.env.context)
+        context.update(default_res_model='hr.employee', default_res_id=self.id, default_composition_mode='comment', default_number_field_name='mobile_phone')
+
+        template = self.env.ref('hr_presence.sms_template_presence', False)
+        if not template:
+            context['default_body'] = _("""Exception made if there was a mistake of ours, it seems that you are not at your office and there is not request of time off from you.
+Please, take appropriate measures in order to carry out this work absence.
+Do not hesitate to contact your manager or the human resource department.""")
+        else:
+            context['default_template_id'] = template.id
+
+        return {
+            "type": "ir.actions.act_window",
+            "res_model": "sms.composer",
+            "view_mode": 'form',
+            "context": context,
+            "name": "Send SMS Text Message",
+            "target": "new",
+        }
+
+    def action_send_mail(self):
+        self.ensure_one()
+        if not self.env.user.has_group('hr.group_hr_manager'):
+            raise UserError(_("You don't have the right to do this. Please contact an Administrator."))
+        if not self.work_email:
+            raise UserError(_("There is no professional email address for this employee."))
+        template = self.env.ref('hr_presence.mail_template_presence', False)
+        compose_form = self.env.ref('mail.email_compose_message_wizard_form', False)
+        ctx = dict(
+            default_model="hr.employee",
+            default_res_id=self.id,
+            default_use_template=bool(template),
+            default_template_id=template.id,
+            default_composition_mode='comment',
+            default_is_log=True,
+            custom_layout='mail.mail_notification_light',
+        )
+        return {
+            'name': _('Compose Email'),
+            'type': 'ir.actions.act_window',
+            'view_mode': 'form',
+            'res_model': 'mail.compose.message',
+            'views': [(compose_form.id, 'form')],
+            'view_id': compose_form.id,
+            'target': 'new',
+            'context': ctx,
+        }
diff --git a/addons/hr_presence/models/res_company.py b/addons/hr_presence/models/res_company.py
new file mode 100644
index 00000000..834fd26c
--- /dev/null
+++ b/addons/hr_presence/models/res_company.py
@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import fields, models
+
+
+class ResCompany(models.Model):
+    _inherit = 'res.company'
+
+    hr_presence_last_compute_date = fields.Datetime()
diff --git a/addons/hr_presence/models/res_users_log.py b/addons/hr_presence/models/res_users_log.py
new file mode 100644
index 00000000..9ad3c97d
--- /dev/null
+++ b/addons/hr_presence/models/res_users_log.py
@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+
+from odoo import api, fields, models
+
+
+class ResUsersLog(models.Model):
+    _inherit = 'res.users.log'
+
+    create_uid = fields.Integer(index=True)
+    ip = fields.Char(string="IP Address")