:j &dZddlmZmZddlmZmZmZmZm Z ddl m Z ddl m Z e dZeGdd ZGd d ZeZ dd edeeedeeedededef dZdefdZd edeefdZdeefdZGddZy )z Stage registry system for dynamic pipeline management. Provides registration, discovery, and metadata management for pipeline stages. ) dataclassfield)CallableDictListOptionalType) get_logger)BasePipelineStagestageceZdZUdZeed<eded<eeZ e eed<eeZ e eed<dZ e ed <d Zeed <d Zy )StageDefinitionz=Definition of a pipeline stage with metadata and dependenciesnamer stage_class)default_factory depends_on produces_forTrequired descriptionc`|js td|js tdy)zValidate stage definitionzStage name cannot be emptyzStage class cannot be NoneN)r ValueErrorrselfs 5/root/.openclaw/workspace/harvester/stage/registry.py __post_init__zStageDefinition.__post_init__s/yy9: :9: : N)__name__ __module__ __qualname____doc__str__annotations__r rlistrrrrboolrrrrrrsUG I)**!$7JS 7#D9L$s)9HdK;rrcxeZdZdZdZdeddfdZdedeefdZ de efd Z de efd Z dede fd Zd d Zy) StageRegistryz0Registry for managing pipeline stage definitionsc i|_d|_y)NF)_stages _initializedrs r__init__zStageRegistry.__init__(s35 !r definitionreturnNc|j|jvr"tjd|j||j|j<tj d|jy)zRegister a stage definitionz&Overriding existing stage definition: zRegistered stage: N)rr+loggerwarningdebug)rr.s rregisterzStageRegistry.register,sV ??dll * NNCJOOCTU V(2 Z__% )*//):;r&r@rBr'rrr)r)%sn:"=?=t=&& 9&+$/+)DI)$3$4$8rr)Nrrrrrr/cHdtddtdffd }|S)z)Decorator for registering pipeline stagesrr r/cdt|xsgxsg}tj||S)N)rrrrrr)r _registryr4)rr.rrrrrs r decoratorz!register_stage..decoratorWs?$#!'R%+#   :&r)r )rrrrrrFs````` rregister_stagerGNs/ t$78 TBU=V   rctS)zGet the global stage registryrEr'rr get_registryrJfs rc,tj|Sr6)rEr7)rs rget_stage_definitionrLks == rc*tjS)zList all registered stage names)rEr>r'rrlist_registered_stagesrNps    !!rcLeZdZdZedefdZdedee fdZ dede fdZ y)StageRegistryMixinz7Mixin to provide registry access to pipeline componentsr/ctS)zGet the stage registryrIrs rregistryzStageRegistryMixin.registryxs rrc8|jj|Sr6)rRr7r8s r get_stage_defz StageRegistryMixin.get_stage_def}s}}  &&rc8|jj|S)z!Check if stage exists in registry)rRr@r8s r stage_existszStageRegistryMixin.stage_existss}}##D))rN) rr r!r"propertyr)rRr#rrrTr&rVr'rrrPrPusIA -'#'(?*C'***rrP)NNTr)r" dataclassesrrtypingrrrrr tools.loggerr baser r1rr)rEr#r&rGrJrLrNrPr'rrr\s )77## G  ;; ;$"8"8L O '+(,  c#49%     0m sx'@ "S " **r