# -*- coding: utf-8 -*- # Part of Odoo. See LICENSE file for full copyright and licensing details. import base64 import os import re from odoo import api, fields, models, tools, _ from odoo.exceptions import ValidationError, UserError class Company(models.Model): _name = "res.company" _description = 'Companies' _order = 'sequence, name' @api.multi def copy(self, default=None): raise UserError(_('Duplicating a company is not allowed. Please create a new company instead.')) def _get_logo(self): return base64.b64encode(open(os.path.join(tools.config['root_path'], 'addons', 'base', 'static', 'img', 'res_company_logo.png'), 'rb') .read()) @api.model def _get_euro(self): return self.env['res.currency.rate'].search([('rate', '=', 1)], limit=1).currency_id @api.model def _get_user_currency(self): currency_id = self.env['res.users'].browse(self._uid).company_id.currency_id return currency_id or self._get_euro() name = fields.Char(related='partner_id.name', string='Company Name', required=True, store=True, readonly=False) sequence = fields.Integer(help='Used to order Companies in the company switcher', default=10) parent_id = fields.Many2one('res.company', string='Parent Company', index=True) child_ids = fields.One2many('res.company', 'parent_id', string='Child Companies') partner_id = fields.Many2one('res.partner', string='Partner', required=True) report_header = fields.Text(string='Company Tagline', help="Appears by default on the top right corner of your printed documents (report header).") report_footer = fields.Text(string='Report Footer', translate=True, help="Footer text displayed at the bottom of all reports.") logo = fields.Binary(related='partner_id.image', default=_get_logo, string="Company Logo", readonly=False) # logo_web: do not store in attachments, since the image is retrieved in SQL for # performance reasons (see addons/web/controllers/main.py, Binary.company_logo) logo_web = fields.Binary(compute='_compute_logo_web', store=True) currency_id = fields.Many2one('res.currency', string='Currency', required=True, default=lambda self: self._get_user_currency()) user_ids = fields.Many2many('res.users', 'res_company_users_rel', 'cid', 'user_id', string='Accepted Users') account_no = fields.Char(string='Account No.') street = fields.Char(compute='_compute_address', inverse='_inverse_street') street2 = fields.Char(compute='_compute_address', inverse='_inverse_street2') zip = fields.Char(compute='_compute_address', inverse='_inverse_zip') city = fields.Char(compute='_compute_address', inverse='_inverse_city') state_id = fields.Many2one('res.country.state', compute='_compute_address', inverse='_inverse_state', string="Fed. State") bank_ids = fields.One2many('res.partner.bank', 'company_id', string='Bank Accounts', help='Bank accounts related to this company') country_id = fields.Many2one('res.country', compute='_compute_address', inverse='_inverse_country', string="Country") email = fields.Char(related='partner_id.email', store=True, readonly=False) phone = fields.Char(related='partner_id.phone', store=True, readonly=False) website = fields.Char(related='partner_id.website', readonly=False) vat = fields.Char(related='partner_id.vat', string="Tax ID", readonly=False) company_registry = fields.Char() paperformat_id = fields.Many2one('report.paperformat', 'Paper format', default=lambda self: self.env.ref('base.paperformat_euro', raise_if_not_found=False)) external_report_layout_id = fields.Many2one('ir.ui.view', 'Document Template') _sql_constraints = [ ('name_uniq', 'unique (name)', 'The company name must be unique !') ] base_onboarding_company_state = fields.Selection([ ('not_done', "Not done"), ('just_done', "Just done"), ('done', "Done")], string="State of the onboarding company step", default='not_done') @api.model_cr def init(self): for company in self.search([('paperformat_id', '=', False)]): paperformat_euro = self.env.ref('base.paperformat_euro', False) if paperformat_euro: company.write({'paperformat_id': paperformat_euro.id}) sup = super(Company, self) if hasattr(sup, 'init'): sup.init() def _get_company_address_fields(self, partner): return { 'street' : partner.street, 'street2' : partner.street2, 'city' : partner.city, 'zip' : partner.zip, 'state_id' : partner.state_id, 'country_id' : partner.country_id, } # TODO @api.depends(): currently now way to formulate the dependency on the # partner's contact address def _compute_address(self): for company in self.filtered(lambda company: company.partner_id): address_data = company.partner_id.sudo().address_get(adr_pref=['contact']) if address_data['contact']: partner = company.partner_id.browse(address_data['contact']).sudo() company.update(company._get_company_address_fields(partner)) def _inverse_street(self): for company in self: company.partner_id.street = company.street def _inverse_street2(self): for company in self: company.partner_id.street2 = company.street2 def _inverse_zip(self): for company in self: company.partner_id.zip = company.zip def _inverse_city(self): for company in self: company.partner_id.city = company.city def _inverse_state(self): for company in self: company.partner_id.state_id = company.state_id def _inverse_country(self): for company in self: company.partner_id.country_id = company.country_id @api.depends('partner_id', 'partner_id.image') def _compute_logo_web(self): for company in self: company.logo_web = tools.image_resize_image(company.partner_id.image, (180, None)) @api.onchange('state_id') def _onchange_state(self): self.country_id = self.state_id.country_id @api.multi def on_change_country(self, country_id): # This function is called from account/models/chart_template.py, hence decorated with `multi`. self.ensure_one() currency_id = self._get_user_currency() if country_id: currency_id = self.env['res.country'].browse(country_id).currency_id return {'value': {'currency_id': currency_id.id}} @api.onchange('country_id') def _onchange_country_id_wrapper(self): res = {'domain': {'state_id': []}} if self.country_id: res['domain']['state_id'] = [('country_id', '=', self.country_id.id)] values = self.on_change_country(self.country_id.id)['value'] for fname, value in values.items(): setattr(self, fname, value) return res @api.model def _name_search(self, name, args=None, operator='ilike', limit=100, name_get_uid=None): context = dict(self.env.context) newself = self if context.pop('user_preference', None): # We browse as superuser. Otherwise, the user would be able to # select only the currently visible companies (according to rules, # which are probably to allow to see the child companies) even if # she belongs to some other companies. companies = self.env.user.company_id + self.env.user.company_ids args = (args or []) + [('id', 'in', companies.ids)] newself = newself.sudo() return super(Company, newself.with_context(context))._name_search(name=name, args=args, operator=operator, limit=limit, name_get_uid=name_get_uid) @api.model @api.returns('self', lambda value: value.id) def _company_default_get(self, object=False, field=False): """ Returns the default company (usually the user's company). The 'object' and 'field' arguments are ignored but left here for backward compatibility and potential override. """ return self.env['res.users']._get_company() @api.model @tools.ormcache('self.env.uid', 'company') def _get_company_children(self, company=None): if not company: return [] return self.search([('parent_id', 'child_of', [company])]).ids @api.multi def _get_partner_hierarchy(self): self.ensure_one() parent = self.parent_id if parent: return parent._get_partner_hierarchy() else: return self._get_partner_descendance([]) @api.multi def _get_partner_descendance(self, descendance): self.ensure_one() descendance.append(self.partner_id.id) for child_id in self._get_company_children(self.id): if child_id != self.id: descendance = self.browse(child_id)._get_partner_descendance(descendance) return descendance # deprecated, use clear_caches() instead def cache_restart(self): self.clear_caches() @api.model def create(self, vals): if not vals.get('name') or vals.get('partner_id'): self.clear_caches() return super(Company, self).create(vals) partner = self.env['res.partner'].create({ 'name': vals['name'], 'is_company': True, 'image': vals.get('logo'), 'customer': False, 'email': vals.get('email'), 'phone': vals.get('phone'), 'website': vals.get('website'), 'vat': vals.get('vat'), }) vals['partner_id'] = partner.id self.clear_caches() company = super(Company, self).create(vals) # The write is made on the user to set it automatically in the multi company group. self.env.user.write({'company_ids': [(4, company.id)]}) partner.write({'company_id': company.id}) # Make sure that the selected currency is enabled if vals.get('currency_id'): currency = self.env['res.currency'].browse(vals['currency_id']) if not currency.active: currency.write({'active': True}) return company @api.multi def write(self, values): self.clear_caches() # Make sure that the selected currency is enabled if values.get('currency_id'): currency = self.env['res.currency'].browse(values['currency_id']) if not currency.active: currency.write({'active': True}) return super(Company, self).write(values) @api.constrains('parent_id') def _check_parent_id(self): if not self._check_recursion(): raise ValidationError(_('You cannot create recursive companies.')) @api.multi def open_company_edit_report(self): self.ensure_one() return self.env['res.config.settings'].open_company() @api.multi def write_company_and_print_report(self, values): res = self.write(values) report_name = values.get('default_report_name') active_ids = values.get('active_ids') active_model = values.get('active_model') if report_name and active_ids and active_model: docids = self.env[active_model].browse(active_ids) return (self.env['ir.actions.report'].search([('report_name', '=', report_name)], limit=1) .with_context(values) .report_action(docids)) else: return res @api.model def action_open_base_onboarding_company(self): """ Onboarding step for company basic information. """ action = self.env.ref('base.action_open_base_onboarding_company').read()[0] action['res_id'] = self.env.user.company_id.id return action def set_onboarding_step_done(self, step_name): if self[step_name] == 'not_done': self[step_name] = 'just_done' def get_and_update_onbarding_state(self, onboarding_state, steps_states): """ Needed to display onboarding animations only one time. """ old_values = {} all_done = True for step_state in steps_states: old_values[step_state] = self[step_state] if self[step_state] == 'just_done': self[step_state] = 'done' all_done = all_done and self[step_state] == 'done' if all_done: if self[onboarding_state] == 'not_done': # string `onboarding_state` instead of variable name is not an error old_values['onboarding_state'] = 'just_done' else: old_values['onboarding_state'] = 'done' self[onboarding_state] = 'done' return old_values @api.multi def action_save_onboarding_company_step(self): if bool(self.street): self.set_onboarding_step_done('base_onboarding_company_state')