:j>dZddlZddlmZmZmZmZddlmZm Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZdd lmZmZmZmZdd lmZGd deZGddeZGddeZGddeZ edZ!GddZ"y)a  Status Builders - System Status Construction Utilities This module provides a robust, type-safe builder pattern implementation for constructing SystemStatus objects with the following features: - Type-safe interfaces with Protocol definitions for external dependencies - Comprehensive parameter validation with custom exception types - Dependency injection support for better testability - Unified error handling with structured logging - Method responsibility separation for maintainability - Factory methods for convenient builder creation - Build state tracking to prevent misuse Key Components: StatusBuilder: Main builder class with full validation and error handling Custom Exceptions: Specific exception types for better error handling IPipelineBase: Abstract base class for pipeline implementations Type Protocols: Interface definitions for external dependencies Usage Examples: # Basic usage with validation builder = StatusBuilder.quick() status = (builder .with_basic_info(runtime=10.5, state=SystemState.RUNNING) .with_task_metrics(completed=100, failed=2) .build()) # Advanced usage with dependency injection custom_collector = StatusCollector(...) builder = StatusBuilder.create(collector=custom_collector) N)AnyDictListOptional) ErrorType SystemState) TaskMetrics)IPipelineStats IProvider) get_logger)StatusCollector) ProviderState)PerformanceMetricsPersistenceMetricsProviderStatus SystemStatus)ICollectorWithAlertsceZdZdZy)StatusBuilderErrorz'Base exception for StatusBuilder errorsN__name__ __module__ __qualname____doc__4/root/.openclaw/workspace/harvester/state/builder.pyrr3s1rrceZdZdZy)BuilderAlreadyBuiltErrorzBRaised when trying to modify a builder that has already been builtNrrrrr r 9sLrr ceZdZdZy)InvalidParameterErrorz>Raised when invalid parameters are provided to builder methodsNrrrrr"r"?sHrr"ceZdZdZy)DataCollectionErrorz7Raised when data collection from external sources failsNrrrrr$r$EsArr$statec eZdZdZd*deefdZed*deeddfdZed+dZ d,dZ d e ddfd Z d e d eddfd Zd e deddfdZd-dedededdfdZd.de de ddfdZdeddfdZdee efddfdZdee efddfdZdee efddfdZdee efddfdZdeeddfd Z d!e d"e!ddfd#Z"d+d$Z#d/d%e d&e$defd'Z%deeddfd(Z&de'fd)Z(y)0 StatusBuilderzOBuilder pattern for constructing SystemStatus objects with dependency injectionN collectorc>||_t|_d|_y)zInitialize builder with dependency injection Args: collector: StatusCollector instance for pipeline data collection FN)r(rstatus_built)selfr(s r__init__zStatusBuilder.__init__Qs #"n  rreturnc||S)zFactory method to create a new StatusBuilder instance Args: collector: Optional StatusCollector instance Returns: New StatusBuilder instance r)clsr(s rcreatezStatusBuilder.create[s9~rc|S)zCreate a StatusBuilder with default dependencies for quick usage Returns: StatusBuilder with default dependencies r)r0s rquickzStatusBuilder.quickgs u rc2|jr tdy)z$Ensure builder hasn't been built yetz9Cannot modify StatusBuilder after build() has been calledN)r+r )r,s r_ensure_not_builtzStatusBuilder._ensure_not_builtps ;;*+fg g rruntimecVt|ttfr|dkrtd|y)zValidate runtime parameterrz,Runtime must be a non-negative number, got: N) isinstanceintfloatr")r,r6s r_validate_runtimezStatusBuilder._validate_runtimeus1'C<0GaK'*VW^V_(`a a5@r operationerrorc~d|d|jjd|}tj|t |j t rA |j j|jtjj|yy#t$r"}tjd|Yd}~yd}~wwxYw)zHandle data collection errors consistently Args: operation: Description of the operation that failed error: The original exception z Failed to : zFailed to add error alert: N) __class__rloggerr=r8r(r_add_error_alertr*rDATA_COLLECTION_ERRORvalue Exceptionwarning)r,r<r= error_msg alert_errors r_handle_collection_errorz&StatusBuilder._handle_collection_errorzs! 2eoo.F.F-Gr%Q  Y dnn&: ; L// Y=\=\=b=bdij < L!<[MJKK Ls?B B<B77B<r%c|j|j|||j_||j_t j |j_|S)akSet basic system information Args: runtime: System runtime in seconds (must be non-negative) state: Current system state Returns: Self for method chaining Raises: BuilderAlreadyBuiltError: If builder has already been built InvalidParameterError: If runtime is invalid )r5r;r*r6r%time timestamp)r,r6r%s rwith_basic_infozStatusBuilder.with_basic_infosM   w'% !  $   r completedfailedpendingc|jd|fd|fd|ffD]*\}}t|tr|dkst|d|t ||||j _|S)aSet task metrics Args: completed: Completed tasks (must be non-negative) failed: Failed tasks (must be non-negative) pending: Pending tasks (must be non-negative) Returns: Self for method chaining Raises: RuntimeError: If builder has already been built ValueError: If any metric is negative rNrOrPrz& must be a non-negative integer, got: )rNrOrP)r5r8r9r"r r*tasks)r,rNrOrPnamerDs rwith_task_metricszStatusBuilder.with_task_metricss  ))4x6H9V]J^_ dKD%eS)UQY+tf4Z[`Za,bcc d()FT[\  r throughput success_ratec*|jt|ttfr|dkrt d|t|ttfr d|cxkrdksnt d|t ||||dkDrd|z nd|j _|S)arSet performance metrics Args: throughput: Tasks per second (must be non-negative) success_rate: Success rate (must be between 0.0 and 1.0) Returns: Self for method chaining Raises: RuntimeError: If builder has already been built ValueError: If metrics are out of valid range rz&Throughput must be non-negative, got: g?z/Success rate must be between 0.0 and 1.0, got: )rUtasks_per_secondrV error_rate)r5r8r9r:r"rr* performance)r,rUrVs rwith_performance_metricsz&StatusBuilder.with_performance_metricss  *sEl3zA~'*PQ[P\(]^ ^,e 5c\>XUX>X'*YZfYg(hi i"4!'%-9A-=s\)3 #    rpipelinecj|j|r% |j}||j_|S|S#t$rF|j }t jdt|||j_Y|SwxYw#t$r}|jd|Yd}~|Sd}~wwxYw)aSet pipeline statistics using collector Args: pipeline: Pipeline object implementing IPipelineBase interface Returns: Self for method chaining Raises: RuntimeError: If builder has already been built zGot dynamic pipeline stats: zcollect pipeline statisticsN) r5 get_all_statsr*r]AttributeErrorget_dynamic_statsrAdebugtyperErI)r,r]pipeline_statses rwith_pipeline_statsz!StatusBuilder.with_pipeline_statss   P:%-%;%;%=N+9DKK( t &:%-%?%?%ANLL#?^@T?U!VW+9DKK(  :  P--.KQOO  Ps.!:A B B B  B B2B--B2 result_statscn|j|r"|j||j||S)a1Set result statistics using direct field mapping Args: result_stats: Dictionary mapping provider names to their persistence metrics Returns: Self for method chaining Raises: BuilderAlreadyBuiltError: If builder has already been built )r5_update_system_level_metrics_update_provider_level_metrics)r,rgs rwith_result_statszStatusBuilder.with_result_statss4    - -l ;  / / = rcZ |j||jjtj d|jj j d|jj jy#t$r}|jd|Yd}~yd}~wwxYw)zzUpdate system-level aggregated metrics Args: result_stats: Dictionary of provider statistics zUpdated system metrics: keys=, links=zupdate system-level metricsN) rjr*calculate_overall_metricsrArbresourcetotallinksrErI)r,rgres rriz*StatusBuilder._update_system_level_metrics s  L  / / = KK 1 1 3 LL/ 0D0D0J0J/K8TXT_T_ThThTnTnSop  L  ) )*G K K LsBB B*B%%B*c ,|jD]\}}|rt|tstj d|d|4 ||j j vrq|j j |}|j|_tjd|d|jjd|jjntj d|dy #t$r}|jd||Yd }~d }~wwxYw) zvUpdate individual provider metrics Args: result_stats: Dictionary of provider statistics zSkip update provider z due to invalid metrics: Updated provider z metrics: valid=rmz Provider z7 not found in status.providers, skipping metrics updatezupdate metrics for provider N) itemsr8rrArFr* providersrorbvalidrqrErI)r,rgrSstatsprovider_statusres rrjz,StatusBuilder._update_provider_level_metricss& (--/ XKD% 52D E!6tfTop=qqx y    ( (): ; rfieldrDc|jt|j|std|dt |j||t j d|d||S)afSet a custom field on the status object Args: field: Name of the field to set value: Value to set Returns: Self for method chaining Raises: BuilderAlreadyBuiltError: If builder has already been built InvalidParameterError: If field doesn't exist on status object zStatus object has no field ''zSet custom field  = )r5hasattrr*r"setattrrArb)r,rrDs rwith_custom_fieldzStatusBuilder.with_custom_fieldZsb  t{{E*'*FugQ(OP P UE* (s5':; rc ||j|jD]g\}} t|j|r3t |j||t j d|d|nt j d|i|S#t$r%}t jd|d|Yd}~d}~wwxYw)aSet additional data fields Args: **kwargs: Additional data to set as custom fields Returns: Self for method chaining Raises: BuilderAlreadyBuiltError: If builder has already been built zSet additional field rzSkipping unknown field zFailed to set additional field r?N) r5rtrr*rrArbrErF)r,kwargsrrDres rwith_additional_dataz"StatusBuilder.with_additional_dataqs  "LLN OLE5 O4;;.DKK6LL#8s5'!JKLL#:5'!BC  O  O!@r!MNN OsA!B  B;B66B;rSenabledcht|||rtjStjS)zCreate a ProviderStatus object with default values Args: name: Name of the provider enabled: Whether the provider is enabled Returns: ProviderStatus: Configured provider status object )rSrr%)rrACTIVEDISABLED)r,rSrs rr{z%StatusBuilder._create_provider_statuss8*1-&&  8E7M7M  rc  |D](}|j|jjvr|jj|j}|j|_|j|_|j |_|j |_tjd|jd|jd|jd|j d|j tjd|j||jj|j<+y#t$r}|jd|Yd}~yd}~wwxYw) z$Update provider stage configurationsrsz stages: S=z, G=z, V=z, I=z!Creating new provider status for zupdate provider stagesN) rSr*ru searchable gatherable checkable inspectablerArbrErI)r,r}rxpsres rrz%StatusBuilder._update_provider_statussM G#4 R"''4;;+@+@@../C/CDB$3$>$>BM$3$>$>BM#2#<#*>? R G  ) )*BA F F GsD.D11 E:EEc|jr td |jjd|_|jS#t$r"}t j d|Yd}~9d}~wwxYw)zBuild final SystemStatus object Returns: Constructed SystemStatus object Raises: BuilderAlreadyBuiltError: If build() has already been called z:build() can only be called once per StatusBuilder instancez%Failed to calculate derived metrics: NT)r+r r*rnrErArF)r,res rbuildzStatusBuilder.buildsk ;;*+gh h H KK 1 1 3 {{  H NNB1#F G G HsA A1A,,A1)N)r.r')r.N)rrr)rXrX)T))rrrrrrr- classmethodr1r3r5r:r;strrErIrrMr9rTr\r rfrrrkrirjr r|rrrrrrboolr{rrrrrrr'r'NsY(?"; x8 O  h bb4b L#LiLDL&u[_,3CcZi25e^m<N>d38J3J.KP_(LcCU>U9VL[_L X4EW@W;XX]aX.T#y.-Ao"d>6J*s3?.4 C $ . Gn9MGRVG*|rr')#rrKtypingrrrr core.enumsrr core.metricsr core.typesr r tools.loggerr r(renumsrmodelsrrrrtypesrrErr r"r$rAr'rrrrs|B ,,-$0#& XX'   1  .  , G uur