You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

77 lines
3.2 KiB

  1. # -*- coding: utf-8 -*-
  2. from openerp import models, fields, api, _
  3. from openerp.exceptions import ValidationError
  4. from openerp.addons.beesdoo_base.tools import concat_names
  5. class Partner(models.Model):
  6. _inherit = 'res.partner'
  7. first_name = fields.Char('First Name')
  8. last_name = fields.Char('Last Name')
  9. eater = fields.Selection([('eater', 'Eater'), ('worker_eater', 'Worker and Eater')], string="Eater/Worker")
  10. child_eater_ids = fields.One2many("res.partner", "parent_eater_id", domain=[('customer', '=', True),
  11. ('eater', '=', 'eater')])
  12. parent_eater_id = fields.Many2one("res.partner", string="Parent Worker", readonly=True)
  13. barcode = fields.Char(compute="_get_bar_code", string='Bar Code', store=True)
  14. parent_barcode = fields.Char(compute="_get_bar_code", string='Parent Bar Code', store=True)
  15. member_card_ids = fields.One2many('member.card', 'partner_id')
  16. member_card_to_be_printed = fields.Boolean('Print BEES card?')
  17. last_printed = fields.Datetime('Last printed on')
  18. @api.onchange('first_name', 'last_name')
  19. def _on_change_name(self):
  20. self.name = concat_names(self.first_name, self.last_name)
  21. @api.one
  22. @api.depends('parent_eater_id', 'parent_eater_id.barcode', 'eater', 'member_card_ids')
  23. def _get_bar_code(self):
  24. if self.eater == 'eater':
  25. self.parent_barcode = self.parent_eater_id.barcode
  26. elif self.member_card_ids:
  27. for c in self.member_card_ids:
  28. if c.valid:
  29. self.barcode = c.barcode
  30. @api.one
  31. @api.constrains('child_eater_ids', 'parent_eater_id')
  32. def _only_two_eaters(self):
  33. if len(self.child_eater_ids) > 2 or len(self.parent_eater_id.child_eater_ids) > 2:
  34. raise ValidationError(_('You can only set two additional eaters per worker'))
  35. @api.multi
  36. def write(self, values):
  37. if values.get('parent_eater_id') and self.parent_eater_id:
  38. raise ValidationError(_('You try to assign a eater to a worker but this easer is alread assign to %s please remove it before') % self.parent_eater_id.name)
  39. #replace many2many command when writing on child_eater_ids to just remove the link
  40. if 'child_eater_ids' in values:
  41. for command in values['child_eater_ids']:
  42. if command[0] == 2:
  43. command[0] = 3
  44. return super(Partner, self).write(values)
  45. @api.one
  46. def _deactivate_active_cards(self):
  47. for card in self.member_card_ids.filtered('valid'):
  48. card.valid = False
  49. card.end_date = fields.Date.today()
  50. @api.multi
  51. def _new_card(self, reason, user_id, barcode=False):
  52. card_data = {
  53. 'partner_id' : self.id,
  54. 'responsible_id' : user_id,
  55. 'comment' : reason,
  56. }
  57. if barcode:
  58. card_data['barcode'] = barcode
  59. self.env['member.card'].create(card_data)
  60. @api.noguess
  61. def _auto_init(self, cr, context=None):
  62. res = super(Partner, self)._auto_init(cr, context=context)
  63. cr.execute("UPDATE res_partner set last_name = name where last_name IS NULL")
  64. return res