from api import PortfoliosRisk, SignalType
from framework import component
from rebalance.base_signal import BaseRebalanceSignal
from rebalance.dao import robo_rebalance_signal as rrs


@component(bean_name='init')
class InitSignalBuilder(BaseRebalanceSignal):

    @property
    def signal_type(self) -> SignalType:
        return SignalType.INIT

    def is_trigger(self, day, risk: PortfoliosRisk) -> bool:
        return rrs.get_count(risk=risk) == 0