GOLEM/golem_family/models/golem_family.py

82 lines
2.9 KiB
Python

# -*- coding: utf-8 -*-
# Copyright 2016 Fabien Bourgeois <fabien@yaltik.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero 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 Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from openerp import models, fields, api, _
class ResPartner(models.Model):
_inherit = 'res.partner'
_description = 'GOLEM Partner Family'
family_id = fields.Many2one('golem.family', string='Family', index=True)
family_role = fields.Many2one('golem.member.family.role',
string='Role', index=True)
family_count = fields.Integer('Family Count', related='family_id.count')
@api.multi
def button_family_members(self):
self.ensure_one()
return {'name': _('Family Members'),
'type': 'ir.actions.act_window',
'res_model': 'golem.family',
'view_mode': 'form',
'domain': [('id', '=', self.family_id.id)]}
class GolemFamily(models.Model):
_name = 'golem.family'
_description = 'GOLEM Family Entity'
_inherit = 'mail.thread'
@api.model
def _get_default_nationality_id(self):
return self.env.ref('base.main_company').country_id
name = fields.Char('Name', index=True, required=True)
street = fields.Char('Street')
street2 = fields.Char('Street2')
zip = fields.Char('Zip')
city = fields.Char('City')
state_id = fields.Many2one('res.country.state', 'State',
ondelete='restrict')
country_id = fields.Many2one('res.country', 'Country',
ondelete='restrict',
default=_get_default_nationality_id)
phone = fields.Char('Phone')
mobile = fields.Char('Mobile')
email = fields.Char('Email')
website = fields.Char('Website')
member_ids = fields.One2many('res.partner', 'family_id', 'Members')
note = fields.Text('Note')
count = fields.Integer('Count', compute='_compute_count', store=True)
@api.one
@api.depends('member_ids')
def _compute_count(self):
self.count = len(self.member_ids)
class GolemFamilyRole(models.Model):
_name = 'golem.member.family.role'
_description = 'GOLEM Member Family Role'
_sql_constraints = [('golem_member_family_role_name_uniq',
'UNIQUE (name)',
'Family role must be unique.')]
name = fields.Char('Role')