You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
8 months ago
|
import collections
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from odoo import models, Command
|
||
|
|
from odoo.tools import populate
|
||
|
|
|
||
|
|
_logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class Partner(models.Model):
|
||
|
|
_inherit = "res.company"
|
||
|
|
|
||
|
|
_populate_sizes = {
|
||
|
|
'small': 5,
|
||
|
|
'medium': 10,
|
||
|
|
'large': 50,
|
||
|
|
}
|
||
|
|
|
||
|
|
def _populate_factories(self):
|
||
|
|
# Activate currency to avoid fail iterator
|
||
|
|
(self.env.ref('base.USD') | self.env.ref('base.EUR')).active = True
|
||
|
|
|
||
|
|
# remaining: paperformat_id, parent_id, partner_id, favicon, font, report_header, external_report_layout_id, report_footer
|
||
|
|
def get_name(values=None, counter=0, **kwargs):
|
||
|
|
return 'company_%s_%s' % (counter, self.env['res.currency'].browse(values['currency_id']).name)
|
||
|
|
|
||
|
|
active_currencies = self.env['res.currency'].search([('active', '=', True)]).ids
|
||
|
|
return [
|
||
|
|
('name', populate.constant('company_{counter}')),
|
||
|
|
('sequence', populate.randint(0, 100)),
|
||
|
|
('company_registry', populate.iterate([False, 'company_registry_{counter}'])),
|
||
|
|
('base_onboarding_company_state', populate.iterate(
|
||
|
|
[False] + [e[0] for e in type(self).base_onboarding_company_state.selection])),
|
||
|
|
('primary_color', populate.iterate([False, '', '#ff7755'])),
|
||
|
|
('secondary_color', populate.iterate([False, '', '#ffff55'], seed='primary_color')),
|
||
|
|
('currency_id', populate.iterate(active_currencies)),
|
||
|
|
('name', populate.compute(get_name)),
|
||
|
|
]
|
||
|
|
|
||
|
|
def _populate(self, size):
|
||
|
|
records = super()._populate(size)
|
||
|
|
self.env.ref('base.user_admin').write({'company_ids': [Command.link(rec.id) for rec in records]}) # add all created companies on user admin
|
||
|
|
return records
|