113 lines
4.4 KiB
Python
113 lines
4.4 KiB
Python
# The COPYRIGHT file at the top level of this repository contains the full
|
|
# copyright notices and license terms.
|
|
from collections import OrderedDict
|
|
|
|
from trytond.model import ModelView, ModelSQL, fields
|
|
from trytond.report import Report
|
|
from trytond.pool import Pool, PoolMeta
|
|
from trytond.transaction import Transaction
|
|
|
|
|
|
class Account(metaclass=PoolMeta):
|
|
'Income Statement'
|
|
__name__ = 'account.income_statement.context'
|
|
utility_temp = fields.Boolean('Utility Temp', help='Permited see the utility without have any account move')
|
|
|
|
|
|
class IncomeStatement(Report):
|
|
'Income Statement'
|
|
__name__ = 'account.income_statement'
|
|
|
|
@classmethod
|
|
def get_context(cls, records, header, data):
|
|
pool = Pool()
|
|
Type = pool.get('account.account.type')
|
|
Account = pool.get('account.account')
|
|
Period = pool.get('account.period')
|
|
Fiscalyear = pool.get('account.fiscalyear')
|
|
Company = pool.get('company.company')
|
|
Context = pool.get('account.income_statement.context')
|
|
context = Transaction().context
|
|
|
|
report_context = super(IncomeStatement, cls).get_context(records, header, data)
|
|
|
|
context_fields = Context.fields_get(['start_period', 'fiscalyear'])
|
|
types = Type.search([
|
|
('statement', '=', 'income')
|
|
])
|
|
accounts_types = []
|
|
company_id = Transaction().context.get('company')
|
|
company = Company(company_id)
|
|
print(context)
|
|
print('---------------------------------------------------')
|
|
print(context_fields)
|
|
print('---------------------------------------------------')
|
|
print(report_context['data'])
|
|
print('---------------------------------------------------')
|
|
records = Type(report_context['data']['id'])
|
|
fiscalyear_id = Transaction().context.get('fiscalyear')
|
|
fiscalyear_cmp = Transaction().context.get('fiscalyear_cmp')
|
|
start_period = Transaction().context.get('start_period')
|
|
start_period_cmp = Transaction().context.get('start_period_cmp')
|
|
end_period = Transaction().context.get('end_period')
|
|
end_period_cmp = Transaction().context.get('end_period_cmp')
|
|
comparison = Transaction().context.get('comparison')
|
|
if start_period:
|
|
start_period = Period(start_period)
|
|
if end_period:
|
|
end_period = Period(end_period)
|
|
|
|
dom_periods= [
|
|
('type', '=', 'standard'),
|
|
('fiscalyear', '=', fiscalyear_id),
|
|
]
|
|
if start_period:
|
|
dom_periods.append(('start_date', '>=', start_period.start_date))
|
|
if end_period:
|
|
dom_periods.append(('start_date', '<=', end_period.start_date))
|
|
range_periods = Period.search(dom_periods)
|
|
periods_ids = [p.id for p in range_periods]
|
|
od = {}
|
|
with Transaction().set_context(periods=periods_ids):
|
|
while types:
|
|
type_ = types.pop()
|
|
if type_.statement == 'income':
|
|
accounts = Account.search([
|
|
('type', '=', ''),
|
|
])
|
|
|
|
if accounts:
|
|
setattr(type_, 'accounts', accounts)
|
|
accounts_types.append((type_.sequence, type_))
|
|
if type_.childs:
|
|
types.extend(list(type_.childs))
|
|
|
|
if accounts_types:
|
|
od = OrderedDict(sorted(dict(accounts_types).items()))
|
|
|
|
types_added = []
|
|
filtered_records = []
|
|
for k, v in od.items():
|
|
childs = []
|
|
for c in v.childs:
|
|
if c.id not in types_added:
|
|
childs.append(c)
|
|
types_added.extend([v.id, c.id])
|
|
|
|
setattr(v, 'childs', childs)
|
|
filtered_records.append(v)
|
|
|
|
report_context['start_period'] = start_period
|
|
report_context['start_period_cmp'] = Period(start_period_cmp)
|
|
report_context['end_period'] = end_period
|
|
report_context['end_period_cmp'] = Period(end_period_cmp)
|
|
report_context['fiscalyear'] = Fiscalyear(fiscalyear_id).name
|
|
report_context['fiscalyear_cmp'] = Fiscalyear(fiscalyear_cmp).name if fiscalyear_cmp else ''
|
|
report_context['records'] = filtered_records
|
|
report_context['comparison'] = comparison
|
|
report_context['company'] = Company(Transaction().context.get('company'))
|
|
report_context['date'] = Transaction().context.get('date')
|
|
|
|
|
|
return report_context
|