U e{j@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)SAMPProxyErrorcCsDddlm}|jsdSztdddWdStk r>YdSXdS)NrconfFzhttp://google.comg?)timeoutT)rZ use_internetr Exceptionrr Q/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/samp/utils.py internet_ons  rSAMPMsgReplierWrapper.cCs6d|kr(|dd\}}tt|||St||SdS)z% Get attributes recursively. rrN)splitgetattr_recursivegetattr)variable attributetop remainingr r r r$src@s$eZdZddZddZddZdS)_ServerProxyPoolMethodcCs||_||_dSN)_ServerProxyPoolMethod__proxies_ServerProxyPoolMethod__name)selfproxiesnamer r r __init__3sz_ServerProxyPoolMethod.__init__cCst|j|jd|SNr)rrrrrr r r __getattr__7sz"_ServerProxyPoolMethod.__getattr__c Osp|j}t||j}zFz|||}Wn2tjk rX}zt|j|j W5d}~XYnXW5|j|X|Sr) rgetrrputxmlrpcFaultr faultCode faultString)rargsZkwrdsproxyfunctionresponseexcr r r __call__:s  $z_ServerProxyPoolMethod.__call__N__name__ __module__ __qualname__rr"r.r r r r r/src@s(eZdZdZddZddZddZdS) ServerProxyPoolz= A thread-safe pool of `xmlrpc.ServerProxy` objects. cOs0t||_t|D]}|j|||qdSr)queueQueue_proxiesranger$)rsizeZ proxy_classr)keywordsir r r rKs  zServerProxyPool.__init__cCs t|j|Sr)rr6r!r r r r"PszServerProxyPool.__getattr__cCs8z|j}Wntjk r(Yq4YnX|dqdS)z;Shut down the proxy pool by closing all active connections.closeN)r6 get_nowaitr4Empty)rr*r r r shutdownTs  zServerProxyPool.shutdownN)r0r1r2__doc__rr"r>r r r r r3Fsr3c@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)rr@r r r rmszSAMPMsgReplierWrapper.__init__csfdd}|S)Nc stdks|ddkr"|nz2|}|rRjjj|dt|dWnTtk rt}tj |d| }jjj|dtd|idYnXdS)N)z samp.statusz samp.result)filetxt) get_num_argsr@ZhubZreplyZget_private_keyrr r traceback print_excgetvalue)r)resulterrrDfrr r wrapped_fqs&   z1SAMPMsgReplierWrapper.__call__..wrapped_fr )rrLrMr rKr r.pszSAMPMsgReplierWrapper.__call__N)r0r1r2r?rr.r r r r ras c@seZdZddZddZdS) _HubAsClientcCs ||_dSr)_handler)rhandlerr r r rsz_HubAsClient.__init__cCs t|j|Sr)_HubAsClientMethodrOr!r r r r"sz_HubAsClient.__getattr__N)r0r1r2rr"r r r r rNsrNc@s$eZdZddZddZddZdS)rQcCs||_||_dSr_HubAsClientMethod__send_HubAsClientMethod__name)rsendrr r r rsz_HubAsClientMethod.__init__cCst|j|jd|Sr )rQrSrTr!r r r r"sz_HubAsClientMethod.__getattr__cGs||j|SrrR)rr)r r r r.sz_HubAsClientMethod.__call__Nr/r r r r rQsrQcCs6t|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)inspectismethod__func____code__ co_argcount isfunction TypeError)rLr r r rEs   rE)r?rVr4rF xmlrpc.clientclientr%iorurllib.requestr constantsrerrorsrr__all__Z__doctest_skip__rrr3rrNrQrEr r r r s$       *