a ߙfb@sdZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZdd Zd gZd gZd d ZGdddZGdddZGdd d ZGdddZGdddZddZdS)z Utility functions and classes N)StringIO)urlopen)SAMP_STATUS_ERROR)SAMPProxyErrorcCsBddlm}|jsdSztdddWdSty<YdS0dS)NrconfFzhttp://google.comg?)ZtimeoutT)rZ use_internetr Exceptionrr 1lib/python3.9/site-packages/astropy/samp/utils.py internet_ons   r SAMPMsgReplierWrapper.cCs6d|vr(|dd\}}tt|||St||SdS)z% Get attributes recursively. rrN)splitgetattr_recursivegetattr)ZvariableZ attributetopZ remainingr r r r#src@s$eZdZddZddZddZdS)_ServerProxyPoolMethodcCs||_||_dSN)_ServerProxyPoolMethod__proxies_ServerProxyPoolMethod__name)selfZproxiesnamer r r __init__3sz_ServerProxyPoolMethod.__init__cCst|j|jd|SNr)rrrrrr r r __getattr__7sz"_ServerProxyPoolMethod.__getattr__c Os|j}t||j}zXz||i|}Wn4tjy^}zt|j|jWYd}~n d}~00W|j |n|j |0|Sr) rgetrrxmlrpcZFaultrZ faultCodeZ faultStringput)rargsZkwrdsproxyZfunctionZresponseexcr r r __call__:s  &z_ServerProxyPoolMethod.__call__N__name__ __module__ __qualname__rrr$r r r r r.src@s(eZdZdZddZddZddZdS) ServerProxyPoolz= A thread-safe pool of `xmlrpc.ServerProxy` objects. cOs4t||_t|D]}|j||i|qdSr)queueZQueue_proxiesranger )rsizeZ proxy_classr!keywordsir r r rKs  zServerProxyPool.__init__cCs t|j|Sr)rr+rr r r rQszServerProxyPool.__getattr__cCs6z|j}Wntjy&Yq2Yn0|dqdS)z;Shut down the proxy pool by closing all active connections.closeN)r+Z get_nowaitr*ZEmpty)rr"r r r shutdownUs  zServerProxyPool.shutdownN)r&r'r(__doc__rrr1r r r r r)Fsr)c@s eZdZdZddZddZdS)ra Function decorator that allows to automatically grab errors and returned maps (if any) from a function bound to a SAMP call (or notify). Parameters ---------- cli : :class:`~astropy.samp.SAMPIntegratedClient` or :class:`~astropy.samp.SAMPClient` SAMP client instance. Decorator initialization, accepting the instance of the client that receives the call or notification. cCs ||_dSr)cli)rr3r r r rnszSAMPMsgReplierWrapper.__init__csfdd}|S)Nc stdks|ddur"|nz2|}|rRjjj|dt|dWnRtyt}tj |d| }jjj|dtd|idYn0dS)N)z samp.statusz samp.result)filetxt) get_num_argsr3ZhubZreplyZget_private_keyrr r traceback print_excgetvalue)r!resulterrr7frr r wrapped_fss&   z1SAMPMsgReplierWrapper.__call__..wrapped_fr )rr?r@r r>r r$qszSAMPMsgReplierWrapper.__call__N)r&r'r(r2rr$r r r r rbs c@seZdZddZddZdS) _HubAsClientcCs ||_dSr)_handler)rZhandlerr r r rsz_HubAsClient.__init__cCs t|j|Sr)_HubAsClientMethodrBrr r r rsz_HubAsClient.__getattr__N)r&r'r(rrr r r r rAsrAc@s$eZdZddZddZddZdS)rCcCs||_||_dSr_HubAsClientMethod__send_HubAsClientMethod__name)rsendrr r r rsz_HubAsClientMethod.__init__cCst|j|jd|Sr)rCrErFrr r r rsz_HubAsClientMethod.__getattr__cGs||j|SrrD)rr!r r r r$sz_HubAsClientMethod.__call__Nr%r r r r rCsrCcCs6t|r|jjjdSt|r*|jjStddS)zW Find the number of arguments a function or method takes (excluding ``self``). rz"f should be a function or a methodN)inspectZismethod__func____code__ co_argcountZ isfunction TypeError)r?r r r r8s   r8)r2r*rHr9iorZ xmlrpc.clientZclientrZurllib.requestrZ constantsrerrorsrr __all__Z__doctest_skip__rrr)rrArCr8r r r r s$       *