a $Fwj|@sRddlmZddlmZddlmZmZeddGdddZGdd d Zd S) ) annotations) dataclass)datetime timedeltaT)frozenc@s>eZdZUdZded<dZded<dZded<dZded <d S) SchedulePolicyint fast_seconds rank_seconds<candle_secondsi,regime_secondsN)__name__ __module__ __qualname__r __annotations__r rrrr;/home/opc/hanz-radar/runtime/src/hanz_realtime/scheduler.pyrs    rc@s@eZdZddddddZddd d d d Zdd dddZdS)SmartSchedulerNzSchedulePolicy | NoneNone)policyreturncCs|pt|_i|_dS)N)rr _last_run)selfrrrr__init__s zSmartScheduler.__init__strrbool)tasknowrcCs<||}|j|}|dus*|||kr8||j|<dSdS)NTF) _intervalrget)rrr intervalpreviousrrrdues    zSmartScheduler.duer)rrcCsB|jj|jj|jj|jjd}||vr4td|t||dS)N)ZfastZrankZcandleZregimezUnknown task: )seconds)rr r rrKeyErrorr)rrmappingrrrr!szSmartScheduler._interval)N)rrrrr%r!rrrrrsrN) __future__r dataclassesrrrrrrrrrs