iA6dZddlZddlZddlmZddlmZmZddlm Z ddl m Z ddl m Z ejeZGdd e ZGd d ZGd d Zy)zEvent-driven architecture for ProjectX SDK v3.0.0. This module provides a unified event system for all SDK components, replacing scattered callback systems with a centralized event bus. N) defaultdict)Callable Coroutine)Enum)Any)WeakSetceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!y )! EventTypez+Unified event types for all SDK components.new_bar data_update quote_update trade_tickorderbook_updatemarket_depth_update order_placed order_filledorder_partial_fillorder_cancelledorder_rejected order_expiredorder_modifiedposition_openedposition_closedposition_updatedposition_pnl_updaterisk_limit_warningrisk_limit_exceededstop_loss_triggeredtake_profit_triggered connected disconnected reconnecting authenticatederrorwarningmemory_warningrate_limit_warninglatency_warningN)"__name__ __module__ __qualname____doc__NEW_BAR DATA_UPDATE QUOTE_UPDATE TRADE_TICKORDERBOOK_UPDATEMARKET_DEPTH_UPDATE ORDER_PLACED ORDER_FILLEDORDER_PARTIAL_FILLORDER_CANCELLEDORDER_REJECTED ORDER_EXPIREDORDER_MODIFIEDPOSITION_OPENEDPOSITION_CLOSEDPOSITION_UPDATEDPOSITION_PNL_UPDATERISK_LIMIT_WARNINGRISK_LIMIT_EXCEEDEDSTOP_LOSS_TRIGGEREDTAKE_PROFIT_TRIGGERED CONNECTED DISCONNECTED RECONNECTING AUTHENTICATEDERRORWARNINGMEMORY_WARNINGRATE_LIMIT_WARNINGLATENCY_WARNINGO/home/work/apex_v16/venv/lib/python3.12/site-packages/project_x_py/event_bus.pyr r s5GK!LJ)/"L!L-'O%N#M%N(O'O)/.//3I!L!L#M EG&N-'OrLr c2eZdZdZddeezdededzfdZy)EventzContainer for event data.Ntypedatasourcec|t|tr||_n t||_||_||_t jj|_ y#t$r ||_YHwxYw)zInitialize event. Args: type: Event type (EventType enum or string) data: Event payload data source: Optional source component name N) isinstancer rP ExceptionrQrRasyncioget_running_looptime timestamp)selfrPrQrRs rM__init__zEvent.__init__Csi dI &DI !%dO    11388:  !   !sA""A54A5N)r)r*r+r,r strrr[rKrLrMrOrO@s'#;Y_;C;t;rLrOc eZdZdZddZdeezdeege e e dffddfdZ dedeezdeege e e dffddfd Z deezdeege e e dffddfd Z deege e e dffddfd Z ddeezdzdeege e e dffdzddfd Z ddeezd e dedzddfdZdeege e e dffdeddfdZ ddeezdedzdefdZddeddfdZdeefdZddZddeezdzdefdZddZy)EventBusaUnified event system for all SDK components. Provides centralized event handling with support for: - Multiple handlers per event - One-time handlers - Wildcard event subscriptions - Async event emission - Weak references to prevent memory leaks returnNctt|_tt|_g|_t |_d|_g|_d|_ tt|_ y)zInitialize EventBus.FN) rlist _handlers_once_handlers_wildcard_handlersr _active_tasks_history_enabled_event_history_max_history_size_legacy_handlersrZs rMr[zEventBus.__init__fsj      WY:A!&+-!%   rLeventhandlerc<Kt|tr|n t|}tj|st d|j d|j |j|tjd|j d|jyw)zRegister handler for event type. Args: event: Event type to listen for handler: Async callable to handle events Handler  must be asynczRegistered handler  for N) rTr rViscoroutinefunction ValueErrorr)rdappendloggerdebugvaluerZrmrn event_types rMonz EventBus.ons) :U %@P **73x(8(8'9HI I z"))'2 *7+;+;*IeDTJz*002##J/557 $~~j9.Q'\A.z* $22:>3!w,A3##J/+.3s=A"E$ D7/D73A4E' D<2D<6!E E"E& ErQrRcKt|||}|jrX|jj|t |j|j kDr|jj dg}t|jtr|j|jj|jg|jj|jg}|j||rXg|j|j<n>|j|jjt|jg|j|j |rxg}|D]S}t#j$|j'||} |j(j+| |j| Ut#j,|ddid{yy7w)zEmit event to all registered handlers. Args: event: Event type to emit data: Event payload data source: Optional source component name rreturn_exceptionsTN)rOrhrirulenrjpoprTrPr extendrdgetrerkr]rfrV create_task_execute_handlerrgaddgather) rZrmrQrR event_objhandlers once_handlerstasksrntasks rMemitz EventBus.emits%v.      & &y 14&&'$*@*@@##''*HJ inni 0 OODNN..y~~rB C //33INNBGM OOM *68##INN3 OOD1155c)..6I2N O //0 -/E# #**4+@+@)+TU""&&t, T" # ..%@4@ @ @  AsG#G.%G,&G.c K ||d{y7#t$r}t|jtr|jjnt |j}t jd|jd|d||jtjk7rG|jtj||jt |ddd{7Yd}~yYd}~yd}~wwxYww)zExecute event handler with error handling. Args: handler: Async callable to execute event: Event object to pass to handler NzError in event handler rrz: )original_eventrnr$r_)rR) rUrTrPr rxr]rvr$r)rFr)rZrnrmeevent_type_strs rMrzEventBus._execute_handlers %.  ejj)4   _  LL)'*:*:);5@PPRSTRUV zzY__,iiOO*8#*#3#3!$Q & - sCC> C> C;CC6&C)'C6, C>6C;;C>timeoutcXKt|tr|n t|}tjdtddffd }|j ||d{ tj |d{S7$7#t$r|j||d{7wxYww)aWait for specific event to occur. Args: event: Event type to wait for timeout: Optional timeout in seconds Returns: Event object when received Raises: asyncio.TimeoutError: If timeout expires evtr`NcRKjsj|yywr\)done set_result)rfutures rMrnz"EventBus.wait_for..handlerKs#;;=!!#&!s$')r) rTr rVFuturerOrwait_for TimeoutErrorr)rZrmrrzrnrs @rMrzEventBus.wait_for9s) :U %@P (/(8 'u ' 'ii G,,,  ))&'BB B -C ((:w/ / /  sHAB*A=B*B8A?9B<B*?BB'B" B''B*max_sizec.d|_||_g|_y)zsEnable event history for debugging. Args: max_size: Maximum number of events to store TN)rhrjri)rZrs rMenable_historyzEventBus.enable_historyXs !%!) rLc6|jjS)zkGet event history. Returns: List of recent events (empty if history disabled) )ricopyrls rM get_historyzEventBus.get_historybs ""''))rLc8|jjy)zClear event history.N)rirrls rM clear_historyzEventBus.clear_historyjs !!#rLc|td|jjD}|td|jjDz }|t |j z }|td|j jDz }|St|trgt |jj|g}|t |jj|gz }|t |j z }|St |j jt|g}|t |j z }|S)zGet number of registered handlers. Args: event: Event type to check (None for total) Returns: Number of handlers c32K|]}t|ywr\r.0rs rM z-EventBus.get_handler_count..xsN(H Nc32K|]}t|ywr\rrs rMrz-EventBus.get_handler_count..ysT8XTrc32K|]}t|ywr\rrs rMrz-EventBus.get_handler_count..{sV8XVr) sumrdvaluesrerrfrkrTr rr])rZrmtotalcounts rMget_handler_countzEventBus.get_handler_countns$ =Ndnn6K6K6MNNE STt7J7J7Q7Q7STT TE S001 1E SVt7L7L7S7S7UVV VEL%+DNN..ub9:T0044UB?@@T4455 D1155c%j"EFT4455 rLc`Kdtddffd }|j|d{y7w)a> Forward all events from this bus to the target bus. This sets up a wildcard handler that forwards all events to another EventBus, enabling event propagation from instrument-specific buses to the main suite bus. Args: target_bus: The EventBus to forward events to rmr`NcKj|j|j|jd{y7w)zForward event to target bus.N)rrPrQrR)rm target_buss rM forwarderz&EventBus.forward_to..forwarders)//%**ejj%,,G G Gs5A>A)rOr)rZrrs ` rM forward_tozEventBus.forward_tos1 H5 HT Hkk)$$$s #.,.)r`N)NNr\)rb)rr_r`N)r)r*r+r,r[r r]rrOrrr{rrrrrrfloatrintrrcrrrrrKrLrMr_r_[s/2V3V5'9S#t^#<<=V  V( > >3 >5'9S#t^#<<= >  > 3 5'9S#t^#<<=    . H3T>)B BC H  H )-GK#3%#5'9S#t^#<<=D#  #LGK,A_,A,/,A9)B BCLQ B?C_/4t| >!s!d!*T%[*$y3'=4%rLr_)r,rVlogging collectionsrcollections.abcrrenumrtypingrweakrefr getLoggerr)rvr rOr_rKrLrMrsV #/   8 $+(+(\;;6|%|%rL