2
0

Port select template to the new API

This commit is contained in:
Alex Comba 2014-12-04 15:04:41 +01:00 committed by Abraham Anes
parent 9d9499a1b8
commit fa34898601

View File

@ -19,35 +19,32 @@
# #
############################################################################## ##############################################################################
from openerp.osv import fields, orm from openerp import models, fields, api, _
from openerp import exceptions
import time import time
from openerp.tools.translate import _
class WizardSelectMoveTemplate(orm.TransientModel): class WizardSelectMoveTemplate(models.TransientModel):
_name = "wizard.select.move.template" _name = "wizard.select.move.template"
_columns = {
'template_id': fields.many2one( template_id = fields.Many2one(
'account.move.template', comodel_name='account.move.template',
'Move Template', string='Move Template',
required=True required=True
), )
'partner_id': fields.many2one('res.partner', 'Partner'), partner_id = fields.Many2one(
comodel_name='res.partner',
'line_ids': fields.one2many( string='Partner'
'wizard.select.move.template.line', )
'template_id', line_ids = fields.One2many(
'Lines' comodel_name='wizard.select.move.template.line',
), inverse_name='template_id',
string='Lines'
'state': fields.selection( )
[ state = fields.Selection(
('template_selected', 'Template selected'), [('template_selected', 'Template selected')],
], string='State'
'State' )
),
}
def on_change_template_id(self, cr, uid, ids, template_id): def on_change_template_id(self, cr, uid, ids, template_id):
res = {} res = {}
@ -65,16 +62,14 @@ class WizardSelectMoveTemplate(orm.TransientModel):
}) })
return res return res
def load_lines(self, cr, uid, ids, context=None): @api.multi
wizard = self.browse(cr, uid, ids, context=context)[0] def load_lines(self):
template_pool = self.pool.get('account.move.template') for wizard in self:
wizard_line_pool = self.pool.get('wizard.select.move.template.line') template = self.env['account.move.template'].browse(
model_data_obj = self.pool.get('ir.model.data') wizard.template_id.id)
template = template_pool.browse(cr, uid, wizard.template_id.id)
for line in template.template_line_ids: for line in template.template_line_ids:
if line.type == 'input': if line.type == 'input':
wizard_line_pool.create(cr, uid, { self.env['wizard.select.move.template.line'].create({
'template_id': wizard.id, 'template_id': wizard.id,
'sequence': line.sequence, 'sequence': line.sequence,
'name': line.name, 'name': line.name,
@ -83,11 +78,11 @@ class WizardSelectMoveTemplate(orm.TransientModel):
'move_line_type': line.move_line_type, 'move_line_type': line.move_line_type,
}) })
if not wizard.line_ids: if not wizard.line_ids:
return self.load_template(cr, uid, ids) return self.load_template()
wizard.write({'state': 'template_selected'}) wizard.write({'state': 'template_selected'})
view_rec = model_data_obj.get_object_reference( view_rec = self.env['ir.model.data'].get_object_reference(
cr, uid, 'account_move_template', 'wizard_select_template') 'account_move_template', 'wizard_select_template')
view_id = view_rec and view_rec[1] or False view_id = view_rec and view_rec[1] or False
return { return {
@ -98,40 +93,38 @@ class WizardSelectMoveTemplate(orm.TransientModel):
'res_id': wizard.id, 'res_id': wizard.id,
'type': 'ir.actions.act_window', 'type': 'ir.actions.act_window',
'target': 'new', 'target': 'new',
'context': context, 'context': self.env.context,
} }
def load_template(self, cr, uid, ids, context=None): @api.multi
template_obj = self.pool.get('account.move.template') def load_template(self):
account_period_obj = self.pool.get('account.period') for wizard in self:
template_model = self.env['account.move.template']
wizard = self.browse(cr, uid, ids, context=context)[0] account_period_model = self.env['account.period']
if not template_obj.check_zero_lines(cr, uid, wizard): if not template_model.check_zero_lines(wizard):
raise orm.except_orm( raise exceptions.Warning(
_('Error !'), _('Error !'),
_('At least one amount has to be non-zero!') _('At least one amount has to be non-zero!')
) )
input_lines = {} input_lines = {}
for template_line in wizard.line_ids: for template_line in wizard.line_ids:
input_lines[template_line.sequence] = template_line.amount input_lines[template_line.sequence] = template_line.amount
period_id = account_period_obj.find(cr, uid, context=context) period_id = account_period_model.find()
if not period_id: if not period_id:
raise orm.except_orm( raise exceptions.Warning(
_('No period found !'), _('No period found !'),
_('Unable to find a valid period !') _('Unable to find a valid period !')
) )
period_id = period_id[0] period_id = period_id[0]
computed_lines = template_obj.compute_lines( computed_lines = template_model.compute_lines(
cr, uid, wizard.template_id.id, input_lines) wizard.template_id.id, input_lines)
moves = {} moves = {}
for line in wizard.template_id.template_line_ids: for line in wizard.template_id.template_line_ids:
if line.journal_id.id not in moves: if line.journal_id.id not in moves:
moves[line.journal_id.id] = self._make_move( moves[line.journal_id.id] = self._make_move(
cr, uid,
wizard.template_id.name, wizard.template_id.name,
period_id, period_id,
line.journal_id.id, line.journal_id.id,
@ -139,7 +132,6 @@ class WizardSelectMoveTemplate(orm.TransientModel):
) )
self._make_move_line( self._make_move_line(
cr, uid,
line, line,
computed_lines, computed_lines,
moves[line.journal_id.id], moves[line.journal_id.id],
@ -149,8 +141,6 @@ class WizardSelectMoveTemplate(orm.TransientModel):
if wizard.template_id.cross_journals: if wizard.template_id.cross_journals:
trans_account_id = wizard.template_id.transitory_acc_id.id trans_account_id = wizard.template_id.transitory_acc_id.id
self._make_transitory_move_line( self._make_transitory_move_line(
cr,
uid,
line, line,
computed_lines, computed_lines,
moves[line.journal_id.id], moves[line.journal_id.id],
@ -169,23 +159,22 @@ class WizardSelectMoveTemplate(orm.TransientModel):
'target': 'current', 'target': 'current',
} }
def _make_move(self, cr, uid, ref, period_id, journal_id, partner_id): @api.model
account_move_obj = self.pool.get('account.move') def _make_move(self, ref, period_id, journal_id, partner_id):
move_id = account_move_obj.create(cr, uid, { return self.env['account.move'].create({
'ref': ref, 'ref': ref,
'period_id': period_id, 'period_id': period_id,
'journal_id': journal_id, 'journal_id': journal_id,
'partner_id': partner_id, 'partner_id': partner_id,
}) })
return move_id
def _make_move_line(self, cr, uid, line, computed_lines, def _make_move_line(self, line, computed_lines,
move_id, period_id, partner_id): move_id, period_id, partner_id):
account_move_line_obj = self.pool.get('account.move.line') account_move_line_model = self.env['account.move.line']
analytic_account_id = False analytic_account_id = False
if line.analytic_account_id: if line.analytic_account_id:
if not line.journal_id.analytic_journal_id: if not line.journal_id.analytic_journal_id:
raise orm.except_orm( raise exceptions.Warning(
_('No Analytic Journal !'), _('No Analytic Journal !'),
_("You have to define an analytic " _("You have to define an analytic "
"journal on the '%s' journal!") "journal on the '%s' journal!")
@ -193,7 +182,7 @@ class WizardSelectMoveTemplate(orm.TransientModel):
) )
analytic_account_id = line.analytic_account_id.id analytic_account_id = line.analytic_account_id.id
val = { vals = {
'name': line.name, 'name': line.name,
'move_id': move_id, 'move_id': move_id,
'journal_id': line.journal_id.id, 'journal_id': line.journal_id.id,
@ -207,27 +196,27 @@ class WizardSelectMoveTemplate(orm.TransientModel):
'partner_id': partner_id, 'partner_id': partner_id,
} }
if line.move_line_type == 'cr': if line.move_line_type == 'cr':
val['credit'] = computed_lines[line.sequence] vals['credit'] = computed_lines[line.sequence]
if line.move_line_type == 'dr': if line.move_line_type == 'dr':
val['debit'] = computed_lines[line.sequence] vals['debit'] = computed_lines[line.sequence]
id_line = account_move_line_obj.create(cr, uid, val) id_line = account_move_line_model.create(vals)
return id_line return id_line
def _make_transitory_move_line(self, cr, uid, line, def _make_transitory_move_line(self, line,
computed_lines, move_id, period_id, computed_lines, move_id, period_id,
trans_account_id, partner_id): trans_account_id, partner_id):
account_move_line_obj = self.pool.get('account.move.line') account_move_line_model = self.env['account.move.line']
analytic_account_id = False analytic_account_id = False
if line.analytic_account_id: if line.analytic_account_id:
if not line.journal_id.analytic_journal_id: if not line.journal_id.analytic_journal_id:
raise orm.except_orm( raise exceptions.Warning(
_('No Analytic Journal !'), _('No Analytic Journal !'),
_("You have to define an analytic journal " _("You have to define an analytic journal "
"on the '%s' journal!") "on the '%s' journal!")
% (line.template_id.journal_id.name,) % (line.template_id.journal_id.name,)
) )
analytic_account_id = line.analytic_account_id.id analytic_account_id = line.analytic_account_id.id
val = { vals = {
'name': 'transitory', 'name': 'transitory',
'move_id': move_id, 'move_id': move_id,
'journal_id': line.journal_id.id, 'journal_id': line.journal_id.id,
@ -238,33 +227,33 @@ class WizardSelectMoveTemplate(orm.TransientModel):
'partner_id': partner_id, 'partner_id': partner_id,
} }
if line.move_line_type != 'cr': if line.move_line_type != 'cr':
val['credit'] = computed_lines[line.sequence] vals['credit'] = computed_lines[line.sequence]
if line.move_line_type != 'dr': if line.move_line_type != 'dr':
val['debit'] = computed_lines[line.sequence] vals['debit'] = computed_lines[line.sequence]
id_line = account_move_line_obj.create(cr, uid, val) id_line = account_move_line_model.create(vals)
return id_line return id_line
class WizardSelectMoveTemplateLine(orm.TransientModel): class WizardSelectMoveTemplateLine(models.TransientModel):
_description = 'Template Lines' _description = 'Template Lines'
_name = "wizard.select.move.template.line" _name = "wizard.select.move.template.line"
_columns = {
'template_id': fields.many2one('wizard.select.move.template', template_id = fields.Many2one(
'Template'), comodel_name='wizard.select.move.template',
'sequence': fields.integer('Number', required=True), string='Template'
'name': fields.char('Name', size=64, required=True, readonly=True), )
'account_id': fields.many2one( sequence = fields.Integer(string='Number', required=True)
'account.account', name = fields.Char(required=True, readonly=True),
'Account', account_id = fields.Many2one(
comodel_name='account.account',
string='Account',
required=True, required=True,
readonly=True readonly=True
), )
'move_line_type': fields.selection( move_line_type = fields.Selection(
[('cr', 'Credit'), [('cr', 'Credit'), ('dr', 'Debit')],
('dr', 'Debit')], string='Move Line Type',
'Move Line Type',
required=True, required=True,
readonly=True readonly=True
), )
'amount': fields.float('Amount', required=True), amount = fields.Float(required=True)
}