181 lines
6.3 KiB
Python
181 lines
6.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from flectra import api, fields, models
|
|
from flectra import SUPERUSER_ID
|
|
# from flectra.exceptions import ValidationError
|
|
|
|
def migrate_company_branch(cr, registry):
|
|
env = api.Environment(cr, SUPERUSER_ID, {})
|
|
company = env.ref('base.main_company')
|
|
company.write({'branch_id': env.ref('base_branch_company.data_branch_1').id})
|
|
cr.commit()
|
|
user_ids = env['res.users'].search([])
|
|
for user_id in user_ids:
|
|
if not user_id.user_has_groups('base_branch_company.group_multi_branch'):
|
|
user_id.sudo().write({'default_branch_id': user_id.company_id.branch_id.id,
|
|
'branch_ids': [(6, 0, [user_id.company_id.branch_id.id])]})
|
|
cr.commit()
|
|
|
|
|
|
class Company(models.Model):
|
|
_name = "res.company"
|
|
_inherit = ["res.company"]
|
|
|
|
branch_id = fields.Many2one('res.branch', 'Branch', ondelete="cascade")
|
|
|
|
@api.model
|
|
def create(self, vals):
|
|
branch = self.env['res.branch'].create({
|
|
'name': vals['name'],
|
|
'code': vals['name'],
|
|
})
|
|
vals['branch_id'] = branch.id
|
|
self.clear_caches()
|
|
company = super(Company, self).create(vals)
|
|
branch.write({'partner_id': company.partner_id.id,
|
|
'company_id': company.id})
|
|
return company
|
|
|
|
|
|
class ResBranch(models.Model):
|
|
_name = "res.branch"
|
|
|
|
name = fields.Char(string='Name', required=True)
|
|
code = fields.Char(string='Code', required=True)
|
|
active = fields.Boolean(string='Active', default=True)
|
|
partner_id = fields.Many2one('res.partner', string='Partner',
|
|
ondelete='restrict')
|
|
company_id = fields.Many2one(
|
|
'res.company', string="Company",
|
|
default=lambda self: self.env.user.company_id, required=True)
|
|
street = fields.Char()
|
|
street2 = fields.Char()
|
|
zip = fields.Char(change_default=True)
|
|
city = fields.Char()
|
|
state_id = fields.Many2one("res.country.state", string='State',
|
|
ondelete='restrict')
|
|
country_id = fields.Many2one('res.country', string='Country',
|
|
ondelete='restrict')
|
|
email = fields.Char()
|
|
phone = fields.Char()
|
|
mobile = fields.Char()
|
|
|
|
_sql_constraints = [('branch_code_company_uniq',
|
|
'unique (code,company_id)',
|
|
'The branch code must be unique per company!')]
|
|
|
|
@api.model
|
|
def create(self, vals):
|
|
res = super(ResBranch, self).create(vals)
|
|
vals.pop("name", None)
|
|
vals.pop("code", None)
|
|
vals.pop("partner_id", None)
|
|
vals.update({'branch_id': res.id})
|
|
res.partner_id.write(vals)
|
|
return res
|
|
|
|
@api.multi
|
|
def write(self, vals):
|
|
res = super(ResBranch, self).write(vals)
|
|
vals.pop("name", None)
|
|
vals.pop("code", None)
|
|
vals.pop("company_id", None)
|
|
vals.pop("partner_id", None)
|
|
ctx = self.env.context.copy()
|
|
if 'branch' not in ctx:
|
|
for record in self:
|
|
record.partner_id.write(vals)
|
|
return res
|
|
|
|
|
|
class Users(models.Model):
|
|
|
|
_inherit = "res.users"
|
|
|
|
@api.multi
|
|
def read(self, fields=None, load='_classic_read'):
|
|
result = super(Users, self).read(fields, load=load)
|
|
self.with_context({'check_branch': True}).check_missing_branch()
|
|
return result
|
|
|
|
@api.multi
|
|
def check_missing_branch(self):
|
|
for user_id in self:
|
|
if self._context.get('check_branch', False) and user_id.company_id.branch_id and not user_id.default_branch_id:
|
|
user_id.default_branch_id = user_id.company_id.branch_id.id
|
|
user_id.branch_ids = [(4, user_id.company_id.branch_id.id)]
|
|
|
|
@api.model
|
|
def branch_default_get(self, user):
|
|
if not user:
|
|
user = self._uid
|
|
branch_id = self.env['res.users'].browse(user).default_branch_id
|
|
if not branch_id:
|
|
branch_id = \
|
|
self.env['res.users'].browse(user).company_id.branch_id
|
|
return branch_id
|
|
|
|
@api.model
|
|
def _get_branch(self):
|
|
return self.env.user.default_branch_id
|
|
|
|
@api.model
|
|
def _get_default_branch(self):
|
|
return self.branch_default_get(self._uid)
|
|
|
|
def _branches_count(self):
|
|
return self.env['res.branch'].sudo().search_count([])
|
|
|
|
branch_ids = fields.Many2many('res.branch',
|
|
'res_branch_users_rel',
|
|
'user_id',
|
|
'branch_id',
|
|
'Branches', default=_get_branch,
|
|
domain="[('company_id','=',company_id)]")
|
|
default_branch_id = fields.Many2one('res.branch', 'Default branch',
|
|
default=_get_branch,
|
|
domain="[('company_id','=',company_id)"
|
|
"]")
|
|
branches_count = fields.Integer(
|
|
compute='_compute_branches_count',
|
|
string="Number of Companies", default=_branches_count)
|
|
|
|
@api.onchange('company_id')
|
|
def _onchange_company_id(self):
|
|
if self.company_id.branch_id:
|
|
self.default_branch_id = self.company_id.branch_id.id
|
|
self.branch_ids = [(4, self.company_id.branch_id.id)]
|
|
|
|
# To do : Check with all base module test cases
|
|
# @api.multi
|
|
# @api.constrains('default_branch_id', 'branch_ids')
|
|
# def _check_branches(self):
|
|
# for user in self:
|
|
# if user.branch_ids \
|
|
# and user.default_branch_id not in user.branch_ids:
|
|
# raise ValidationError(_('The selected Default Branch (%s) '
|
|
# 'is not in the Branches!') % (
|
|
# user.default_branch_id.name))
|
|
|
|
@api.multi
|
|
def _compute_branches_count(self):
|
|
branches_count = self._branches_count()
|
|
for user in self:
|
|
user.branches_count = branches_count
|
|
|
|
@api.model
|
|
def create(self, vals):
|
|
res = super(Users, self).create(vals)
|
|
if 'company_id' in vals:
|
|
vals.update({
|
|
'default_branch_id': self.company_id.branch_id.id,
|
|
})
|
|
return res
|
|
|
|
@api.multi
|
|
def write(self, vals):
|
|
res = super(Users, self).write(vals)
|
|
if 'company_id' in vals:
|
|
self.default_branch_id = self.company_id.branch_id.id
|
|
return res
|