trytondo-account_co_co/account.py

110 lines
3.7 KiB
Python
Raw Permalink Normal View History

from trytond.pool import Pool, PoolMeta
from trytond.model.exceptions import AccessError
from trytond.i18n import gettext
2021-03-24 02:41:17 +01:00
from trytond.model import ModelView
2021-03-24 02:47:27 +01:00
from decimal import Decimal
2021-03-24 02:59:46 +01:00
from itertools import groupby
2021-03-24 02:47:27 +01:00
__all__ = ['Account','Move','Line']
class Account(metaclass=PoolMeta):
__name__ = 'account.account'
@classmethod
def __setup__(cls):
super(Account, cls).__setup__()
cls.party_required.domain = [()]
cls.party_required.states = {}
class Move(metaclass=PoolMeta):
__name__ = 'account.move'
@classmethod
def create(cls, vlist):
pool = Pool()
Sequence = pool.get('ir.sequence')
Journal = pool.get('account.journal')
Account_Configuration = pool.get('account.configuration')
configuration = Account_Configuration(1)
draft_default = configuration.default_draft_sequence
vlist = [x.copy() for x in vlist]
for vals in vlist:
if not vals.get('number'):
journal_id = (vals.get('journal')
or Transaction().context.get('journal'))
if journal_id:
journal = Journal(journal_id)
if journal.sequence:
vals['number'] = Sequence.get_id(draft_default.id)
moves = super(Move, cls).create(vlist)
cls.validate_move(moves)
return moves
@classmethod
@ModelView.button
def post(cls, moves):
pool = Pool()
Date = pool.get('ir.date')
Line = pool.get('account.move.line')
2021-03-24 02:51:19 +01:00
Sequence = pool.get('ir.sequence')
for move in moves:
amount = Decimal('0.0')
if not move.lines:
raise PostError(
gettext('account.msg_post_empty_move', move=move.rec_name))
company = None
for line in move.lines:
amount += line.debit - line.credit
if not company:
company = line.account.company
if not company.currency.is_zero(amount):
raise PostError(
gettext('account.msg_post_unbalanced_move',
move=move.rec_name))
for move in moves:
move.state = 'posted'
move.number = Sequence.get_id(move.journal.sequence.id)
if not move.post_number:
move.post_date = Date.today()
2021-03-24 02:56:37 +01:00
move.post_number = Sequence.get_id(
move.period.post_move_sequence_used.id)
def keyfunc(l):
return l.party, l.account
to_reconcile = [l for l in move.lines
if ((l.debit == l.credit == Decimal('0'))
and l.account.reconcile)]
to_reconcile = sorted(to_reconcile, key=keyfunc)
for _, zero_lines in groupby(to_reconcile, keyfunc):
Line.reconcile(list(zero_lines))
cls.save(moves)
class Line(metaclass=PoolMeta):
__name__ = 'account.move.line'
@classmethod
def __setup__(cls):
super(Line, cls).__setup__()
cls.party.states = {}
def check_account(self):
if not self.account.type or self.account.closed:
raise AccessError(
gettext('account.msg_line_closed_account',
account=self.account.rec_name))
if self.account.party_required:
if bool(self.party) != bool(self.account.party_required):
error = 'party_set' if self.party else 'party_required'
raise AccessError(
gettext('account.msg_line_%s' % error,
account=self.account.rec_name,
line=self.rec_name))