Cédric Krier
1c3e728871
We add the flake8 configuration used so we ensure everyone uses the same. We remove the usage of __all__ for non public API. When possible, we rationalize the class name according to its __name__ and module. issue9082 review297061002
74 lines
2.6 KiB
Python
74 lines
2.6 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 itertools import groupby
|
|
from operator import itemgetter
|
|
|
|
from sql.operators import Concat
|
|
|
|
from trytond.i18n import gettext
|
|
from trytond.model import ModelView
|
|
from trytond.pool import PoolMeta, Pool
|
|
from trytond.transaction import Transaction
|
|
from trytond.tools import grouped_slice, reduce_ids
|
|
|
|
from trytond.modules.account.exceptions import PostError
|
|
|
|
|
|
class Journal(metaclass=PoolMeta):
|
|
__name__ = 'account.journal'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(Journal, cls).__setup__()
|
|
cls.type.selection.append(('statement', "Statement"))
|
|
|
|
|
|
class Move(metaclass=PoolMeta):
|
|
__name__ = 'account.move'
|
|
|
|
@classmethod
|
|
def __register__(cls, module_name):
|
|
pool = Pool()
|
|
StatementLine = pool.get('account.statement.line')
|
|
cursor = Transaction().connection.cursor()
|
|
sql_table = cls.__table__()
|
|
|
|
super(Move, cls).__register__(module_name)
|
|
|
|
# Migration from 3.4:
|
|
# account.statement.line origin changed to account.statement
|
|
statement_line = StatementLine.__table__()
|
|
cursor.execute(*sql_table.join(statement_line,
|
|
condition=(
|
|
Concat(StatementLine.__name__ + ',', statement_line.id)
|
|
== sql_table.origin
|
|
)
|
|
).select(sql_table.id, statement_line.statement,
|
|
order_by=(sql_table.id, statement_line.statement)))
|
|
for statement_id, values in groupby(cursor.fetchall(), itemgetter(1)):
|
|
ids = [x[0] for x in values]
|
|
for sub_ids in grouped_slice(ids):
|
|
red_sql = reduce_ids(sql_table.id, sub_ids)
|
|
cursor.execute(*sql_table.update(
|
|
columns=[sql_table.origin],
|
|
values=['account.statement,%s' % statement_id],
|
|
where=red_sql))
|
|
|
|
@classmethod
|
|
def _get_origin(cls):
|
|
return super(Move, cls)._get_origin() + ['account.statement']
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
def post(cls, moves):
|
|
pool = Pool()
|
|
Statement = pool.get('account.statement')
|
|
for move in moves:
|
|
if (isinstance(move.origin, Statement)
|
|
and move.origin.state != 'posted'):
|
|
raise PostError(
|
|
gettext('account_statement.msg_post_statement_move',
|
|
move=move.rec_name,
|
|
statement=move.origin.rec_name))
|
|
super().post(moves)
|