flectra/addons/account_asset/wizard/asset_depreciation_summary.py

# Part of Flectra. See LICENSE file for full copyright and licensing
# details.

from flectra import api, fields, models
from datetime import datetime
from flectra.tools import DEFAULT_SERVER_DATE_FORMAT as DF


class AssetDepreciationWizard(models.TransientModel):
    _name = "asset.depreciation.wizard"

    @api.depends('start_date')
    def _compute_get_end_date(self):
        if self.start_date:
            last_day = self.env.user.company_id.fiscalyear_last_day
            last_month = self.env.user.company_id.fiscalyear_last_month
            date = datetime.strptime(self.start_date, DF).date()
            year = date.year
            fiscal_date = date.replace(month=last_month, day=last_day, year=year)
            if fiscal_date < date:
                fiscal_date = date.replace(month=last_month, day=last_day, year=year + 1)
            self.end_date = datetime.strftime(fiscal_date, '%Y-%m-%d')

    start_date = fields.Date(string="Start Date", required=True)
    end_date = fields.Date(compute='_compute_get_end_date', store=True)

    @api.multi
    def print_depreciation_lines(self):
        data = {'form': self.read(['start_date', 'end_date'])[0]}
        return self.env.ref(
            'account_asset.asset_depreciation_report').report_action(
            self,
            data=data,
            config=False)