add gym module

gym/models/trainer.py

@@ -0,0 +1,66 @@
+# See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models, _
+from odoo.exceptions import ValidationError
+
+
+class HrEmployee(models.Model):
+    """Inherited this model for trainer's information."""
+
+    _inherit = 'hr.employee'
+
+    is_trainer = fields.Boolean(string='Is Trainer ?')
+    specialist_ids = fields.Many2many('gym.skills', 'trainer_skill_rel',
+                                      'trainer_id', 'skill_id',
+                                      string='Specialization')
+    total_session = fields.Integer(
+        compute='_compute_calculation_trainer_session',
+        string='Total Session')
+    user_id = fields.Many2one('res.users', 'User',
+                              related='resource_id.user_id', store=True,
+                              readonly=False, ondelete="cascade")
+
+    def unlink(self):
+        for rec in self:
+            user_id = rec.user_id
+            rec.user_id = False
+            user_id.unlink()
+        return super(HrEmployee, self).unlink()
+
+    def _compute_calculation_trainer_session(self):
+        """Trainer session count for session smart
+        button in trainer form(hr.employee)."""
+        for trainer_rec in self:
+            trainer_rec.total_session = \
+                self.env['calendar.event'].search_count([
+                    ('trainer_id', '=', trainer_rec.id)])
+
+    @api.constrains('work_email')
+    def _check_work_email(self):
+        rec = self.search([('work_email', '=', self.work_email),
+                           ('id', '!=', self.id)])
+        if rec:
+            raise ValidationError(_("You can not have two users with the "
+                                    "same login !"))
+
+    @api.model
+    def create(self, vals):
+        """Method for creating User at trainer creation."""
+        res = super(HrEmployee, self).create(vals)
+        if not vals.get('user_id'):
+            user = self.env['res.users'].create({
+                'name': vals.get('name'),
+                'login': vals.get('work_email', False),
+                'password': self.env.user.company_id.default_password or '',
+                'groups_id': [(6, 0, [self.env.ref(
+                    'gym.group_gym_trainer').id])],
+            })
+            if user and user.partner_id:
+                user.partner_id.write({'email': vals.get('work_email', False)})
+            res.write({'user_id': user and user.id or False})
+        return res
+
+    def copy(self, default=None):
+        if self.is_trainer:
+            raise ValidationError(_("Sorry, you can't duplicate a trainer. "))
+        return super(HrEmployee, self).copy(default)