Yaltik/account-financial-tools
2
0
Fork 0
Code Pull Requests Activity

Port document template to the new API

Browse Source
This commit is contained in:
Alex Comba 2014-12-04 15:01:38 +01:00 committed by Abraham Anes
parent bc32c437da
commit 9d9499a1b8
1 changed files with 36 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
account_move_template/account_document_template.py
View File

@ -20,52 +20,48 @@
# #
############################################################################## ##############################################################################
from openerp.osv import fields, orm from openerp import models, fields, api
from openerp.tools.translate import _ from openerp.tools.translate import _
from openerp import exceptions
import re import re
class AccountDocumentTemplate(orm.Model): class AccountDocumentTemplate(models.Model):
_computed_lines = {}
_current_template_id = 0
_cr = None
_uid = None
_name = 'account.document.template' _name = 'account.document.template'
_columns = { name = fields.Char(required=True)
'name': fields.char('Name', size=64, required=True),
}
def _input_lines(self, cr, uid, template): @api.model
def _input_lines(self, template):
count = 0 count = 0
for line in template.template_line_ids: for line in template.template_line_ids:
if line.type == 'input': if line.type == 'input':
count += 1 count += 1
return count return count
def _get_template_line(self, cr, uid, template_id, line_number): @api.model
for line in self.browse(cr, uid, template_id).template_line_ids: def _get_template_line(self, template_id, line_number):
template = self.browse(template_id)
for line in template.template_line_ids:
if line.sequence == line_number: if line.sequence == line_number:
return line return line
return False return False
def _generate_empty_lines(self, cr, uid, template_id): @api.model
def _generate_empty_lines(self, template_id):
lines = {} lines = {}
t_lines = self.browse(cr, uid, template_id).template_line_ids template = self.browse(template_id)
for template_line in t_lines: for line in template.template_line_ids:
lines[template_line.sequence] = None lines[line.sequence] = None
return lines return lines
@api.model
def lines(self, line_number): def lines(self, line_number):
if self._computed_lines[line_number] is not None: if self._computed_lines[line_number] is not None:
return self._computed_lines[line_number] return self._computed_lines[line_number]
line = self._get_template_line(self._cr, line = self._get_template_line(self._current_template_id, line_number)
self._uid,
self._current_template_id,
line_number)
if re.match(r'L\( *' + str(line_number) + r' *\)', line.python_code): if re.match(r'L\( *' + str(line_number) + r' *\)', line.python_code):
raise orm.except_orm( raise exceptions.Warning(
_('Error'), _('Error'),
_('Line %s can\'t refer to itself') % str(line_number) _('Line %s can\'t refer to itself') % str(line_number)
) )
@ -74,32 +70,32 @@ class AccountDocumentTemplate(orm.Model):
line.python_code.replace('L', 'self.lines') line.python_code.replace('L', 'self.lines')
) )
except KeyError: except KeyError:
raise orm.except_orm( raise exceptions.Warning(
_('Error'), _('Error'),
_('Code "%s" refers to non existing line') % line.python_code) _('Code "%s" refers to non existing line') % line.python_code)
return self._computed_lines[line_number] return self._computed_lines[line_number]
def compute_lines(self, cr, uid, template_id, input_lines): @api.model
def compute_lines(self, template_id, input_lines):
# input_lines: dictionary in the form {line_number: line_amount} # input_lines: dictionary in the form {line_number: line_amount}
# returns all the lines (included input lines) # returns all the lines (included input lines)
# in the form {line_number: line_amount} # in the form {line_number: line_amount}
template = self.browse(cr, uid, template_id) template = self.browse(template_id)
if len(input_lines) != self._input_lines(cr, uid, template): if len(input_lines) != self._input_lines(template):
raise orm.except_orm( raise exceptions.Warning(
_('Error'), _('Error'),
_('Inconsistency between input lines and ' _('Inconsistency between input lines and '
'filled lines for template %s') % template.name 'filled lines for template %s') % template.name
) )
self._current_template_id = template.id self._current_template_id = template.id
self._cr = cr self._computed_lines = self._generate_empty_lines(template_id)
self._uid = uid
self._computed_lines = self._generate_empty_lines(cr, uid, template_id)
self._computed_lines.update(input_lines) self._computed_lines.update(input_lines)
for line_number in self._computed_lines: for line_number in self._computed_lines:
self.lines(line_number) self.lines(line_number)
return self._computed_lines return self._computed_lines
def check_zero_lines(self, cr, uid, wizard): @api.model
def check_zero_lines(self, wizard):
if not wizard.line_ids: if not wizard.line_ids:
return True return True
for template_line in wizard.line_ids: for template_line in wizard.line_ids:
@ -108,17 +104,14 @@ class AccountDocumentTemplate(orm.Model):
return False return False
class AccountDocumentTemplateLine(orm.Model): class AccountDocumentTemplateLine(models.Model):
_name = 'account.document.template.line' _name = 'account.document.template.line'
_columns = { name = fields.Char(required=True)
'name': fields.char('Name', size=64, required=True), sequence = fields.Integer(string='Sequence', required=True)
'sequence': fields.integer('Sequence', required=True), type = fields.Selection(
'type': fields.selection( [('computed', 'Computed'), ('input', 'User input')],
[('computed', 'Computed'), ('input', 'User input')], string='Type',
'Type', required=True
required=True )
), python_code = fields.Text(string='Python Code')
'python_code': fields.text('Python Code'),
}