:jQdZddlmZmZmZmZddlmZddlm Z m Z m Z m Z ddl mZddlmZddlmZddlmZd d lmZed ZGd d Zy)z Task Recovery Strategy - Enhanced Task Recovery Management This module provides enhanced task recovery functionality using type-safe stage management and configuration-driven approach. )DictListOptionalSet) PipelineStage)AllRecoveredTasks ProviderTaskRecoveredTasksService) IProvider) StageUtils) TaskFactory) get_logger)PipelinemanagerceZdZdZdddeeeffdZdeeee fddfd Z d e ddfd Z d ed e ddfdZd edeedeeddfdZd edeeddfdZdedeefdZy)TaskRecoveryManagerz@Enhanced task recovery strategy using type-safe stage managementpipeliner providersc ||_||_y)zInitialize recovery strategy Args: pipeline: Pipeline instance providers: Dictionary of provider instances N)rr)selfrrs 7/root/.openclaw/workspace/harvester/manager/recovery.py__init__zTaskRecoveryManager.__init__s! " queue_tasksreturnNc P|sytjd|jD]\}}|s |j|}|stj d|5|j j |j}|stj d|u|D]/} |j|tjd|d|1tjdt|d y#t$r%}tjd|d|Yd}~d}~wwxYw) zRecover queue tasks using stage registry Args: queue_tasks: Dictionary mapping stage names to task lists NzStarting queue task recoveryzUnknown stage name: zStage not found: zRecovered task to : zFailed to recover task to z"Queue task recovery completed for z stages) loggerinfoitems_get_stage_enumwarningr get_stagevalueput_taskdebug Exceptionerrorlen)rr stage_nametasks stage_enumstagetaskes rrecover_queue_tasksz'TaskRecoveryManager.recover_queue_tasks%s,   23!,!2!2!4 Q J--j9J!5j\BCMM++J,<,<=E!2:,?@ QQNN4(LL#5j\D6!JK Q# Q0  8[9I8J'RS!QLL#=j\A3!OPPQs$,C77 D%D  D% result_taskscZ|r |jsytjd|jjD]>\}}||jvrtj d|-|j ||@tjdt |jdy)zzRecover result tasks with stage enum validation Args: result_tasks: All recovered tasks data NzStarting result task recoveryProvider not found: z#Result task recovery completed for z providers)rr r!r"r$_recover_provider_tasksr+)rr3 provider_nameprovider_taskss rrecover_result_tasksz(TaskRecoveryManager.recover_result_tasksJs <#9#9  34-9-C-C-I-I-K H )M>DNN2!5m_EF  ( ( G  H  9#l>T>T:U9VV`abrnamer-c|jjj|}|stj d|yt j |tjr'|j||j |jt j |tjr|j||jyy)zRecover tasks for specific provider Args: name: Name of the provider tasks: Recovered tasks for the provider zNo config found for provider: N)r task_configsgetr r$r checkrCHECK_recover_check_tasksinvalidGATHER_recover_acquisition_tasks acquisition)rr:r-configs rr6z+TaskRecoveryManager._recover_provider_tasks^s++//5 NN;D6B C    FM$7$7 8  % %dEKK G   FM$8$8 9  + +D%2C2C D :r check_tasks invalid_keysc.|jjtjj}|st j d|y|jj|}|st j d|yd}|D]8}|r|r||vr tj||}|j||dz }:|dkDrt jd|d |yy#t$r%} t jd|d| Yd} ~ d} ~ wwxYw) zRecover check tasks for provider Args: name: Name of the provider check_tasks: List of services to check invalid_keys: Set of invalid services zCheck stage not found for Nr5rrz Failed to create check task for r Recovered z check tasks for )rr%rr?r&r r$rr=rcreate_check_taskr'r)r*r!) rr:rFrGstage_instanceproviderrecovered_countservice check_taskr1s rr@z(TaskRecoveryManager._recover_check_tasksrs001D1D1J1JK NN7v> ? >>%%d+ NN1$8 9 # MG|<0G M(::4I '' 31$ M Q  KK*_$55FtfM N  M ?vRsKLL Ms,C&& D/DDacquisition_tasksc>|jjtjj}|st j d|y|jj|}|st j d|y|j}d}|D]0} tj|||}|j||dz }2|dkDrt jd|d |yy#t$r%} t jd|d| Yd} ~ ~d} ~ wwxYw) zRecover acquisition tasks for provider Args: name: Name of the provider acquisition_tasks: List of URLs to acquire z Acquisition stage not found for Nr5rrz&Failed to create acquisition task for rrIz acquisition tasks for )rr%rrBr&r r$rr= get_patternsrcreate_acquisition_taskr'r)r*r!) rr:rPrKrLpatternsrMurlacquisition_taskr1s rrCz.TaskRecoveryManager._recover_acquisition_taskss 001E1E1K1KL NN=dVD E >>%%d+ NN1$8 9 ((*$ SC S#.#F#FtSRZ#[ ''(891$  S Q  KK*_$55LTFS T  S EdV2aSQRR Ss-C.. D7DDr,c8 t|S#t$rYywxYw)zGet stage enum from stage name Args: stage_name: Name of the stage Returns: PipelineStage enum or None if not found N)r ValueError)rr,s rr#z#TaskRecoveryManager._get_stage_enums$  , ,  s )__name__ __module__ __qualname____doc__rstrr rrr r2rr9r r6r rr@rCrrr#rrrrsJ##S)^8L##TtCl9K4K/L#TQU#TJc1Bctc(ECEE4E("O"O4="OX[\cXd"Oim"OHUsUtCyUUYUB # (=2I rrN)r\typingrrrr core.enumsr core.modelsrr r r core.typesr stage.baser stage.factoryr tools.loggerrrrr rr^rrrfs@-,$PP !%# I kkr