trytonpsk-account_col/asset.py

from trytond.pool import Pool, PoolMeta
from trytond.model import ModelView, ModelSQL, fields
from datetime import timedelta
from trytond.pyson import Eval


class AssetLocation(ModelView, ModelSQL):
    'Location Asset'
    __name__ = 'account_col.asset_location'

    name = fields.Char('name')
    address = fields.Char('address')


class DepreciationKind(ModelView, ModelSQL):
    'Depreciation Kind'
    __name__ = 'account_col.depreciation_kind'

    name = fields.Char('Name')
    years = fields.Integer('Years Depreciation')
    colgaap = fields.Boolean('Colgaap')

    @staticmethod
    def default_colgaap():
        return False


class Asset(metaclass=PoolMeta):
    __name__ = 'account.asset'

    kind = fields.Many2One('account_col.depreciation_kind', 'Kind',
        required=True, domain=[('colgaap', '=', Eval('colgaap', False))])
    location = fields.Many2One('account_col.asset_location', "Location")
    type = fields.Selection([
        ('AFP', 'Activo fijo propios'),
        ('CD', 'Comodato'),
        ('AC', 'Activo Controlado')
        ], 'Type', required=True)
    code = fields.Char('Code')
    colgaap = fields.Boolean('Colgaap')

    @fields.depends('kind', 'start_date')
    def on_change_with_end_date(self, name=None):
        if self.kind:
            days = 365 * self.kind.years
            return self.start_date + timedelta(days=days)

    @staticmethod
    def default_colgaap():
        return False

    def get_move(self, line):
        move = super(Asset, self).get_move(line)
        Company = Pool().get('company.company')
        company = Company.search([])
        if line.asset.colgaap:
            move.method = 'colgaap'
        else:
            move.method = 'ifrs'
        if move:
            for line in move.lines:
                if line.account.type.statement == 'income':
                    line.party = company[0].party
        return move

    @classmethod
    def create_lines(cls, assets):
        for asset in assets:
            if asset.type == 'AC' or asset.type == 'CD':
                continue
            super(Asset, cls).create_lines([asset])