)Jf9dZddlZddlZddlmZddlmZddlmZmZej e Z dZ dZ dZGd d eZdS) z^Callback management class, common area for keeping track of all callbacks in the Pika stack. N)frame) amqp_object)xrange canonical_strc t|tjr|jSn#t$rYnwxYwt |t jr |jjSt |tjr|jSt|S)zWill take Frame objects, classes, etc and attempt to return a valid string identifier for them. :param pika.amqp_object.AMQPObject|pika.frame.Frame|int|str value: The value to sanitize :rtype: str ) issubclassr AMQPObjectNAME TypeError isinstancerMethodmethodr)values P/home/alex/cs2snipeproduction/venv/lib/python3.11/site-packages/pika/callback.py name_or_valuers e[3 4 4 :        %&&!|  %/00z   s $ 11cFtjfd}|S)z9Automatically call name_or_value on the prefix passed in.ct|}d}d|vrt|d|d<n3t|dz |krt||||<|dz }d|vrt|d|d<n.t|dz |krt||||<t|i|S)Nprefixkey)listrlentuple)argskwargsoffsetfunctions rwrapperz sanitize_prefix..wrapper.sDzz v  ,VH-=>>F8   YY]f $ $(f66DL aKF F??)&-88F5MM YY]f $ $(f66DLxt//// functoolswrapsrrs` rsanitize_prefixr$+s:_X 0 0 0 0 0 NrcFtjfd}|S)zcAutomatically return false if the key or prefix is not in the callbacks for the instance. c*d}d|vrt|d}nt||}|dz }d|vrt|d}nt||}||djvs||dj|vrdS|i|S)NrrrrF)r_stack)rrrrrrs rrz)check_for_prefix_and_key..wrapperGs v  "6(#344FF"4<00F aKF F??u ..CCV --C a ' '3d1gnV6L+L+L5x((((rr r#s` rcheck_for_prefix_and_keyr(As: _X))))), Nrc@eZdZdZdZdZdZdZdZdZ dZ e dd Z d Z e d Ze dZe edZe eddZe edZdZdZddZedZedZdZdZd S)CallbackManagerzCallbackManager is a global callback system designed to be a single place where Pika can manage callbacks and process them. It should be referenced by the CallbackManager.instance() method instead of constructing new instances of it. calls argumentsz$Duplicate callback found for "%s:%s"callbackone_shotonlyc,t|_dS)z)Create an instance of the CallbackManagerN)dictr'selfs r__init__zCallbackManager.__init__osff rTNc||jvrt|j|<||j|vrt|j||<|j||D]}||j|kr||j|krx||j|krg||jdur1||jxxdz cc<t d|n!t |j ||||fcS| ||||}|j|| |t d|||fS)aAdd a callback to the stack for the specified key. If the call is specified as one_shot, it will be removed after being fired The prefix is usually the channel number but the class is generic and prefix and key may be any value. If you pass in only_caller CallbackManager will restrict processing of the callback to only the calling function/object that you specify. :param str|int prefix: Categorize the callback :param str|dict key: The key for the callback :param callable callback: The callback to call :param bool one_shot: Remove this callback after it is called :param object only_caller: Only allow one_caller value to call the event that fires the callback. :param dict arguments: Arguments to validate when processing :rtype: tuple(prefix, key) Trz*Incremented callback reference counter: %rz Added: %r)r'r1rCALLBACK ARGUMENTS ONLY_CALLERONE_SHOTCALLSLOGGERdebugwarningDUPLICATE_WARNING_callback_dictappend)r3rrr-r. only_callerr, callback_dicts raddzCallbackManager.addssw6  $ $"&&&DK  dk&) ) )'+vvDK  $"[05 # #Mdm,88!$.1Y>>!$"23{BB /477!$*---2---LL!M!.0000NN4#963GGGs{"""++Hh ,577  FC '' 666 [-000s{rc`t|_tddS)z1Clear all the callbacks if there are any defined.zCallbacks clearedN)r1r'r;r<r2s rclearzCallbackManager.clears&ff  ()))))rc|td|||jvs |j|sdS|j|=dS)zRemove all callbacks from the stack by a prefix. Returns True if keys were there to be removed :param str or int prefix: The prefix for keeping track of callbacks with :rtype: bool zClearing out %r from the stackFT)r;r<r')r3rs rcleanupzCallbackManager.cleanupsF  5v>>>  $ $DK,? $5 K trcv||jvs||j|vrdSt|j||S)zReturn count of callbacks for a given prefix or key or None :param str|int prefix: Categorize the callback :param object|str|dict key: The key for the callback :rtype: None or int N)r'rr3rrs rpendingzCallbackManager.pendingsA$$C4;v3F,F,F44;v&s+,,,rc@td||||jvs||j|vrdSt}t|j||D]j}|||t|rD|||j||jr||||k|D]K}td||| ||i|)#t d|||xYwdS)aRun through and process all the callbacks for the specified keys. Caller should be specified at all times so that callbacks which require a specific function to call CallbackManager.process will not be processed. :param str|int prefix: Categorize the callback :param object|str|int key: The key for the callback :param object caller: Who is firing the event :param list args: Any optional arguments :param dict keywords: Optional keyword arguments :rtype: bool zProcessing %s:%sFzCalling %s for "%s:%s"zCalling %s for "%s:%s" failedT) r;r<r'r_should_process_callbackr@r6r9_use_one_shot_callback exception) r3rrcallerrkeywords callbacksrBr-s rprocesszCallbackManager.processsK  '555  $ $4;v3F(F(F5FF !$+f"5c":;; L LM,,]FDJJOO L  t}!=>>> /L//]KKK"  H LL18VS I I I $+(++++   !@(!'...ts 2C;; Dc(|rt}tt|j||ddD]^}|j|||dz }||j|kr/|||gr||dz _|D]Z} td||j||||j|||=K#t$rYWwxYw| ||dS)aRemove a callback from the stack by prefix, key and optionally the callback itself. If you only pass in prefix and key, all callbacks for that prefix and key will be removed. :param str or int prefix: The prefix for keeping track of callbacks with :param str key: The callback key :param callable callback_value: The method defined to call on callback :param dict arguments: Optional arguments to check :rtype: bool rrzRemoving callback #%i: %rT) rrrr'r6_arguments_matchr@r;r<KeyError_cleanup_callback_dict)r3rrcallback_valuer,offsets_to_removerrBs rremovezCallbackManager.removes?   $  T[%8%=!>!>2FF 9 9 $ F 3C 8! D !$-0NBB--mi[IIC%,,VaZ888+  LL!>>  Cyy~~S!111 A3GGGuu2trc|D]}t||s,tdt||dSt ||||kr,tdt||dSdS)zChecks an object to see if it has attributes that meet the expectation. :param object value: The object to evaluate :param dict expectation: The values to check against :rtype: bool z'%r does not have required attribute: %sFz Values in %s do not match for %sT)r`r;r<typegetattrres rr_z$CallbackManager._obj_arguments_matchbs  C5#&&  F!%[[#///uuuc""k#&666 ?e """uu7trc|||std||dS||jdup||jo||j|kS)zReturns True if the callback should be processed. :param dict callback_dict: The callback configuration :param object caller: Who is firing the event :param list args: Any optional arguments :rtype: bool z!Arguments do not match for %r, %rFN)rUr;r<r8)r3rBrOrs rrLz(CallbackManager._should_process_callbackwsy$$]D99  LLr6r9r8r4r$rCrErGrJr(rRrZr\rUr?rW staticmethodr^r_rLrMrrr*r*as EI>HHK  222_2h***   _  - -_ -##_#J_@11_1+++(( $ $ $ $  \ \(==="77777rr*)ror!loggingpikarr pika.compatrr getLoggerrlr;rr$r(objectr*rqrrrws--------  8 $ $   8,@v7v7v7v7v7fv7v7v7v7v7r