account-financial-tools/account_renumber/wizard/wizard_renumber.py

# -*- coding: utf-8 -*-
##############################################################################
#
#    OpenERP - Account renumber wizard
#    Copyright (C) 2009 Pexego Sistemas Informáticos. All Rights Reserved
#    $Id$
#
#    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/>.
#
##############################################################################

"""
Account renumber wizard
"""

from openerp.osv import fields
from openerp.osv import orm
from openerp.tools.translate import _
import logging


class wizard_renumber(orm.TransientModel):
    _name = "wizard.renumber"
    _columns = {
                'journal_ids': fields.many2many('account.journal', 'account_journal_wzd_renumber_rel',
                                        'wizard_id', 'journal_id',
                                        required=True,
                                        help="Journals to renumber",
                                        string="Journals"),
                'period_ids': fields.many2many('account.period', 'account_period_wzd_renumber_rel',
                                        'wizard_id', 'period_id',
                                        required=True,
                                        help='Fiscal periods to renumber',
                                        string="Periods", ondelete='null'),
                'number_next': fields.integer('First Number', required=True,
                                        help="Journal sequences will start counting on this number"),
                'state': fields.selection([
                                           ('init', 'Initial'),
                                           ('renumber', 'Renumbering')
                                           ], readonly=True)
                }

    _defaults = {
        'number_next': 1,
        'state': 'init'
    }

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

    def get_sequence_id_for_fiscalyear_id(self, cr, uid, sequence_id, fiscalyear_id, context=None):
        """
        Based on ir_sequence.get_id from the account module.
        Allows us to get the real sequence for the given fiscal year.
        """
        sequences = self.pool.get('ir.sequence').browse(
                                                        cr,
                                                        uid,
                                                        sequence_id,
                                                        context=context)
        for line in sequences.fiscal_ids:
            if line.fiscalyear_id.id == fiscalyear_id:
                return line.sequence_id.id
        return sequence_id

    ##########################################################################
    # Renumber form/action
    ##########################################################################

    def renumber(self, cr, uid, ids, context):
        """
        Action that renumbers all the posted moves on the given
        journal and periods, and returns their ids.
        """
        logger = logging.getLogger("account_renumber")
        obj = self.browse(cr, uid, ids[0])

        period_ids = [x.id for x in obj.period_ids]
        journal_ids = [x.id for x in obj.journal_ids]
        number_next = obj.number_next or 1

        if not (period_ids and journal_ids):
            raise orm.except_orm(
                                 _('No Data Available'),
                                 _('No records found for your selection!'))

        logger.debug("Searching for account moves to renumber.")
        move_facade = self.pool.get('account.move')
        obj_sequence = self.pool.get('ir.sequence')
        sequences_seen = []
        for period in period_ids:
            move_ids = move_facade.search(
                            cr,
                            uid,
                            [
                            ('journal_id', 'in', journal_ids),
                            ('period_id', '=', period),
                            ('state', '=', 'posted')],
                            limit=0, order='date,id',
                            context=context)
            if len(move_ids) == 0:
                continue

            for move in move_facade.browse(cr, uid, move_ids):

                sequence_id = self.get_sequence_id_for_fiscalyear_id(
                                cr,
                                uid,
                                sequence_id=move.journal_id.sequence_id.id,
                                fiscalyear_id=move.period_id.fiscalyear_id.id)
                if not sequence_id in sequences_seen:
                    self.pool.get('ir.sequence').write(
                                            cr,
                                            uid,
                                            [sequence_id],
                                            {'number_next': number_next})
                    sequences_seen.append(sequence_id)
                #
                # Generate (using our own get_id) and write the new move number
                #
                c = {'fiscalyear_id': move.period_id.fiscalyear_id.id}
                new_name = obj_sequence.next_by_id(
                                            cr,
                                            uid,
                                            move.journal_id.sequence_id.id,
                                            c)
                # Note: We can't just do a
                # "move_facade.write(cr, uid, [move.id], {'name': new_name})"
                # cause it might raise a
                #"You can't do this modification on a confirmed entry"
                # exception.
                cr.execute('UPDATE account_move SET name=%s WHERE id=%s',
                               (new_name, move.id))
                logger.debug("%d account moves renumbered." % len(move_ids))
                logger.debug("Renumbering %d account moves." % len(move_ids))
        sequences_seen = []
        obj.write({'state': 'renumber'})

        view_ref = self.pool.get('ir.model.data').get_object_reference(
                                                cr,
                                                uid,
                                                'account',
                                                'view_move_tree')
        view_id = view_ref and view_ref[1] or False,
        res = {
            'type': 'ir.actions.act_window',
            'name': _("Renumbered account moves"),
            'res_model': 'account.move',
            'domain': "[('journal_id','in',%s), ('period_id','in',%s), ('state','=','posted')]" % (repr(journal_ids), repr(period_ids)),
            'view_type': 'form',
            'view_mode': 'tree',
            'view_id': view_id,
            'context': context,
            'target': 'current',
            }
        return res

wizard_renumber()