o iS@sUdZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z ddlm Z ddl m Z mZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZddlZddl Zddl!Zddl"Zddl#Zdd l$m%Z%erydd l&m'Z'ed\ddZ(edddd]ddZ(edddd^ddZ( d_ddddd`ddZ(e ddGd d!d!Z)e*d"Z+d#e,d$<e Gd%d&d&Z-e ddGd'd(d(Z.e Gd)d*d*Z/e Gd+d,d,Z0dad.d/Z1dbd0d1Z2dcd2d3Z3ddd5d6Z4ded:d;Z5dbdZ?djdSdTZ@dkdVdWZAGdXdYdYejBZBeBeCeDdZE e ddGdZd[d[ZFdS)laFunctions that can be called inside of activities. Most of these functions use :py:mod:`contextvars` to obtain the current activity in context. This is already set before the start of the activity. Activities that make calls that do not automatically propagate the context, such as calls in another thread, should not use the calls herein unless the context is explicitly propagated. ) annotationsN)AbstractContextManagercontextmanager) dataclass)datetime timedelta) TYPE_CHECKINGAnyCallableIteratorListMappingMutableMappingNoReturnOptionalSequenceTupleTypeUnionoverload) CallableType)ClientfnrreturncCdSNrrrY/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/temporalio/activity.pydefn2sr Fnameno_thread_cancel_exceptionr" Optional[str]r#bool&Callable[[CallableType], CallableType]cCrrrr!rrrr 6r#dynamicr)cCrrrr(rrrr <r')r"r#r)Optional[CallableType]cs&dfdd }|dur||S|S)aeDecorator for activity functions. Activities can be async or non-async. Args: fn: The function to decorate. name: Name to use for the activity. Defaults to function ``__name__``. This cannot be set if dynamic is set. no_thread_cancel_exception: If set to true, an exception will not be raised in synchronous, threaded activities upon cancellation. dynamic: If true, this activity will be dynamic. Dynamic activities have to accept a single 'Sequence[RawValue]' parameter. This cannot be set to true if name is present. rrrcs"tj|s p |jndd|S)N) activity_namer#) _Definition_apply_to_callable__name__rr)r"r#rr decoratorXs zdefn..decoratorNrrrrr)rr"r#r)r0rr/rr Bs T)frozenc@seZdZUdZded<ded<ded<ded<d ed <d ed <d ed<d ed<ded<d ed<ded<ded<ded<ded<ded<ded<ded<ded<ded< d#d d!Zd"S)$InfozcInformation about the running activity. Retrieved inside an activity via :py:func:`info`. str activity_id activity_typeintattemptrcurrent_attempt_scheduled_timez Sequence[Any]heartbeat_detailszOptional[timedelta]heartbeat_timeoutr%is_localschedule_to_close_timeoutscheduled_timestart_to_close_timeout started_time task_queuebytes task_token workflow_idworkflow_namespaceworkflow_run_id workflow_typeztemporalio.common.Prioritypriorityz'Optional[temporalio.common.RetryPolicy] retry_policyrMapping[str, Any]c Cs&|j|j|j|j|j|j|j|jdS)N)r5r6r8 namespacerArDrFrG)r5r6r8rErArDrFrGselfrrr_logger_detailsszInfo._logger_detailsNrrJ)r. __module__ __qualname____doc____annotations__rNrrrrr3fs. r3activityz contextvars.ContextVar[_Context]_current_contextc@seZdZUdZded<dS)"_ActivityCancellationDetailsHolderN%Optional[ActivityCancellationDetails]details)r.rPrQrXrSrrrrrVs rVc@sheZdZUdZdZded<dZded<dZded<dZded<dZ ded<dZ ded <e dd dZ dS)ActivityCancellationDetailszsProvides the reasons for the activity's cancellation. Cancellation details are set once and do not change once set.Fr% not_foundcancel_requestedpausedreset timed_outworker_shutdownprotoAtemporalio.bridge.proto.activity_task.ActivityCancellationDetailsrcCs t|j|j|j|j|j|jdS)N)rZr[r\r^r_r])rY is_not_found is_cancelled is_paused is_timed_outis_worker_shutdownis_reset)r`rrr _from_protosz'ActivityCancellationDetails._from_protoN)r`rarrY) r.rPrQrRrZrSr[r\r]r^r_ staticmethodrhrrrrrYs       rYc@seZdZUded<ded<ded<ded<ded <d ed <d ed <ded<ded<dZded<dZded<dZd ed<ed,ddZed-ddZ ed.d!d"Z e d/d$d%Z e d0d'd(Z e d1d*d+ZdS)2_ContextzCallable[[], Info]infozOptional[Callable[..., None]] heartbeat_CompositeEventcancelled_eventworker_shutdown_eventz.Optional[Callable[[], AbstractContextManager]]shield_thread_cancel_exceptionzYUnion[Type[temporalio.converter.PayloadConverter], temporalio.converter.PayloadConverter]#payload_converter_class_or_instancez'Optional[temporalio.common.MetricMeter]runtime_metric_meterzOptional[Client]clientrVcancellation_detailsNOptional[Mapping[str, Any]]rNz/Optional[temporalio.converter.PayloadConverter]_payload_converter _metric_meterrcCstd}|s td|S)NzNot in activity context)rUget RuntimeErrorcontextrrrcurrents z_Context.currentr{contextvars.TokencCs t|Sr)rUsetrzrrrr~s z _Context.settokenNonecCst|dSr)rUr])rrrrr]sz_Context.resetrJcCs|jdur ||_|jSr)rNrkrLrrrlogger_detailss z_Context.logger_details%temporalio.converter.PayloadConvertercCs4|jst|jtjjr|j|_|jS||_|jSr)rv isinstancerq temporalio converterPayloadConverterrLrrrpayload_converters z_Context.payload_convertertemporalio.common.MetricMetercCs>|jstd|js|}|j|j|j|jd|_|jS)NzMMetrics meter not available in non-threaded sync activities like mulitprocess)rKrAr6)rrryrwrkwith_additional_attributesrErAr6)rMrkrrr metric_metersz_Context.metric_meter)rrj)r{rjrr})rr}rrrOrrrr)r.rPrQrSrNrvrwrir|r~r]propertyrrrrrrrrjs2          rjc@sHeZdZUded<ded<dddZdd d Zdd d ZddddZdS)rmzOptional[threading.Event] thread_eventzOptional[asyncio.Event] async_eventrrcCs0|jstd|j|jr|jdSdSNz Missing event)rryr~rrLrrrr~ s  z_CompositeEvent.setr%cCs|jstd|jSr)rryis_setrLrrrrs z_CompositeEvent.is_setcs$|jstd|jIdHdS)Nznot in async activity)rrywaitrLrrrrsz_CompositeEvent.waitNtimeoutOptional[float]cCs|jstd|j|dSr)rryr)rMrrrr wait_syncsz_CompositeEvent.wait_syncrrrr%r)rrrr)r.rPrQrSr~rrrrrrrrms    rmrcCstj}|s td|S)aReturn a Temporal Client for use in the current activity. The client is only available in `async def` activities. In tests it is not available automatically, but you can pass a client when creating a :py:class:`temporalio.testing.ActivityEnvironment`. Returns: :py:class:`temporalio.client.Client` for use in the current activity. Raises: RuntimeError: When the client is not available. zNo client available. The client is only available in `async def` activities; not in `def` activities. In tests you can pass a client when creating ActivityEnvironment.)rjr|rsry)rsrrrrs s rscCstdduS)zsWhether the current code is inside an activity. Returns: True if in an activity, False otherwise. N)rUrxrrrr in_activity8srcCs tS)zCurrent activity's info. Returns: Info for the currently running activity. Raises: RuntimeError: When not in an activity. )rjr|rkrrrrrkAs rkrWcCs tjjS)zcCancellation details of the current activity, if any. Once set, cancellation details do not change.)rjr|rtrXrrrrrtMs rtrXr rcGs"tj}|s td||dS)zkSend a heartbeat for the current activity. Raises: RuntimeError: When not in an activity. z1Can only execute heartbeat after interceptor initN)rjr|rlry)rX heartbeat_fnrrrrlRs  rlcCtjS)zWhether a cancellation was ever requested on this activity. Returns: True if the activity has had a cancellation request, False otherwise. Raises: RuntimeError: When not in an activity. )rjr|rnrrrrrrc^ rcIterator[None]ccsLtj}|s dVdS| dVWddS1swYdS)aContext manager for synchronous multithreaded activities to delay cancellation exceptions. By default, synchronous multithreaded activities have an exception thrown inside when cancellation occurs. Code within a "with" block of this context manager will delay that throwing until the end. Even if the block returns a value or throws its own exception, if a cancellation exception is pending, it is thrown instead. Therefore users are encouraged to not throw out of this block and can surround this with a try/except if they wish to catch a cancellation. This properly supports nested calls and will only throw after the last one. This just runs the blocks with no extra effects for async activities or synchronous multiprocess/other activities. Raises: temporalio.exceptions.CancelledError: If a cancellation occurs anytime during this block and this is not nested in another shield block. N)rjr|rp)shield_contextrrrrpjs  "rpctjIdHdS)zAsynchronously wait for this activity to get a cancellation request. Raises: RuntimeError: When not in an async activity. N)rjr|rnrrrrrwait_for_cancelledrr!Optional[Union[timedelta, float]]cC,tjt|tr|dS|dS)a'Synchronously block while waiting for a cancellation request on this activity. This is essentially a wrapper around :py:meth:`threading.Event.wait`. Args: timeout: Max amount of time to wait for cancellation. Raises: RuntimeError: When not in an activity. N)rjr|rnrrr total_secondsrrrrwait_for_cancelled_syncs rcCr)zWhether shutdown has been invoked on the worker. Returns: True if shutdown has been called on the worker, False otherwise. Raises: RuntimeError: When not in an activity. )rjr|rorrrrrrfrrfcr)zAsynchronously wait for shutdown to be called on the worker. Raises: RuntimeError: When not in an async activity. N)rjr|rorrrrrwait_for_worker_shutdownrrcCr)aFSynchronously block while waiting for shutdown to be called on the worker. This is essentially a wrapper around :py:meth:`threading.Event.wait`. Args: timeout: Max amount of time to wait for shutdown to be called on the worker. Raises: RuntimeError: When not in an activity. N)rjr|rorrrrrrrrwait_for_worker_shutdown_syncs rrcCst)zPRaise an error that says the activity will be completed asynchronously. )_CompleteAsyncErrorrrrrraise_complete_asyncsrc@s eZdZdS)rN)r.rPrQrrrrrsrrcC tjS)zGet the payload converter for the current activity. The returned converter has :py:class:`temporalio.converter.ActivitySerializationContext` set. This is often used for dynamic activities to convert payloads. )rjr|rrrrrrs rrcCr)aGet the metric meter for the current activity. .. warning:: This is only available in async or synchronous threaded activities. An error is raised on non-thread-based sync activities when trying to access this. Returns: Current metric meter for this activity for recording metrics. Raises: RuntimeError: When not in an activity or in a non-thread-based synchronous activity. )rjr|rrrrrrs rcs:eZdZdZdfdd ZdddZedddZZS) LoggerAdaptera=Adapter that adds details to the log about the running activity. Attributes: activity_info_on_message: Boolean for whether a string representation of a dict of some activity info will be appended to each message. Default is True. activity_info_on_extra: Boolean for whether a ``temporal_activity`` dictionary value will be added to the ``extra`` dictionary with some activity info, making it present on the ``LogRecord.__dict__`` for use by others. Default is True. full_activity_info_on_extra: Boolean for whether an ``activity_info`` value will be added to the ``extra`` dictionary with the entire activity info, making it present on the ``LogRecord.__dict__`` for use by others. Default is False. loggerlogging.Loggerextrarurrcs(t||pid|_d|_d|_dS)zCreate the logger adapter.TFN)super__init__activity_info_on_messageactivity_info_on_extrafull_activity_info_on_extra)rMrr __class__rrrs zLoggerAdapter.__init__msgr kwargsMutableMapping[str, Any]$Tuple[Any, MutableMapping[str, Any]]cCs|js |js |jrEtd}|rE|jr|d|jd}|jr0|ddp&i}|j|d<||d<|jrE|ddp:i}||d<||d<||fS)z!Override to add activity details.Nz ()rtemporal_activity activity_info)rrrrUrxrrk)rMrrr{rrrrprocess s&   zLoggerAdapter.processcCs|jS)zYUnderlying logger usable for actions such as adding handlers/formatters. )rrLrrr base_logger$szLoggerAdapter.base_logger)rrrrurr)rr rrrr)rr) r.rPrQrRrrrr __classcell__rrrrrs  rc@seZdZUded<ded<ded<ded<dZd ed <dZd ed <edddZedddZedddddZ dddZ dS)r,r$r"r rr%is_asyncr#NzOptional[List[Type]] arg_typeszOptional[Type]ret_typerOptional[_Definition]cCs(t|dd}t|trtj||d}|S)N__temporal_activity_definitionr)getattrrr, dataclassesreplace)rr rrr from_callable:s  z_Definition.from_callablecCs.t|}|r |St|dd}td|d)Nr.z z Activity z: missing attributes, was it decorated with @activity.defn?)r,rr TypeError)rretfn_namerrrmust_from_callableEs   z_Definition.must_from_callableF)r#r+rc Cst|dr tdt|stdt|}|jD] }|jtj j kr(tdqt |dt ||t |p9t |j|ddS)Nrz-Function already contains activity definitionzActivity is not callablez+Activity cannot have keyword-only arguments)r"rrr#)hasattr ValueErrorcallablerinspect signature parametersvalueskind Parameter KEYWORD_ONLYsetattrr,iscoroutinefunction__call__)rr+r#sigparamrrrr-Os(    z_Definition._apply_to_callablecCs|jdur@|jdurB|jdu}tj|j\}}|r0|r,t|dks,|dttjj kr0t dt |d|t |d|dSdSdS)NrrzJDynamic activity must accept a single Sequence[temporalio.common.RawValue]rr) rrr"rcommon_type_hints_from_funcrlenrRawValuerobject __setattr__)rMr)rrrrr __post_init__ps  z_Definition.__post_init__)rr rr)rr rr,)rr r+r$r#r%rrr) r.rPrQrSrrrirrr-rrrrrr,0s       r,r1)r"r$r#r%rr&)r#r%r)r%rr&r)rr*r"r$r#r%r)r%)rrr)rr3)rrW)rXr rr)rrr)rrrr)rrrr)GrR __future__rasyncio contextvarsrrlogging threading contextlibrrrrrtypingrr r r r r rrrrrrrrtemporalio.bridgertemporalio.bridge.proto%temporalio.bridge.proto.activity_tasktemporalio.commontemporalio.convertertypesrtemporalio.clientrr r3 ContextVarrUrSrVrYrjrmrsrrkrtrlrcrprrrfrrr BaseExceptionrrrr getLoggerr.rr,rrrrs  @   $.K           ;