Files
tryton/account.py
T
2009-03-20 12:22:19 +01:00

182 lines
8.4 KiB
Python

#This file is part of Tryton. The COPYRIGHT file at the top level of
#this repository contains the full copyright notices and license terms.
from trytond.model import ModelView, ModelSQL, fields
class FiscalYear(ModelSQL, ModelView):
_name = 'account.fiscalyear'
out_invoice_sequence = fields.Many2One('ir.sequence.strict',
'Customer Invoice Sequence', required=True,
domain="[('code', '=', 'account.invoice')," \
"['OR', ('company', '=', company)," \
"('company', '=', False)]]",
context="{'code': 'account.invoice', 'company': company}")
in_invoice_sequence = fields.Many2One('ir.sequence.strict',
'Supplier Invoice Sequence', required=True,
domain="[('code', '=', 'account.invoice')," \
"['OR', ('company', '=', company)," \
"('company', '=', False)]]",
context="{'code': 'account.invoice', 'company': company}")
out_credit_note_sequence = fields.Many2One('ir.sequence.strict',
'Customer Credit Note Sequence', required=True,
domain="[('code', '=', 'account.invoice')," \
"['OR', ('company', '=', company)," \
"('company', '=', False)]]",
context="{'code': 'account.invoice', 'company': company}")
in_credit_note_sequence = fields.Many2One('ir.sequence.strict',
'Supplier Credit Note Sequence', required=True,
domain="[('code', '=', 'account.invoice')," \
"['OR', ('company', '=', company)," \
"('company', '=', False)]]",
context="{'code': 'account.invoice', 'company': company}")
def __init__(self):
super(FiscalYear, self).__init__()
self._constraints += [
('check_invoice_sequences', 'different_invoice_sequence'),
]
self._error_messages.update({
'change_invoice_sequence': 'You can not change ' \
'the invoice sequence if there is already ' \
'an invoice opened in the fiscalyear',
'different_invoice_sequence': 'You must have different ' \
'invoice sequences per fiscal year!',
})
def check_invoice_sequences(self, cursor, user, ids):
for fiscalyear in self.browse(cursor, user, ids):
for sequence in ('out_invoice_sequence', 'in_invoice_sequence',
'out_credit_note_sequence', 'in_credit_note_sequence'):
if self.search(cursor, user, [
(sequence, '=', fiscalyear[sequence].id),
('id', '!=', fiscalyear.id),
]):
return False
return True
def write(self, cursor, user, ids, vals, context=None):
invoice_obj = self.pool.get('account.invoice')
for sequence in ('out_invoice_sequence', 'in_invoice_sequence',
'out_credit_note_sequence', 'in_credit_note_sequence'):
if vals.get(sequence):
for fiscalyear in self.browse(cursor, user, ids,
context=context):
if fiscalyear[sequence] and \
fiscalyear[sequence].id != \
vals[sequence]:
if invoice_obj.search(cursor, user, [
('invoice_date', '>=', fiscalyear.start_date),
('invoice_date', '<=', fiscalyear.end_date),
('number', '!=', False),
('type', '=', sequence[:-9]),
], context=context):
self.raise_user_error(cursor,
'change_invoice_sequence', context=context)
return super(FiscalYear, self).write(cursor, user, ids, vals,
context=context)
FiscalYear()
class Period(ModelSQL, ModelView):
_name = 'account.period'
out_invoice_sequence = fields.Many2One('ir.sequence.strict',
'Customer Invoice Sequence',
domain="[('code', '=', 'account.invoice')]",
context="{'code': 'account.invoice'}",
states={
'required': "type == 'standard'",
'invisible': "type != 'standard'",
})
in_invoice_sequence = fields.Many2One('ir.sequence.strict',
'Supplier Invoice Sequence',
domain="[('code', '=', 'account.invoice')]",
context="{'code': 'account.invoice'}",
states={
'required': "type == 'standard'",
'invisible': "type != 'standard'",
})
out_credit_note_sequence = fields.Many2One('ir.sequence.strict',
'Customer Credit Note Sequence',
domain="[('code', '=', 'account.invoice')]",
context="{'code': 'account.invoice'}",
states={
'required': "type == 'standard'",
'invisible': "type != 'standard'",
})
in_credit_note_sequence = fields.Many2One('ir.sequence.strict',
'Supplier Credit Note Sequence',
domain="[('code', '=', 'account.invoice')]",
context="{'code': 'account.invoice'}",
states={
'required': "type == 'standard'",
'invisible': "type != 'standard'",
})
def __init__(self):
super(Period, self).__init__()
self._constraints += [
('check_invoice_sequences', 'check_invoice_sequences'),
]
self._error_messages.update({
'change_invoice_sequence': 'You can not change ' \
'the invoice sequence if there is already ' \
'an invoice opened in the period',
'check_invoice_sequences': 'You must have different ' \
'invoice sequences per fiscal year and ' \
'in the same company!',
})
def check_invoice_sequences(self, cursor, user, ids):
for period in self.browse(cursor, user, ids):
for sequence in ('out_invoice_sequence', 'in_invoice_sequence',
'out_credit_note_sequence', 'in_credit_note_sequence'):
if self.search(cursor, user, [
(sequence, '=', period[sequence].id),
('fiscalyear', '!=', period.fiscalyear.id),
]):
return False
if period[sequence].company \
and period[sequence].company.id != \
period.fiscalyear.company.id:
return False
return True
def create(self, cursor, user, vals, context=None):
fiscalyear_obj = self.pool.get('account.fiscalyear')
vals = vals.copy()
if vals.get('fiscalyear'):
fiscalyear = fiscalyear_obj.browse(cursor, user, vals['fiscalyear'],
context=context)
for sequence in ('out_invoice_sequence', 'in_invoice_sequence',
'out_credit_note_sequence', 'in_credit_note_sequence'):
if not vals.get(sequence):
vals[sequence] = fiscalyear[sequence].id
return super(Period, self).create(cursor, user, vals, context=context)
def write(self, cursor, user, ids, vals, context=None):
invoice_obj = self.pool.get('account.invoice')
if isinstance(ids, (int, long)):
ids = [ids]
for sequence in ('out_invoice_sequence', 'in_invoice_sequence',
'out_credit_note_sequence', 'in_credit_note_sequence'):
if vals.get(sequence):
for period in self.browse(cursor, user, ids, context=context):
if period[sequence] and \
period[sequence].id != \
vals[sequence]:
if invoice_obj.search(cursor, user, [
('date', '>=', period.start_date),
('date', '<=', period.end_date),
('number', '!=', False),
('type', '=', sequence[:-9]),
], context=context):
self.raise_user_error(cursor,
'change_invoice_sequence', context=context)
return super(Period, self).write(cursor, user, ids, vals,
context=context)
Period()