~openerp-commiter/openobject-addons/trunk-extra-addons

# -*- encoding: utf-8 -*-
##############################################################################
#
#    Copyright (c) 2010 Zikzakmedia S.L. (http://www.zikzakmedia.com)
#    Copyright (c) 2010 Pexego Sistemas Informáticos S.L. (http://www.pexego.es)
#    @authors: Jordi Esteve (Zikzakmedia), Borja López Soilán (Pexego)
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU 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 General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################
"""
Account Chart Update Wizard
"""
__authors__ = [ "Jordi Esteve <jesteve@zikzakmedia.com>",
                "Borja López Soilán <borjals@pexego.es>" ]

from osv import fields, osv
from tools.translate import _
import netsvc

class WizardLog:
    """
    Small helper class to store the messages and errors on the wizard.
    """
    def __init__(self):
        self.messages = []
        self.errors = []

    def add(self, message, is_error=False):
        """
        Adds a message to the log.
        """
        if is_error:
            netsvc.Logger().notifyChannel('account_chart_update', netsvc.LOG_WARNING, u"Log line: %s" % message)
            self.errors.append(message)
        else:
            netsvc.Logger().notifyChannel('account_chart_update', netsvc.LOG_DEBUG, u"Log line: %s" % message)
        self.messages.append(message)

    def has_errors(self):
        """
        Returns whether errors where logged.
        """
        return self.errors

    def __call__(self):
        return "".join(self.messages)

    def __str__(self):
        return "".join(self.messages)
    
    def get_errors_str(self):
        return "".join(self.errors)


class wizard_update_charts_accounts(osv.osv_memory):
    """
    Updates an existing account chart for a company.
    Wizards ask for:
        * a company
        * an account chart template
        * a number of digits for formatting code of non-view accounts
    Then, the wizard:
        * generates/updates all accounts from the template and assigns them to the right company
        * generates/updates all taxes and tax codes, changing account assignations
        * generates/updates all accounting properties and assigns them correctly
    """
    _name = 'wizard.update.charts.accounts'

    def _get_lang_selection_options(self, cr, uid, context={}):
        """
        Gets the available languages for the selection.
        """
        obj = self.pool.get('res.lang')
        ids = obj.search(cr, uid, [], context=context)
        res = obj.read(cr, uid, ids, ['code', 'name'], context)
        return [(r['code'], r['name']) for r in res] + [('','')]


    _columns = {
        'state': fields.selection([
                ('init', 'Step 1'),
                ('ready', 'Step 2'),
                ('done', 'Wizard Complete')
            ],'Status',readonly=True),
        'company_id': fields.many2one('res.company', 'Company', required=True),
        'chart_template_id': fields.many2one('account.chart.template', 'Chart Template', required=True),
        'code_digits': fields.integer('# of Digits', required=True, help="No. of Digits to use for account code. Make sure it is the same number as existing accounts."),
        'lang': fields.selection(_get_lang_selection_options, 'Language', size=5, help="For records searched by name (taxes, tax codes, fiscal positions), the template name will be matched against the record name on this language."),
        'update_tax_code': fields.boolean('Update tax codes', help="Existing tax codes are updated. Tax codes are searched by name."),
        'update_tax': fields.boolean('Update taxes', help="Existing taxes are updated. Taxes are searched by name."),
        'update_account': fields.boolean('Update accounts', help="Existing accounts are updated. Accounts are searched by code."),
        'update_fiscal_position': fields.boolean('Update fiscal positions', help="Existing fiscal positions are updated. Fiscal positions are searched by name."),
        'update_children_accounts_parent': fields.boolean("Update children accounts parent",
                    help="Update the parent of accounts that seem (based on the code) to be children of the newly created ones. If you had an account 430 with a child 4300000, and a 4300 account is created, the 4300000 parent will be set to 4300."),
        'continue_on_errors': fields.boolean("Continue on errors", help="If set, the wizard will continue to the next step even if there are minor errors (for example the parent account of a new account couldn't be set)."),
        'tax_code_ids': fields.one2many('wizard.update.charts.accounts.tax.code', 'update_chart_wizard_id', 'Tax codes'),
        'tax_ids': fields.one2many('wizard.update.charts.accounts.tax', 'update_chart_wizard_id', 'Taxes'),
        'account_ids': fields.one2many('wizard.update.charts.accounts.account', 'update_chart_wizard_id', 'Accounts'),
        'fiscal_position_ids': fields.one2many('wizard.update.charts.accounts.fiscal.position', 'update_chart_wizard_id', 'Fiscal positions'),
        'new_tax_codes': fields.integer('New tax codes', readonly=True),
        'new_taxes': fields.integer('New taxes', readonly=True),
        'new_accounts': fields.integer('New accounts', readonly=True),
        'new_fps': fields.integer('New fiscal positions', readonly=True),
        'updated_tax_codes': fields.integer('Updated tax codes', readonly=True),
        'updated_taxes': fields.integer('Updated taxes', readonly=True),
        'updated_accounts': fields.integer('Updated accounts', readonly=True),
        'updated_fps': fields.integer('Updated fiscal positions', readonly=True),
        'log': fields.text('Messages and Errors', readonly=True)
    }

    def name_search(self, cr, user, name, args=None, operator='ilike', context=None, limit=80):
        """
        Redefine the search to search by company name.
        """
        if not name:
            name = '%'
        if not args:
            args = []
        if not context:
            context = {}
        args = args[:]
        ids = []
        ids = self.search(cr, user, [('company_id', operator, name)]+ args, limit=limit)
        return self.name_get(cr, user, ids, context=context)

    def name_get(self, cr, uid, ids, context=None):
        """
        Use the company name and template as name.
        """
        if context is None:
            context = {}
        if not len(ids):
            return []
        records = self.browse(cr, uid, ids, context)
        res = []
        for record in records:
            res.append((record.id, record.company_id.name+' - '+record.chart_template_id.name))
        return res

    def _get_chart(self, cr, uid, context=None):
        """
        Returns the default chart template.
        """
        if context is None:
            context = {}
        ids = self.pool.get('account.chart.template').search(cr, uid, [], context=context)
        if ids:
            return ids[0]
        return False

    def _get_code_digits(self, cr, uid, context=None, company_id=None):
        """
        Returns the default code size for the accounts.

        To figure out the number of digits of the accounts it look at the
        code size of the default receivable account of the company
        (or user's company if any company is given).
        """
        if context is None:
            context = {}
        property_obj = self.pool.get('ir.property')
        account_obj = self.pool.get('account.account')
        if not company_id:
            user = self.pool.get('res.users').browse(cr, uid, uid, context)
            company_id = user.company_id.id
        property_ids = property_obj.search(cr, uid, [('name', '=', 'property_account_receivable' ), ('company_id', '=', company_id), ('res_id', '=', False), ('value_reference', '!=', False)])
        if not property_ids:
            # Try to get a generic (no-company) property
            property_ids = property_obj.search(cr, uid, [('name', '=', 'property_account_receivable' ), ('res_id', '=', False), ('value_reference', '!=', False)])
        number_digits = 6
        if property_ids:
            prop = property_obj.browse(cr, uid, property_ids[0], context=context)
            account_id = prop.value_reference.id

            if account_id:
                code = account_obj.read(cr, uid, account_id, ['code'], context)['code']
                number_digits = len(code)
        return number_digits

    _defaults = {
        'state': lambda *a: 'init',
        'company_id': lambda self, cr, uid, context: self.pool.get('res.users').browse(cr, uid, [uid], context)[0].company_id.id,
        'chart_template_id': _get_chart,
        'update_tax_code': lambda *a: True,
        'update_tax': lambda *a: True,
        'update_account': lambda *a: True,
        'update_fiscal_position': lambda *a: True,
        'update_children_accounts_parent': lambda *a: True,
        'continue_on_errors': lambda *a: False,
        'lang': lambda self, cr, uid, context: context and context.get('lang') or None,
    }


    def onchange_company_id(self, cr, uid, ids, company_id, context=None):
        """
        Update the code size when the company changes
        """
        res = { 
            'value': {
                'code_digits': self._get_code_digits(cr, uid, context=context, company_id=company_id),
            }
        }
        return res


    def action_init(self, cr, uid, ids, context=None):
        """
        Initial action that sets the initial state.
        """
        if context is None:
            context = {}
        self.write(cr, uid, ids, { 'state': 'init' }, context)
        return True




    ############################################################################
    # Helper methods
    ############################################################################

    def _map_tax_template(self, cr, uid, wizard, tax_template_mapping, tax_template, context=None):
        """
        Adds a tax template -> tax id to the mapping.
        """
        if tax_template and not tax_template_mapping.get(tax_template.id):
            tax_facade = self.pool.get('account.tax')
            tax_ids = tax_facade.search(cr, uid, [
                        ('name', '=', tax_template.name),
                        ('company_id', '=', wizard.company_id.id)
                    ], context=context)
            if tax_ids:
                tax_template_mapping[tax_template.id] = tax_ids[0]

    def _map_tax_code_template(self, cr, uid, wizard, tax_code_template_mapping, tax_code_template, context=None):
        """
        Adds a tax code template -> tax code id to the mapping.
        """
        if tax_code_template and not tax_code_template_mapping.get(tax_code_template.id):
            tax_code_facade = self.pool.get('account.tax.code')
            root_tax_code_id = wizard.chart_template_id.tax_code_root_id.id
            tax_code_name = (tax_code_template.id == root_tax_code_id) and wizard.company_id.name or tax_code_template.name
            tax_code_ids = tax_code_facade.search(cr, uid, [
                        ('name', '=', tax_code_name),
                        ('company_id', '=', wizard.company_id.id)
                    ])
            if tax_code_ids:
                tax_code_template_mapping[tax_code_template.id] = tax_code_ids[0]

    def _map_account_template(self, cr, uid, wizard, account_template_mapping, account_template, context=None):
        """
        Adds an account template -> account id to the mapping
        """
        if account_template and not account_template_mapping.get(account_template.id):
            account_facade = self.pool.get('account.account')
            code = account_template.code or ''
            if account_template.type != 'view':
                if len(code) > 0 and len(code) <= wizard.code_digits:
                    code = '%s%s' % (code, '0' * (wizard.code_digits - len(code)))
            account_ids = account_facade.search(cr, uid, [
                        ('code', '=', code),
                        ('company_id', '=', wizard.company_id.id)
                    ], context=context)
            if account_ids:
                account_template_mapping[account_template.id] = account_ids[0]

    def _map_fp_template(self, cr, uid, wizard, fp_template_mapping, fp_template, context=None):
        """
        Adds a fiscal position template -> fiscal position id to the mapping.
        """
        if fp_template and not fp_template_mapping.get(fp_template.id):
            fp_facade = self.pool.get('account.fiscal.position')
            fp_ids = fp_facade.search(cr, uid, [
                        ('name', '=', fp_template.name),
                        ('company_id', '=', wizard.company_id.id)
                    ], context=context)
            if fp_ids:
                fp_template_mapping[fp_template.id] = fp_ids[0]


    ############################################################################
    # Find records methods 
    ############################################################################


    def _find_tax_codes(self, cr, uid, wizard, context=None):
        """
        Search for, and load, tax code templates to create/update.
        """
        new_tax_codes = 0
        updated_tax_codes = 0
        tax_code_template_mapping = {}
        
        tax_code_templ_facade = self.pool.get('account.tax.code.template')
        tax_code_facade = self.pool.get('account.tax.code')
        wiz_tax_code_facade = self.pool.get('wizard.update.charts.accounts.tax.code')
        
        # Remove previous tax codes
        wiz_tax_code_facade.unlink(cr, uid, wiz_tax_code_facade.search(cr, uid, []))
        
        #
        # Search for new / updated tax codes
        #
        root_tax_code_id = wizard.chart_template_id.tax_code_root_id.id
        children_tax_code_template = tax_code_templ_facade.search(cr, uid, [('parent_id', 'child_of', [root_tax_code_id])], order='id')
        for tax_code_template in tax_code_templ_facade.browse(cr, uid, children_tax_code_template):
            # Ensure the tax code template is on the map (search for the mapped tax code id).
            self._map_tax_code_template(cr, uid, wizard, tax_code_template_mapping, tax_code_template, context)

            tax_code_id = tax_code_template_mapping.get(tax_code_template.id)
            if not tax_code_id:
                new_tax_codes += 1
                wiz_tax_code_facade.create(cr, uid, {
                        'tax_code_id': tax_code_template.id,
                        'update_chart_wizard_id': wizard.id,
                        'type': 'new',
                    }, context)
            elif wizard.update_tax_code:
                #
                # Check the tax code for changes.
                #
                modified = False
                notes = ""
                tax_code = tax_code_facade.browse(cr, uid, tax_code_id, context=context)

                if tax_code.code != tax_code_template.code:
                    notes += _("The code field is different.\n")
                    modified = True
                if tax_code.info != tax_code_template.info:
                    notes += _("The info field is different.\n")
                    modified = True
                if tax_code.sign != tax_code_template.sign:
                    notes += _("The sign field is different.\n")
                    modified = True

                # TODO: We could check other account fields for changes...

                if modified:
                    #
                    # Tax code to update.
                    #
                    updated_tax_codes += 1
                    wiz_tax_code_facade.create(cr, uid, {
                            'tax_code_id': tax_code_template.id,
                            'update_chart_wizard_id': wizard.id,
                            'type': 'updated',
                            'update_tax_code_id': tax_code_id,
                            'notes': notes,
                        }, context)

        return { 'new': new_tax_codes, 'updated': updated_tax_codes, 'mapping': tax_code_template_mapping }
    
    

    def _find_taxes(self, cr, uid, wizard, context=None):
        """
        Search for, and load, tax templates to create/update.
        """
        new_taxes = 0
        updated_taxes = 0
        tax_template_mapping = {}
        
        tax_facade = self.pool.get('account.tax')
        wiz_tax_facade = self.pool.get('wizard.update.charts.accounts.tax')
        
        delay_wiz_tax=[]
        # Remove previous taxes
        wiz_tax_facade.unlink(cr, uid, wiz_tax_facade.search(cr, uid, []))
    
        #
        # Search for new / updated taxes
        #
        for tax_template in wizard.chart_template_id.tax_template_ids:
            # Ensure the tax template is on the map (search for the mapped tax id).
            self._map_tax_template(cr, uid, wizard, tax_template_mapping, tax_template, context)

            tax_id = tax_template_mapping.get(tax_template.id)
            if not tax_id:
                new_taxes += 1
                vals_wiz = {
                        'tax_id': tax_template.id,
                        'update_chart_wizard_id': wizard.id,
                        'type': 'new',
                    }
                if not tax_template.parent_id:
                    wiz_tax_facade.create(cr, uid, vals_wiz, context)
                else:
                    delay_wiz_tax.append(vals_wiz)
            elif wizard.update_tax:
                #
                # Check the tax for changes.
                #
                modified = False
                notes = ""
                tax = tax_facade.browse(cr, uid, tax_id, context=context)

                if tax.sequence != tax_template.sequence:
                    notes += _("The sequence field is different.\n")
                    modified = True
                if tax.amount != tax_template.amount:
                    notes += _("The amount field is different.\n")
                    modified = True
                if tax.type != tax_template.type:
                    notes += _("The type field is different.\n")
                    modified = True
                if tax.applicable_type != tax_template.applicable_type:
                    notes += _("The applicable type field is different.\n")
                    modified = True
                if tax.domain != tax_template.domain:
                    notes += _("The domain field is different.\n")
                    modified = True
                if tax.child_depend != tax_template.child_depend:
                    notes += _("The child depend field is different.\n")
                    modified = True
                if tax.python_compute != tax_template.python_compute:
                    notes += _("The python compute field is different.\n")
                    modified = True
                #if tax.tax_group != tax_template.tax_group:
                    #notes += _("The tax group field is different.\n")
                    #modified = True
                if tax.base_sign != tax_template.base_sign:
                    notes += _("The base sign field is different.\n")
                    modified = True
                if tax.tax_sign != tax_template.tax_sign:
                    notes += _("The tax sign field is different.\n")
                    modified = True
                if tax.include_base_amount != tax_template.include_base_amount:
                    notes += _("The include base amount field is different.\n")
                    modified = True
                if tax.type_tax_use != tax_template.type_tax_use:
                    notes += _("The type tax use field is different.\n")
                    modified = True
                # TODO: We could check other tax fields for changes...

                if modified:
                    #
                    # Tax code to update.
                    #
                    updated_taxes += 1
                    wiz_tax_facade.create(cr, uid, {
                            'tax_id': tax_template.id,
                            'update_chart_wizard_id': wizard.id,
                            'type': 'updated',
                            'update_tax_id': tax_id,
                            'notes': notes,
                        }, context)
                        
        for delay_vals_wiz in delay_wiz_tax:
            wiz_tax_facade.create(cr, uid, delay_vals_wiz, context)
        
        return { 'new': new_taxes, 'updated': updated_taxes, 'mapping': tax_template_mapping }
    
    
    def _find_accounts(self, cr, uid, wizard, context=None):
        """
        Search for, and load, account templates to create/update.
        """
        new_accounts = 0
        updated_accounts = 0
        account_template_mapping = {}
        
        account_facade = self.pool.get('account.account')
        account_template_facade = self.pool.get('account.account.template')
        wiz_account_facade = self.pool.get('wizard.update.charts.accounts.account')
        
        # Remove previous accounts
        wiz_account_facade.unlink(cr, uid, wiz_account_facade.search(cr, uid, []))
    
        #
        # Search for new / updated accounts
        #
        root_account_id = wizard.chart_template_id.account_root_id.id
        children_acc_template = account_template_facade.search(cr, uid, [('parent_id', 'child_of', [root_account_id])], context=context)
        children_acc_template.sort()
        for account_template in account_template_facade.browse(cr, uid, children_acc_template, context=context):
            # Ensure the account template is on the map (search for the mapped account id).
            self._map_account_template(cr, uid, wizard, account_template_mapping, account_template, context)

            account_id = account_template_mapping.get(account_template.id)
            if not account_id:
                new_accounts += 1
                wiz_account_facade.create(cr, uid, {
                        'account_id': account_template.id,
                        'update_chart_wizard_id': wizard.id,
                        'type': 'new',
                    }, context)
            elif wizard.update_account:
                #
                # Check the account for changes.
                #
                modified = False
                notes = ""
                account = account_facade.browse(cr, uid, account_id, context=context)

                if account.name != account_template.name and account.name != wizard.company_id.name:
                    notes += _("The name is different.\n")
                    modified = True
                if account.type != account_template.type:
                    notes += _("The type is different.\n")
                    modified = True
                if account.user_type != account_template.user_type:
                    notes += _("The user type is different.\n")
                    modified = True
                if account.reconcile != account_template.reconcile:
                    notes += _("The reconcile is different.\n")
                    modified = True

                # TODO: We could check other account fields for changes...

                if modified:
                    #
                    # Account to update.
                    #
                    updated_accounts += 1
                    wiz_account_facade.create(cr, uid, {
                            'account_id': account_template.id,
                            'update_chart_wizard_id': wizard.id,
                            'type': 'updated',
                            'update_account_id': account_id,
                            'notes': notes,
                        }, context)

        return { 'new': new_accounts, 'updated': updated_accounts, 'mapping': account_template_mapping }
    
    
    def _find_fiscal_positions(self, cr, uid, wizard, context=None):
        """
        Search for, and load, fiscal position templates to create/update.
        """
        new_fps = 0
        updated_fps = 0
        fp_template_mapping = {}
        
        fp_template_facade = self.pool.get('account.fiscal.position.template')
        fp_facade = self.pool.get('account.fiscal.position')
        wiz_fp_facade = self.pool.get('wizard.update.charts.accounts.fiscal.position')
        
        # Remove previous fiscal positions
        wiz_fp_facade.unlink(cr, uid, wiz_fp_facade.search(cr, uid, []))
    
        #
        # Search for new / updated fiscal positions
        #
        fp_template_ids = fp_template_facade.search(cr, uid, [('chart_template_id', '=', wizard.chart_template_id.id)], context=context)
        for fp_template in fp_template_facade.browse(cr, uid, fp_template_ids, context=context):
            # Ensure the fiscal position template is on the map (search for the mapped fiscal position id).
            self._map_fp_template(cr, uid, wizard, fp_template_mapping, fp_template, context)

            fp_id = fp_template_mapping.get(fp_template.id)
            if not fp_id:
                #
                # New fiscal position template.
                #
                new_fps += 1
                wiz_fp_facade.create(cr, uid, {
                        'fiscal_position_id': fp_template.id,
                        'update_chart_wizard_id': wizard.id,
                        'type': 'new',
                    }, context)
            elif wizard.update_fiscal_position:
                #
                # Check the fiscal position for changes.
                #
                modified = False
                notes = ""
                fp = fp_facade.browse(cr, uid, fp_id, context=context)

                #
                # Check fiscal position taxes for changes.
                #
                if fp_template.tax_ids and fp.tax_ids:
                    for fp_tax_template in fp_template.tax_ids:
                        found = False
                        for fp_tax in fp.tax_ids:
                            if fp_tax.tax_src_id.name == fp_tax_template.tax_src_id.name:
                                if fp_tax_template.tax_dest_id and fp_tax.tax_dest_id:
                                    if fp_tax.tax_dest_id.name == fp_tax_template.tax_dest_id.name:
                                        found = True
                                        break
                                elif not fp_tax_template.tax_dest_id and not fp_tax.tax_dest_id:
                                    found = True
                                    break
                        if not found:
                            if fp_tax_template.tax_dest_id:
                                notes += _("Tax mapping not found on the fiscal position instance: %s -> %s.\n") % (fp_tax_template.tax_src_id.name, fp_tax_template.tax_dest_id.name)
                            else:
                                notes += _("Tax mapping not found on the fiscal position instance: %s -> None.\n") % fp_tax_template.tax_src_id.name
                            modified = True
                elif fp_template.tax_ids and not fp.tax_ids:
                    notes += _("The template has taxes the fiscal position instance does not.\n")
                    modified = True

                #
                # Check fiscal position accounts for changes.
                #
                if fp_template.account_ids and fp.account_ids:
                    for fp_account_template in fp_template.account_ids:
                        found = False
                        for fp_account in fp.account_ids:
                            if fp_account.account_src_id.name == fp_account_template.account_src_id.name:
                                if fp_account.account_dest_id.name == fp_account_template.account_dest_id.name:
                                    found = True
                                    break
                        if not found:
                            notes += _("Account mapping not found on the fiscal position instance: %s -> %s.\n") % (fp_account_template.account_src_id.name, fp_account_template.account_dest_id.name)
                            modified = True
                elif fp_template.account_ids and not fp.account_ids:
                    notes += _("The template has accounts the fiscal position instance does not.\n")
                    modified = True


                if modified:
                    #
                    # Fiscal position template to update.
                    #
                    updated_fps += 1
                    wiz_fp_facade.create(cr, uid, {
                            'fiscal_position_id': fp_template.id,
                            'update_chart_wizard_id': wizard.id,
                            'type': 'updated',
                            'update_fiscal_position_id': fp_id,
                            'notes': notes,
                        }, context)

        return { 'new': new_fps, 'updated': updated_fps, 'mapping': fp_template_mapping }
    
    
    def action_find_records(self, cr, uid, ids, context=None):
        """
        Searchs for records to update/create and shows them
        """
        if context is None:
            context = {}
        wizard = self.browse(cr, uid, ids[0], context=context)

        if wizard.lang:
            context['lang'] = wizard.lang
        elif context.get('lang'):
            del context['lang']

        #
        # Search for, and load, the records to create/update.
        #
        tax_codes_res = self._find_tax_codes(cr, uid, wizard, context=context)
        taxes_res = self._find_taxes(cr, uid, wizard, context=context)
        accounts_res = self._find_accounts(cr, uid, wizard, context=context)
        fps_res = self._find_fiscal_positions(cr, uid, wizard, context=context)

        #
        # Write the results, and go to the next step.
        #
        self.write(cr, uid, [wizard.id], {
            'state': 'ready',
            'new_tax_codes': tax_codes_res.get('new', 0),
            'new_taxes': taxes_res.get('new', 0),
            'new_accounts': accounts_res.get('new', 0),
            'new_fps': fps_res.get('new', 0),
            'updated_tax_codes': tax_codes_res.get('updated', 0),
            'updated_taxes': taxes_res.get('updated', 0),
            'updated_accounts': accounts_res.get('updated', 0),
            'updated_fps': fps_res.get('updated', 0),
        }, context)
        
        return True



    
    ############################################################################
    # Update records methods 
    ############################################################################

    def _update_tax_codes(self, cr, uid, wizard, log, context=None):
        """
        Search for, and load, tax code templates to create/update.
        """
        tax_code_facade = self.pool.get('account.tax.code')

        root_tax_code_id = wizard.chart_template_id.tax_code_root_id.id

        new_tax_codes = 0
        updated_tax_codes = 0
        tax_code_template_mapping = {}

        for wiz_tax_code in wizard.tax_code_ids:
            tax_code_template = wiz_tax_code.tax_code_id
            tax_code_name = (root_tax_code_id == tax_code_template.id) and wizard.company_id.name or tax_code_template.name

            # Ensure the parent tax code template is on the map.
            self._map_tax_code_template(cr, uid, wizard, tax_code_template_mapping, tax_code_template.parent_id, context)

            #
            # Values
            #
            vals = {
                'name': tax_code_name,
                'code': tax_code_template.code,
                'info': tax_code_template.info,
                'parent_id': tax_code_template.parent_id and tax_code_template_mapping.get(tax_code_template.parent_id.id),
                'company_id': wizard.company_id.id,
                'sign': tax_code_template.sign,
            }

            tax_code_id = None
            modified = False

            if wiz_tax_code.type == 'new':
                #
                # Create the tax code
                #
                tax_code_id = tax_code_facade.create(cr, uid, vals)
                log.add(_("Created tax code %s.\n") % tax_code_name)
                new_tax_codes += 1
                modified = True
            elif wizard.update_tax_code and wiz_tax_code.update_tax_code_id:
                #
                # Update the tax code
                #
                tax_code_id = wiz_tax_code.update_tax_code_id.id
                tax_code_facade.write(cr, uid, [tax_code_id], vals)
                log.add(_("Updated tax code %s.\n") % tax_code_name)
                updated_tax_codes += 1
                modified = True
            else:
                tax_code_id = wiz_tax_code.update_tax_code_id and wiz_tax_code.update_tax_code_id.id
                modified = False

            # Store the tax codes on the map
            tax_code_template_mapping[tax_code_template.id] = tax_code_id

            if modified:
                #
                # Detect errors
                #
                if tax_code_template.parent_id and not tax_code_template_mapping.get(tax_code_template.parent_id.id):
                    log.add(_("Tax code %s: The parent tax code %s can not be set.\n") % (tax_code_name, tax_code_template.parent_id.name), True)

        return {
            'new': new_tax_codes,
            'updated': updated_tax_codes,
            'mapping': tax_code_template_mapping
        }



    def _update_taxes(self, cr, uid, wizard, log, tax_code_template_mapping, context=None):
        """
        Search for, and load, tax templates to create/update.
        """
        tax_facade = self.pool.get('account.tax')

        new_taxes = 0
        updated_taxes = 0
        tax_template_mapping = {}
        taxes_pending_for_accounts = {}

        for wiz_tax in wizard.tax_ids:
            tax_template = wiz_tax.tax_id

            # Ensure the parent tax template is on the map.
            self._map_tax_template(cr, uid, wizard, tax_template_mapping, tax_template.parent_id, context)

            #
            # Ensure the referenced tax codes are on the map.
            #
            tax_code_templates_to_find = [
                tax_template.base_code_id,
                tax_template.tax_code_id,
                tax_template.ref_base_code_id,
                tax_template.ref_tax_code_id
            ]
            for tax_code_template in [tmpl for tmpl in tax_code_templates_to_find if tmpl]:
                self._map_tax_code_template(cr, uid, wizard, tax_code_template_mapping, tax_code_template)

            #
            # Values
            #
            vals_tax = {
                'name': tax_template.name,
                'sequence': tax_template.sequence,
                'amount': tax_template.amount,
                'type': tax_template.type,
                'applicable_type': tax_template.applicable_type,
                'domain': tax_template.domain,
                'parent_id': tax_template.parent_id and tax_template_mapping.get(tax_template.parent_id.id),
                'child_depend': tax_template.child_depend,
                'python_compute': tax_template.python_compute,
                'python_compute_inv': tax_template.python_compute_inv,
                'python_applicable': tax_template.python_applicable,
                #'tax_group': tax_template.tax_group,
                'base_code_id': tax_template.base_code_id and tax_code_template_mapping.get(tax_template.base_code_id.id),
                'tax_code_id': tax_template.tax_code_id and tax_code_template_mapping.get(tax_template.tax_code_id.id),
                'base_sign': tax_template.base_sign,
                'tax_sign': tax_template.tax_sign,
                'ref_base_code_id': tax_template.ref_base_code_id and tax_code_template_mapping.get(tax_template.ref_base_code_id.id),
                'ref_tax_code_id': tax_template.ref_tax_code_id and tax_code_template_mapping.get(tax_template.ref_tax_code_id.id),
                'ref_base_sign': tax_template.ref_base_sign,
                'ref_tax_sign': tax_template.ref_tax_sign,
                'include_base_amount': tax_template.include_base_amount,
                'description': tax_template.description,
                'company_id': wizard.company_id.id,
                'type_tax_use': tax_template.type_tax_use
            }

            tax_id = None
            modified = False

            if wiz_tax.type == 'new':
                #
                # Create a new tax.
                #
                tax_id = tax_facade.create(cr, uid, vals_tax)
                log.add(_("Created tax %s.\n") % tax_template.name)
                new_taxes += 1
                modified = True
            elif wizard.update_tax and wiz_tax.update_tax_id:
                #
                # Update a tax.
                #
                tax_id = wiz_tax.update_tax_id.id
                tax_facade.write(cr, uid, [tax_id], vals_tax)
                log.add(_("Updated tax %s.\n") % tax_template.name)
                updated_taxes += 1
                modified = True
            else:
                tax_id = wiz_tax.update_tax_id and wiz_tax.update_tax_id.id

            # Update the tax template map
            tax_template_mapping[tax_template.id] = tax_id

            if modified:
                #
                # Add to the dict of taxes waiting for accounts.
                #
                taxes_pending_for_accounts[tax_id] = {
                    'account_collected_id': tax_template.account_collected_id and tax_template.account_collected_id.id or False,
                    'account_paid_id': tax_template.account_paid_id and tax_template.account_paid_id.id or False,
                }

                #
                # Detect errors
                #
                if tax_template.parent_id and not tax_template_mapping.get(tax_template.parent_id.id):
                    log.add(_("Tax %s: The parent tax %s can not be set.\n") % (tax_template.name, tax_template.parent_id.name), True)
                if tax_template.base_code_id and not tax_code_template_mapping.get(tax_template.base_code_id.id):
                    log.add(_("Tax %s: The tax code for the base %s can not be set.\n") % (tax_template.name, tax_template.base_code_id.name), True)
                if tax_template.tax_code_id and not tax_code_template_mapping.get(tax_template.tax_code_id.id):
                    log.add(_("Tax %s: The tax code for the tax %s can not be set.\n") % (tax_template.name, tax_template.tax_code_id.name), True)
                if tax_template.ref_base_code_id and not tax_code_template_mapping.get(tax_template.ref_base_code_id.id):
                    log.add(_("Tax %s: The tax code for the base refund %s can not be set.\n") % (tax_template.name, tax_template.ref_base_code_id.name), True)
                if tax_template.ref_tax_code_id and not tax_code_template_mapping.get(tax_template.ref_tax_code_id.id):
                    log.add(_("Tax %s: The tax code for the tax refund %s can not be set.\n") % (tax_template.name, tax_template.ref_tax_code_id.name), True)

        return {
            'new': new_taxes,
            'updated': updated_taxes,
            'mapping': tax_template_mapping,
            'pending': taxes_pending_for_accounts
        }



    def _update_children_accounts_parent(self, cr, uid, wizard, log, parent_account_id, context=None):
        """
        Updates the parent_id of accounts that seem to be children of the
        given account (accounts that start with the same code and are brothers
        of the first account).
        """
        account_facade = self.pool.get('account.account')
        parent_account = account_facade.browse(cr, uid, parent_account_id, context=context)

        if not parent_account.parent_id or not parent_account.code:
            return False

        children_ids = account_facade.search(cr, uid, [
                    ('company_id', '=', parent_account.company_id and parent_account.company_id.id),
                    ('parent_id', '=', parent_account.parent_id.id),
                    ('code', '=like', "%s%%" % parent_account.code),
                    ('id', '!=', parent_account.id),
                ], context=context)

        if children_ids:
            try:
                account_facade.write(cr, uid, children_ids, { 'parent_id': parent_account.id }, context=context)
            except osv.except_osv, ex:
                log.add(_("Exception setting the parent of account %s children: %s - %s.\n") % (parent_account.code, ex.name, ex.value), True)

        return True
    
    
    def _update_accounts(self, cr, uid, wizard, log, tax_template_mapping, context=None):
        """
        Search for, and load, account templates to create/update.
        """
        account_facade = self.pool.get('account.account')

        root_account_id = wizard.chart_template_id.account_root_id.id

        # Disable the parent_store computing on account_account during the batch
        # processing, we will force _parent_store_compute afterwards.
        self.pool._init = True

        new_accounts = 0
        updated_accounts = 0
        account_template_mapping = {}

        for wiz_account in wizard.account_ids:
            account_template = wiz_account.account_id


            # Ensure the parent account template is on the map.
            self._map_account_template(cr, uid, wizard, account_template_mapping, account_template.parent_id, context)

            #
            # Ensure the related tax templates are on the map.
            #
            for tax_template in account_template.tax_ids:
                self._map_tax_template(cr, uid, wizard, tax_template_mapping, tax_template, context)

            # Get the tax ids
            tax_ids = [tax_template_mapping[tax_template.id] for tax_template in account_template.tax_ids if tax_template_mapping[tax_template.id]]

            #
            # Calculate the account code (we need to add zeros to non-view
            # account codes)
            #
            code = account_template.code or ''
            if account_template.type != 'view':
                if len(code) > 0 and len(code) <= wizard.code_digits:
                    code = '%s%s' % (code, '0' * (wizard.code_digits - len(code)))

            #
            # Values
            #
            vals = {
                'name': (root_account_id == account_template.id) and wizard.company_id.name or account_template.name,
                #'sign': account_template.sign,
                'currency_id': account_template.currency_id and account_template.currency_id.id or False,
                'code': code,
                'type': account_template.type,
                'user_type': account_template.user_type and account_template.user_type.id or False,
                'reconcile': account_template.reconcile,
                'shortcut': account_template.shortcut,
                'note': account_template.note,
                'parent_id': account_template.parent_id and account_template_mapping.get(account_template.parent_id.id) or False,
                'tax_ids': [(6, 0, tax_ids)],
                'company_id': wizard.company_id.id,
            }

            account_id = None
            modified = False

            if wiz_account.type == 'new':
                #
                # Create the account
                #
                try:
                    account_id = account_facade.create(cr, uid, vals)
                    log.add(_("Created account %s.\n") % code)
                    new_accounts += 1
                    modified = True
                except osv.except_osv, ex:
                    log.add(_("Exception creating account %s: %s - %s.\n") % (code, ex.name, ex.value), True)
            elif wizard.update_account and wiz_account.update_account_id:
                #
                # Update the account
                #
                account_id = wiz_account.update_account_id.id
                try:
                    account_facade.write(cr, uid, [account_id], vals)
                    log.add(_("Updated account %s.\n") % code)
                    updated_accounts += 1
                    modified = True
                except osv.except_osv, ex:
                    log.add(_("Exception writing account %s: %s - %s.\n") % (code, ex.name, ex.value), True)
            else:
                account_id = wiz_account.update_account_id and wiz_account.update_account_id.id

            # Store the account on the map
            account_template_mapping[account_template.id] = account_id

            if modified:
                #
                # Detect errors
                #
                if account_template.parent_id and not account_template_mapping.get(account_template.parent_id.id):
                    log.add(_("Account %s: The parent account %s can not be set.\n") % (code, account_template.parent_id.code), True)

                #
                # Set this account as the parent of the accounts that seem to
                # be its children (brothers starting with the same code).
                #
                if wizard.update_children_accounts_parent:
                    self._update_children_accounts_parent(cr, uid, wizard, log, account_id, context=context)

        #
        # Reenable the parent_store computing on account_account
        # and force the recomputation.
        #
        self.pool._init = False
        self.pool.get('account.account')._parent_store_compute(cr)

        return {
            'new': new_accounts,
            'updated': updated_accounts,
            'mapping': account_template_mapping
        }


    def _update_taxes_pending_for_accounts(self, cr, uid, wizard, log, taxes_pending_for_accounts, account_template_mapping, context=None):
        """
        Updates the taxes (created or updated on previous steps) to set
        the references to the accounts (the taxes where created/updated first,
        when the referenced accounts where still not available).
        """
        tax_facade = self.pool.get('account.tax')
        account_template_facade = self.pool.get('account.account.template')

        for key, value in taxes_pending_for_accounts.items():
            #
            # Ensure the related account templates are on the map.
            #
            if value['account_collected_id']:
                account_template = account_template_facade.browse(cr, uid, value['account_collected_id'], context=context)
                self._map_account_template(cr, uid, wizard, account_template_mapping, account_template, context)
            if value['account_paid_id']:
                account_template = account_template_facade.browse(cr, uid, value['account_paid_id'], context=context)
                self._map_account_template(cr, uid, wizard, account_template_mapping, account_template, context)


            if value['account_collected_id'] or value['account_paid_id']:
                if account_template_mapping.get(value['account_collected_id']) and account_template_mapping.get(value['account_paid_id']):
                    vals = {
                        'account_collected_id': account_template_mapping[value['account_collected_id']],
                        'account_paid_id': account_template_mapping[value['account_paid_id']],
                    }
                    tax_facade.write(cr, uid, [key], vals)
                else:
                    tax = tax_facade.browse(cr, uid, key)
                    if not account_template_mapping.get(value['account_collected_id']):
                        log.add(_("Tax %s: The collected account can not be set.\n") % (tax.name), True)
                    if not account_template_mapping.get(value['account_paid_id']):
                        log.add(_("Tax %s: The paid account can not be set.\n") % (tax.name), True)


    def _update_fiscal_positions(self, cr, uid, wizard, log, tax_template_mapping, account_template_mapping, context=None):
        """
        Search for, and load, fiscal position templates to create/update.
        """        
        fp_facade = self.pool.get('account.fiscal.position')
        fp_tax_facade = self.pool.get('account.fiscal.position.tax')
        fp_account_facade = self.pool.get('account.fiscal.position.account')

        new_fps = 0
        updated_fps = 0

        for wiz_fp in wizard.fiscal_position_ids:
            fp_template = wiz_fp.fiscal_position_id

            fp_id = None
            modified = False
            if wiz_fp.type == 'new':
                #
                # Create a new fiscal position
                #
                vals_fp = {
                   'company_id': wizard.company_id.id,
                   'name': fp_template.name,
                }
                fp_id = fp_facade.create(cr, uid, vals_fp)
                new_fps += 1
                modified = True
            elif wizard.update_fiscal_position and wiz_fp.update_fiscal_position_id:
                #
                # Update the given fiscal position (remove the tax and account
                # mappings, that will be regenerated later)
                #
                fp_id = wiz_fp.update_fiscal_position_id.id
                updated_fps += 1
                modified = True
                # Remove the tax mappings
                fp_tax_ids = fp_tax_facade.search(cr, uid, [('position_id', '=', fp_id)])
                fp_tax_facade.unlink(cr, uid, fp_tax_ids)
                # Remove the account mappings
                fp_account_ids = fp_account_facade.search(cr, uid, [('position_id', '=', fp_id)])
                fp_account_facade.unlink(cr, uid, fp_account_ids)
            else:
                fp_id = wiz_fp.update_fiscal_position_id and wiz_fp.update_fiscal_position_id.id

            if modified:
                #
                # (Re)create the tax mappings
                #
                for fp_tax in fp_template.tax_ids:
                    #
                    # Ensure the related tax templates are on the map.
                    #
                    self._map_tax_template(cr, uid, wizard, tax_template_mapping, fp_tax.tax_src_id, context)
                    if fp_tax.tax_dest_id:
                        self._map_tax_template(cr, uid, wizard, tax_template_mapping, fp_tax.tax_dest_id, context)

                    #
                    # Create the fp tax mapping
                    #
                    vals_tax = {
                        'tax_src_id': tax_template_mapping.get(fp_tax.tax_src_id.id),
                        'tax_dest_id': fp_tax.tax_dest_id and tax_template_mapping.get(fp_tax.tax_dest_id.id),
                        'position_id': fp_id,
                    }
                    fp_tax_facade.create(cr, uid, vals_tax)

                    #
                    # Check for errors
                    #
                    if not tax_template_mapping.get(fp_tax.tax_src_id.id):
                        log.add(_("Fiscal position %s: The source tax %s can not be set.\n") % (fp_template.name, fp_tax.tax_src_id.code), True)
                    if fp_tax.tax_dest_id and not tax_template_mapping.get(fp_tax.tax_dest_id.id):
                        log.add(_("Fiscal position %s: The destination tax %s can not be set.\n") % (fp_template.name, fp_tax.tax_dest_id.name), True)
                #
                # (Re)create the account mappings
                #
                for fp_account in fp_template.account_ids:
                    #
                    # Ensure the related account templates are on the map.
                    #
                    self._map_account_template(cr, uid, wizard, account_template_mapping, fp_account.account_src_id, context)
                    if fp_account.account_dest_id:
                        self._map_account_template(cr, uid, wizard, account_template_mapping, fp_account.account_dest_id, context)

                    #
                    # Create the fp account mapping
                    #
                    vals_account = {
                        'account_src_id': account_template_mapping.get(fp_account.account_src_id.id),
                        'account_dest_id': fp_account.account_dest_id and account_template_mapping.get(fp_account.account_dest_id.id),
                        'position_id': fp_id,
                    }
                    fp_account_facade.create(cr, uid, vals_account)

                    #
                    # Check for errors
                    #
                    if not account_template_mapping.get(fp_account.account_src_id.id):
                        log.add(_("Fiscal position %s: The source account %s can not be set.\n") % (fp_template.name, fp_account.account_src_id.code), True)
                    if fp_account.account_dest_id and not account_template_mapping.get(fp_account.account_dest_id.id):
                        log.add(_("Fiscal position %s: The destination account %s can not be set.\n") % (fp_template.name, fp_account.account_dest_id.code), True)

            log.add(_("Created or updated fiscal position %s.\n") % fp_template.name)
        return { 'new': new_fps, 'updated': updated_fps }


    
    
    def action_update_records(self, cr, uid, ids, context=None):
        """
        Action that creates/updates the selected elements.
        """
        if context is None:
            context = {}
        wizard = self.browse(cr, uid, ids[0], context=context)

        if wizard.lang:
            context['lang'] = wizard.lang
        elif context.get('lang'):
            del context['lang']
            
        log = WizardLog()

        #
        # Create or update the records.
        #
        tax_codes_res = self._update_tax_codes(cr, uid, wizard, log, context=context)
        taxes_res = self._update_taxes(cr, uid, wizard, log, tax_codes_res['mapping'], context=context)
        accounts_res = self._update_accounts(cr, uid, wizard, log, taxes_res['pending'], context=context)
        self._update_taxes_pending_for_accounts(cr, uid, wizard, log, taxes_res['pending'], accounts_res['mapping'], context=context)
        fps_res = self._update_fiscal_positions(cr, uid, wizard, log, taxes_res['mapping'], accounts_res['mapping'], context=context)

        #
        # Check if errors where detected and wether we should stop.
        #
        if log.has_errors() and not wizard.continue_on_errors:
            raise osv.except_osv(_('Error'), _("One or more errors detected!\n\n%s") % log.get_errors_str())

        #
        # Store the data and go to the next step.
        #
        self.write(cr, uid, [wizard.id], {
                    'state': 'done',
                    'new_tax_codes': tax_codes_res.get('new', 0),
                    'new_taxes': taxes_res.get('new', 0),
                    'new_accounts': accounts_res .get('new', 0),
                    'new_fps': fps_res.get('new', 0),
                    'updated_tax_codes': tax_codes_res.get('updated', 0),
                    'updated_taxes': taxes_res.get('updated', 0),
                    'updated_accounts': accounts_res.get('updated', 0),
                    'updated_fps': fps_res.get('updated', 0),
                    'log': log(),
                }, context)
        
        return True

wizard_update_charts_accounts()


class wizard_update_charts_accounts_tax_code(osv.osv_memory):
    """
    Tax code that needs to be updated (new or updated in the template).
    """
    _name = 'wizard.update.charts.accounts.tax.code'

    _columns = {
        'tax_code_id': fields.many2one('account.tax.code.template', 'Tax code template', required=True),
        'update_chart_wizard_id': fields.many2one('wizard.update.charts.accounts', 'Update chart wizard', required=True),
        'type': fields.selection([
                    ('new', 'New template'),
                    ('updated', 'Updated template'),
                ], 'Type'),
        'update_tax_code_id': fields.many2one('account.tax.code', 'Tax code to update', required=False),
        'notes': fields.text('Notes', readonly=True),
    }

    _defaults = {
        #'update_tax_code_id': lambda *a: None,
    }

wizard_update_charts_accounts_tax_code()


class wizard_update_charts_accounts_tax(osv.osv_memory):
    """
    Tax that needs to be updated (new or updated in the template).
    """
    _name = 'wizard.update.charts.accounts.tax'

    _columns = {
        'tax_id': fields.many2one('account.tax.template', 'Tax template', required=True),
        'update_chart_wizard_id': fields.many2one('wizard.update.charts.accounts', 'Update chart wizard', required=True),
        'type': fields.selection([
                    ('new', 'New template'),
                    ('updated', 'Updated template'),
                ], 'Type'),
        'update_tax_id': fields.many2one('account.tax', 'Tax to update', required=False),
        'notes': fields.text('Notes', readonly=True),
    }

    _defaults = {
        #'update_tax_id': lambda *a: None,
    }

wizard_update_charts_accounts_tax()


class wizard_update_charts_accounts_account(osv.osv_memory):
    """
    Account that needs to be updated (new or updated in the template).
    """
    _name = 'wizard.update.charts.accounts.account'
    
    # The chart of accounts can have a lot of accounts, so we need an higher
    # limit for the objects in memory to let the wizard create all the items
    # at once.
    _max_count = 4096

    _columns = {
        'account_id': fields.many2one('account.account.template', 'Account template', required=True),
        'update_chart_wizard_id': fields.many2one('wizard.update.charts.accounts', 'Update chart wizard', required=True),
        'type': fields.selection([
                    ('new', 'New template'),
                    ('updated', 'Updated template'),
                ], 'Type'),
        'update_account_id': fields.many2one('account.account', 'Account to update', required=False),
        'notes': fields.text('Notes', readonly=True),
    }

    _defaults = {
        #'update_account_id': lambda *a: None,
    }

wizard_update_charts_accounts_account()


class wizard_update_charts_accounts_fiscal_position(osv.osv_memory):
    """
    Fiscal position that needs to be updated (new or updated in the template).
    """
    _name = 'wizard.update.charts.accounts.fiscal.position'

    _columns = {
        'fiscal_position_id': fields.many2one('account.fiscal.position.template', 'Fiscal position template', required=True),
        'update_chart_wizard_id': fields.many2one('wizard.update.charts.accounts', 'Update chart wizard', required=True),
        'type': fields.selection([
                    ('new', 'New template'),
                    ('updated', 'Updated template'),
                ], 'Type'),
        'update_fiscal_position_id': fields.many2one('account.fiscal.position', 'Fiscal position to update', required=False),
        'notes': fields.text('Notes', readonly=True),
    }

    _defaults = {
        #'update_fiscal_position_id': lambda *a: None,
    }


wizard_update_charts_accounts_fiscal_position()

# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: