~taktik/openobject-addons/hui-extra-6.1

# -*- encoding: utf-8 -*-

##############################################################################

#

#    OpenERP, Open Source Management Solution

#    

#    Copyright (c) 2012 Noviat nv/sa (www.noviat.be). All rights reserved.

# 

#    This program is free software: you can redistribute it and/or modify

#    it under the terms of the GNU Affero 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 Affero General Public License for more details.

#

#    You should have received a copy of the GNU Affero General Public License

#    along with this program.  If not, see <http://www.gnu.org/licenses/>.

#

##############################################################################

 

import re

from osv import fields, osv

from tools.translate import _

import logging

_logger = logging.getLogger(__name__)

 

class be_legal_financial_reportscheme(osv.osv):

    _name = 'be.legal.financial.reportscheme'

    _description = 'Belgian Legal Financial Report Scheme (Full)'

    _rec_name = 'account_group'

    _columns = {

        'account_group': fields.char('Group', size=4, help='General Account Starting Digits'),

        'report_id': fields.many2one('account.financial.report', 'Report Entry', ondelete='cascade'),

        'account_ids': fields.related('report_id','account_ids', type='one2many', relation='account.account', string='Accounts', readonly=True),

    }

    _sql_constraints = [

        ('group_uniq', 'unique (account_group)', 'The General Account Group must be unique !')

    ]    

be_legal_financial_reportscheme()

 

class account_financial_report(osv.osv):

    _inherit = 'account.financial.report'

    _columns = {    

        'code': fields.char('Code', size=16),

        'invisible': fields.boolean('Invisible', help="Hide this entry from the printed report."),

    }

       

    def _get_children_by_order(self, cr, uid, ids, context=None):

        res = []

        if context.get('get_children_by_sequence'):

            res = self.search(cr, uid, [('id', 'child_of', ids[0]), ('invisible', '=', 0)], order='sequence ASC', context=context)

        else:

            for id in ids:

                res.append(id)

                ids2 = self.search(cr, uid, [('parent_id', '=', id), ('invisible', '=', 0)], order='sequence ASC', context=context)

                res += self._get_children_by_order(cr, uid, ids2, context=context)

        return res

       

    def read(self, cr, user, ids, fields=None, context=None, load='_classic_read'):

        #_logger.warn('read, context = %s', context)

        res = super(account_financial_report, self).read(cr, user, ids, fields, context, load)

        context = context or {}

        if 'name' in fields:

            for entry in res:

                if entry.get('code') and context.get('code_print'):

                    entry['name'] = entry['name'] + ' - (' + entry['code'] + ')' 

        return res

    

account_financial_report()

 

class account_account(osv.osv):

    _inherit = 'account.account'

 

    def create(self, cr, uid, vals, context=None):

        acc_id = super(account_account, self).create(cr, uid, vals, context=context) 

        #_logger.warn('create, vals = %s, context = %s', vals, context)

        scheme_obj = self.pool.get('be.legal.financial.reportscheme')

        scheme_table = scheme_obj.read(cr, uid, scheme_obj.search(cr, uid, []), context=context)

        be_report_ids = [x['report_id'][0] for x in scheme_table]

        acc_code = vals['code']

        account = self.browse(cr, uid, acc_id)

        if account.type not in ['view', 'consolidation'] and account.company_id.country_id.code == 'BE':

            be_report_entries =  filter(lambda x: acc_code[0:len(x['account_group'])] == x['account_group'], scheme_table)

            if be_report_entries:

                if len(be_report_entries) > 1:

                    raise osv.except_osv(_('Configuration Error !'), _('Configuration Error in the Belgian Legal Financial Report Scheme.'))

                be_report_id = be_report_entries[0]['report_id'][0]

                self.write(cr, uid, account.id, {'financial_report_ids': [(4, be_report_id)]})

        return acc_id

 

    def write(self, cr, uid, ids, vals, context=None):

        res = super(account_account, self).write(cr, uid, ids, vals, context=context)

        #_logger.warn('write, vals = %s, context = %s', vals, context)

        if 'code' in vals.keys():

            scheme_obj = self.pool.get('be.legal.financial.reportscheme')

            scheme_table = scheme_obj.read(cr, uid, scheme_obj.search(cr, uid, []), context=context)

            be_report_ids = [x['report_id'][0] for x in scheme_table]

            acc_code = vals['code']

            for account in self.browse(cr, uid, ids):

                updated = False

                if account.type not in ['view', 'consolidation'] and account.company_id.country_id.code == 'BE':

                    be_report_entries =  filter(lambda x: acc_code[0:len(x['account_group'])] == x['account_group'], scheme_table)

                    if len(be_report_entries) > 1:

                        raise osv.except_osv(_('Configuration Error !'), _('Configuration Error in the Belgian Legal Financial Report Scheme.'))                       

                    be_report_id = be_report_entries and be_report_entries[0]['report_id'][0]

                    for fin_report in account.financial_report_ids:

                        if fin_report.id in be_report_ids:

                            if fin_report.id == be_report_id:

                                updated = True

                            else:

                                self.write(cr, uid, account.id, {'financial_report_ids': [(3, fin_report.id)]})

                    if be_report_id and not updated:

                        self.write(cr, uid, account.id, {'financial_report_ids': [(4, be_report_id)]})

        return res