2
0
account-financial-tools/account_move_template/account_document_template.py

107 lines
3.9 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
##############################################################################
#
2014-12-03 17:57:52 +01:00
# Copyright (C) 2011 - 2014 Agile Business Group sagl
# (<http://www.agilebg.com>)
# Copyright (C) 2011 Domsense srl (<http://www.domsense.com>)
#
# 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/>.
#
##############################################################################
2014-12-09 11:43:50 +01:00
from openerp import models, fields, api, exceptions, _
2014-12-17 14:14:35 +01:00
from openerp.tools.safe_eval import safe_eval as eval
from functools import partial
import re
2014-12-04 15:01:38 +01:00
class AccountDocumentTemplate(models.Model):
_name = 'account.document.template'
2014-12-04 15:01:38 +01:00
name = fields.Char(required=True)
@api.multi
def _input_lines(self):
count = 0
for line in self.template_line_ids:
if line.type == 'input':
count += 1
return count
@api.multi
def _get_template_line(self, line_number):
for line in self.template_line_ids:
if line.sequence == line_number:
return line
return False
@api.multi
def _generate_empty_lines(self):
lines = {}
for line in self.template_line_ids:
2014-12-04 15:01:38 +01:00
lines[line.sequence] = None
return lines
@api.multi
def lines(self, line_number, computed_lines=None):
if computed_lines is None:
computed_lines = {}
if computed_lines[line_number] is not None:
return computed_lines[line_number]
line = self._get_template_line(line_number)
2014-11-18 21:07:06 +01:00
if re.match(r'L\( *' + str(line_number) + r' *\)', line.python_code):
2014-12-04 15:01:38 +01:00
raise exceptions.Warning(
_('Line %s can\'t refer to itself') % str(line_number)
)
try:
recurse_lines = partial(self.lines, computed_lines=computed_lines)
computed_lines[line_number] = eval(
2014-12-17 14:14:35 +01:00
line.python_code.replace('L', 'recurse_lines'),
locals_dict={'recurse_lines': recurse_lines}
)
except KeyError:
2014-12-04 15:01:38 +01:00
raise exceptions.Warning(
_('Code "%s" refers to non existing line') % line.python_code)
return computed_lines[line_number]
@api.multi
def compute_lines(self, input_lines):
# input_lines: dictionary in the form {line_number: line_amount}
# returns all the lines (included input lines)
# in the form {line_number: line_amount}
if len(input_lines) != self._input_lines():
2014-12-04 15:01:38 +01:00
raise exceptions.Warning(
_('Inconsistency between input lines and '
'filled lines for template %s') % self.name
)
computed_lines = self._generate_empty_lines()
computed_lines.update(input_lines)
for line_number in computed_lines:
computed_lines[line_number] = self.lines(
line_number, computed_lines)
return computed_lines
2014-12-04 15:01:38 +01:00
class AccountDocumentTemplateLine(models.Model):
_name = 'account.document.template.line'
2014-12-04 15:01:38 +01:00
name = fields.Char(required=True)
sequence = fields.Integer(string='Sequence', required=True)
type = fields.Selection(
[('computed', 'Computed'), ('input', 'User input')],
string='Type',
required=True
)
python_code = fields.Text(string='Python Code')