Files
odoo_source/addons/account/populate/res_company.py
T
william 05dd0f6460 [IMP] account: populate
Task 2230635
Add _populate implementation for:
* Bank Statements
* Journals
* Journal Entries / Invoices
* Payments
* Reconcile Models

Load a chart of account for the first 3 companies, and load data only
for companies with a CoA installed.
2021-01-22 10:14:48 +00:00

56 lines
2.5 KiB
Python

# -*- coding: utf-8 -*-
"""Classes extending the populate factory for Companies and related models.
Only adding specificities of basic accounting applications.
"""
from odoo import models, _
from odoo.tools import populate
from odoo.exceptions import UserError
import logging
from functools import lru_cache
_logger = logging.getLogger(__name__)
class ResCompany(models.Model):
"""Populate factory part for the accountings applications of res.company."""
_inherit = "res.company"
def _populate(self, size):
@lru_cache()
def search_coa_ids(currency_id):
return self.env['account.chart.template'].search([('currency_id', '=', currency_id)])
records = super()._populate(size)
_logger.info('Loading Chart Template')
default_chart_templates = self.env['account.chart.template'].search([], limit=1)
if not default_chart_templates:
# TODO install l10n_generic_coa ?
raise UserError(_(
"At least one localization is needed to be installed in order to populate the "
"database with accounting"
))
random = populate.Random('res.company+chart_template_selector')
# Load the a chart of accounts matching the currency of the company for the 3 first created companies
# We are loading an existing CoA and not populating it because:
# * it reflects best real use cases.
# * it allows checking reports by localization
# * the config is complete with try_loading(), no need to adapt when the model changes
# * it is way easier :-)
# We are loading only for 3 companies because:
# * It takes a few hundreds of a second to create account.move records in batch.
# Because we want to have a lot of entries for at least one company (in order to test
# reports, functions and widgets performances for instance), we can't afford to do it for
# a lot of companies.
# * it would be useless to have entries for all the companies, we can already do everything with
# entries in only a few (but multiple) companies.
# Note that we can still populate some new records on top of the CoA if it makes sense,
# like account.journal for instance.
for company in records[:3]:
chart_templates_cur = search_coa_ids(company.currency_id.id) or default_chart_templates
random.choice(chart_templates_cur).with_company(company.id).try_loading()
return records