[c@`sddlmZmZmZmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZmZddl mZddlmZddlmZdd lmZdd lmZdd lmZmZmZdd lmZmZm Z dd l!m"Z"m#Z#ddl$m%Z%ddl&m'Z'm(Z(ddgZ)ddgZ*de+fdYZ,de+fdYZ-dS(i(tabsolute_importtdivisiontprint_functiontunicode_literalsNi(tqueuetrange(t xmlrpc_client(t urlunparse(tlogi(tSAMP_STATUS_OK(t__profile_version__(t SAMPWarningt SAMPHubErrortSAMPProxyError(t internet_ontServerProxyPoolt _HubAsClient(t read_lockfiletcreate_lock_file(tThreadingXMLRPCServer(tWebProfileXMLRPCServertweb_profile_text_dialogu SAMPHubServeruWebProfileDialogu.uSAMPHubServer.*t SAMPHubServerc B`seZdZdEdEddEddddedEddd ZedZdZd Z d Z d Z dEdEdEdEd Z dEd Z dZdZdZdZedZedZdZdZdZdEdZedZdZdZdZdZdZdZdZ d Z!d!Z"d"Z#d#Z$d$Z%d%Z&d&Z'd'Z(d(Z)d)Z*d*Z+d+Z,d,Z-e.d-Z/d.Z0d/Z1d0Z2d1Z3d2Z4d3Z5d4Z6d5Z7d6Z8d7Z9d8Z:d9Z;d:Z<d;Z=d<Z>dEd=Z?d>Z@d?ZAd@ZBdFdAdBZCdCZDdDZERS(Gu SAMP Hub Server. Parameters ---------- secret : str, optional The secret code to use for the SAMP lockfile. If none is is specified, the :func:`uuid.uuid1` function is used to generate one. addr : str, optional Listening address (or IP). This defaults to 127.0.0.1 if the internet is not reachable, otherwise it defaults to the host name. port : int, optional Listening XML-RPC server socket port. If left set to 0 (the default), the operating system will select a free port. lockfile : str, optional Custom lockfile name. timeout : int, optional Hub inactivity timeout. If ``timeout > 0`` then the Hub automatically stops after an inactivity period longer than ``timeout`` seconds. By default ``timeout`` is set to 0 (Hub never expires). client_timeout : int, optional Client inactivity timeout. If ``client_timeout > 0`` then the Hub automatically unregisters the clients which result inactive for a period longer than ``client_timeout`` seconds. By default ``client_timeout`` is set to 0 (clients never expire). mode : str, optional Defines the Hub running mode. If ``mode`` is ``'single'`` then the Hub runs using the standard ``.samp`` lock-file, having a single instance for user desktop session. Otherwise, if ``mode`` is ``'multiple'``, then the Hub runs using a non-standard lock-file, placed in ``.samp-1`` directory, of the form ``samp-hub-``, where ```` is a unique UUID assigned to the hub. label : str, optional A string used to label the Hub with a human readable name. This string is written in the lock-file assigned to the ``hub.label`` token. web_profile : bool, optional Enables or disables the Web Profile support. web_profile_dialog : class, optional Allows a class instance to be specified using ``web_profile_dialog`` to replace the terminal-based message with e.g. a GUI pop-up. Two `queue.Queue` instances will be added to the instance as attributes ``queue_request`` and ``queue_result``. When a request is received via the ``queue_request`` queue, the pop-up should be displayed, and a value of `True` or `False` should be added to ``queue_result`` depending on whether the user accepted or refused the connection. web_port : int, optional The port to use for web SAMP. This should not be changed except for testing purposes, since web SAMP should always use port 21012. pool_size : int, optional The number of socket connections opened to communicate with the clients. iusingleuiRic C`sttj|_t|_||_d|_||_ ||_ ||_ ||_ ||_ ||_| |_| |_| |_| |_d|_i|_d|_d|_d|_d|_tr$y8tj|_tj|j p|j|j pdWq$tjk r d|_q$Xntj|_ d|_!d|_"d|_#g|_$d|_%i|_&d|_'||_(|j)|_*d|_+i|_,i|_-i|_.i|_/i|_0i|_1d|_2dS(Nu 127.0.0.1iui(3tstrtuuidtuuid1t_idtFalset _is_runningt_customlockfilenametNonet _lockfilet_addrt_portt_modet_labelt_timeoutt_client_timeoutt _pool_sizet _web_profilet_web_profile_dialogt _web_portt_web_profile_servert_web_profile_callbackst_web_profile_requests_queuet_web_profile_requests_resultt_web_profile_requests_semaphoret _host_nameRtsockettgetfqdnt getaddrinfoterrort threadingtLockt _thread_lockt _thread_runt_thread_hub_timeoutt_thread_client_timeoutt_launched_threadst_last_activity_timet_client_activity_timet_hub_msg_id_countert_hub_secret_code_customizedt_create_secret_codet _hub_secrett_hub_public_idt _private_keyst _metadatat _mtype2idst _id2mtypest_xmlrpc_endpointst_sync_msg_ids_heapt_client_id_counter( tselftsecrettaddrtporttlockfilettimeouttclient_timeouttmodetlabelt web_profiletweb_profile_dialogtweb_portt pool_size((s/lib/python2.7/site-packages/astropy/samp/hub.pyt__init__dsZ                                   cC`s|jS(u$ The unique hub ID. (R(RI((s/lib/python2.7/site-packages/astropy/samp/hub.pytidscC`s4|j|jd|j|jd|j|jd|j|jd|j|jd|j|jd|j|jd|j|jd|j|j d |j|j d |j|j d |j|j d |j|j d |j|jd|j|jd|j|jddS(Nu samp.hub.pingusamp.hub.setXmlrpcCallbackusamp.hub.registerusamp.hub.unregisterusamp.hub.declareMetadatausamp.hub.getMetadatausamp.hub.declareSubscriptionsusamp.hub.getSubscriptionsusamp.hub.getRegisteredClientsusamp.hub.getSubscribedClientsusamp.hub.notifyusamp.hub.notifyAllu samp.hub.callusamp.hub.callAllusamp.hub.callAndWaitusamp.hub.reply(tregister_functiont_pingt_set_xmlrpc_callbackt _registert _unregistert_declare_metadatat _get_metadatat_declare_subscriptionst_get_subscriptionst_get_registered_clientst_get_subscribed_clientst_notifyt _notify_allt_callt _call_allt_call_and_waitt_reply(RItserver((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_register_standard_apis cC`sG|j|jd|j|jd|j|jd|j|jd|j|jd|j|jd|j|jd|j|jd|j|j d |j|j d |j|j d |j|j d |j|j d |j|jd|j|jd|j|jd|j|jddS(Nusamp.webhub.pingusamp.webhub.unregisterusamp.webhub.declareMetadatausamp.webhub.getMetadatau samp.webhub.declareSubscriptionsusamp.webhub.getSubscriptionsu samp.webhub.getRegisteredClientsu samp.webhub.getSubscribedClientsusamp.webhub.notifyusamp.webhub.notifyAllusamp.webhub.callusamp.webhub.callAllusamp.webhub.callAndWaitusamp.webhub.replyusamp.webhub.registeru!samp.webhub.allowReverseCallbacksusamp.webhub.pullCallbacks(RXRYR\R]R^R_R`RaRbRcRdReRfRgRht_web_profile_registert"_web_profile_allowReverseCallbackst_web_profile_pullCallbacks(RIRi((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_register_web_profile_apis"cC`st|jp|j|jpdftdtdt|_d}|jjj d|_dj |jpm|j|j}t ||ddddf|_ |jj |j|jdS(Nit logRequestst allow_noneuhttpiu{0}:{1}u(RR R/R!RRtTruet_serverR0t getsocknametformatRt_urlt register_introspection_functionsRj(RItprotRK((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_start_standard_servers!! cC`s/tjd|_tjd|_tjd|_|jdk rf|j|j_|j|j_nynt d|j ft dt dt |_|jjjd|_ |jj|j|jt jdWnTtjk r*t jdj|j tt |_d|_d|_d|_nXdS(Niu localhostRoRpu0Hub set to run with Web Profile support enabled.uLPort {0} already in use. Impossible to run the Hub with Web Profile support.(RtQueueR,R-R.R(Rt queue_requestt queue_resultRR)RRRqR*R0RsRvRntinfoR3twarningRtR R'(RI((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_start_web_profile_servers,       c C`sg}x-|jD]"}|js|j|qqWx|D]}|jj|q=Wtjd|d|d|d|}|j|jj|dS(Ntgroupttargettnametargs(R:tis_alivetappendtremoveR4tThreadtstart(RIRRRRRtt((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_launch_threads  $ cC`s(x!|jD]}|jd|q WdS(NRN(R:tjoin(RIRNR((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_join_launched_threads*scC`s|jdkrdStj}x|jrtjdtj}||dkr"|jK|jdk r||j|jkrtjdt |j dSnWdQX|}q"q"WdS(Nig?g?u&Timeout expired, Hub is shutting down!( R$ttimeRtsleepR6R;RtwarningstwarnR tstop(RItlasttnow((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_timeout_test_hub.s        cC`s|jdkrdStj}x|jrtjdtj}||dkr"xv|jjD]e}||j||jkrd||jkrdtjdj |t |j ||j |qdqdW|}q"q"WdS(Nig?g?uClient {} timeout expired!( R%RRRR<tkeyst_hub_private_keyRRRtR t_notify_disconnectionR\(RIRRt private_key((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_timeout_test_clientAs       cC`sh|dkr|j|S|dkr2|j|S|dkrK|j|S|dkrd|j|SdS(Nusamp.client.receiveCallusamp.client.receiveNotificationusamp.client.receiveResponseu samp.app.ping(t _receive_callt_receive_notificationt_receive_responseRY(RItmethodR((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_hub_as_client_request_handlerUs       cC`sidd6|jd6dd6dd6|jdd 6}|j|j}|d |_|d |_|j|j|j|j|j||j|jiid 6id 6dS(NuAstropy SAMP Hubu samp.nameusamp.description.textuThe Astropy Collaborationu author.nameu)http://docs.astropy.org/en/stable/vo/sampusamp.documentation.urlu /samp/iconu samp.icon.urlu samp.self-idusamp.private-keyu samp.app.pingux-samp.query.by-meta( R#RuR[R@RARRZR]R_(RIt hub_metadatatresult((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_setup_hub_as_client_s      c C`s|jrtdn|jdk r6tdn|jrL|jn|jtd|jd|j d|j d|j |_|j |j |jtjd|r|jr|jjd|_ndS( u< Start the current SAMP Hub instance and create the lock file. Hub start-up can be blocking or non blocking depending on the ``wait`` parameter. Parameters ---------- wait : bool If `True` then the Hub process is joined with the caller, blocking the code flow. Usually `True` option is used to run a stand-alone Hub in an executable script. If `False` (default), then the Hub process runs in a separated thread. `False` is usually used in a Python shell. uHub is already runningu&Hub is not running but lockfile is sett lockfilenameRPthub_idt hub_paramsu Hub startedN(RR RRR'R~RxRRR"RWtparamst_update_last_activity_timeRt_start_threadsRR|R7R(RItwait((s/lib/python2.7/site-packages/astropy/samp/hub.pyRps"         cC`sZi}|j|d<|j|dRRRR(RI((s/lib/python2.7/site-packages/astropy/samp/hub.pyR?scC`s|js dStjd|jt|_|jrtjj|jrt |j}|d|j krtj |jqnd|_|j ddd|_|j|_ d|_i|_i|_i|_i|_i|_d|_tjddS( uN Stop the current SAMP Hub instance and delete the lock file. NuHub is stopping...u samp.secretRNg$@iuu Hub stopped.(RRR|t_notify_shutdownRRtostpathtisfileRR@RRt_join_all_threadsR=R?RARCRBRDRERFR;(RIt lockfiledict((s/lib/python2.7/site-packages/astropy/samp/hub.pyRs*             cC`stj}|j|k rL|jjd||jjsLd|_qLn|jdk r|j|k r|jjd||jjsd|_qn|jdk r|j|k r|jjd||jjsd|_qn|jd|dS(NRN( R4tcurrent_threadR7RRRR8R9R(RIRNR((s/lib/python2.7/site-packages/astropy/samp/hub.pyRs cC`s|jS(uReturn an information concerning the Hub running status. Returns ------- running : bool Is the hub running? (R(RI((s/lib/python2.7/site-packages/astropy/samp/hub.pyt is_running s cC`szxG|jrIy)tj|jjgggdd}Wn/tjk rf}tjdj|tnX|r}|jj n|j r|j dkry|j j}Wntjk rqXt||jny)tj|jjgggdd}Wn/tjk r,}tjdj|tqFX|rF|jj qFqqW|jj|jdk rv|jjndS(Ng{Gz?iu-Call to select() in SAMPHubServer failed: {0}(RtselectRrR0R3RRRtR thandle_requestR'R(RR,t get_nowaitRtEmptyRR-R*t server_close(RIt read_readytexctrequest((s/lib/python2.7/site-packages/astropy/samp/hub.pyRs2 )  )  cC`s}tjd}xg|D]_}||jkrxG|j|D]5}|j|j|j|didd6id6q9WqqWdS(Nusamp.hub.event.shutdowniu samp.mtypeu samp.params(Rtget_mtype_subtypesRDt_notify_RRB(RItmsubstmtypetkey((s/lib/python2.7/site-packages/astropy/samp/hub.pyR<s   c C`stjd}x|D]w}||jkr|j|d}xN|j|D]<}|j|j|j|didd6i|d6d6qJWqqWdS(Nusamp.hub.event.registeriu samp.mtypeuidu samp.params(RRRDRBRcR(RIRRRt public_idR((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_notify_registerFs   c C`stjd}x|D]}||jkr|j|d}x]|j|D]K}||krJ|j|j|j|didd6i|d6d6qJqJWqqWdS(Nusamp.hub.event.unregisteriu samp.mtypeuidu samp.params(RRRDRBRcR(RIRRRRR((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_notify_unregisterRs    c C`stjd}x|D]}||jkr|j|d}x\|j|D]J}|j|j|j|didd6i|d6|j|d6d6qJWqqWdS(Nusamp.hub.event.metadataiu samp.mtypeuidumetadatau samp.params(RRRDRBRcRRC(RIRRRRR((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_notify_metadata^s    c C`stjd}x|D]}||jkr|j|d}x\|j|D]J}|j|j|j|didd6i|d6|j|d6d6qJWqqWdS(Nusamp.hub.event.subscriptionsiu samp.mtypeuidu subscriptionsu samp.params(RRRDRBRcRRE(RIRRRRR((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_notify_subscriptionsls    c C`sd}tjd}|j|d}|j|d}x|D]{}||jkrA||j|krAtjdj||jd|d|||j idd6id d 6d 6fqAqAWdS( NcS`s|jjj|||dS(N(tsamptclienttreceiveNotification(tendpointRt hub_public_idtmessage((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_xmlrpc_call_disconnect{susamp.hub.disconnectiiunotify disconnection to {}RRu samp.mtypeuTimeout expired!ureasonu samp.params( RRRBRFRDRtdebugRtRRA(RIRRRRRR((s/lib/python2.7/site-packages/astropy/samp/hub.pyRys  " cC`s|jtjddS(Nupingu1(RRR(RI((s/lib/python2.7/site-packages/astropy/samp/hub.pyRYs  cC`sfg}xY|jD]N}||j|kr|j|||kr^|j|j|dq^qqW|S(Ni(RCRRB(RIRtvaluetpublic_id_listt private_id((s/lib/python2.7/site-packages/astropy/samp/hub.pyt_query_by_metadatas "cC`s|j|||jkr||jkr\|j|d}|t|jf|j| metadata = {}iuInvalid client IDiu"Private-key {} expired or invalid.( RRBt_public_id_to_private_keyRRRtRRCR (RIRt client_idtclient_private_key((s/lib/python2.7/site-packages/astropy/samp/hub.pyR^ s       cC`s|j|||jkrtjdj|t|||jkr|j|}x=|D]2}y|j|j|Wq^t k rq^Xq^Wnt j ||j|}|j |d r||kr||kr||=qqqWqqWtjdj|t|xZ|D]R}||jkr||j|kr|j|j|qqN|g|j| {}iu"Private-key {} expired or invalid.u(RRBRRRtRRERDRt ValueErrortcopytdeepcopytendswitht startswithRRR (RIRtmtypest prev_mtypesRtoriginal_mtypestmtype2((s/lib/python2.7/site-packages/astropy/samp/hub.pyR_s>              cC`s|j|||jkr|j|}|dk r||jkrwtjdj|t|j||j|Stjdj|iSqt ddnt ddj|dS(Nu-get_subscriptions: client-id = {} mtypes = {}u2get_subscriptions: client-id = {} mtypes = missingiuInvalid client IDiu"Private-key {} expired or invalid.( RRBRRRERRRtRR (RIRRR((s/lib/python2.7/site-packages/astropy/samp/hub.pyR`Ps       cC`s|j|||jkrg}x>|jjD]-}||kr2|j|j|dq2q2Wtjdj|||Stddj|dS(Niu5get_registered_clients: private_key = {} clients = {}iu"Private-key {} expired or invalid.(RRBRRRRRtR (RIRt reg_clientstpkey((s/lib/python2.7/site-packages/astropy/samp/hub.pyRafs     cC`s|j|||jkri}xM|jjD]<}||kr2|j||r2i||j|d>> from astropy.samp import SAMPHubServer >>> SAMPHubServer.get_mtype_subtypes("samp.app.ping") ['samp.app.ping', 'samp.app.*', 'samp.*', '*'] u.iiuu.*u*(tsplittlistRtlentreverseRR(RtsubtypesRtindexestit tmp_mtype((s/lib/python2.7/site-packages/astropy/samp/hub.pyRs       cC`sXt}tj|}x<|D]4}||jkr||j|krPt}qPqqW|S(N(RRRRDRq(RIRRt subscribedRtmsub((s/lib/python2.7/site-packages/astropy/samp/hub.pyRs cC`s|j|||jkr|j|j||dtkrctddj||dn|jd|jd|||fiStddj|dS(Nu samp.mtypeiu$Client {} not subscribed to MType {}RRiu"Private-key {} expired or invalid.( RRBRRRR RtRR(RIRt recipient_idR((s/lib/python2.7/site-packages/astropy/samp/hub.pyRcs    c C`s||jkrdS|j|d}y[tjdj|d|||j|}||f}d}|j||||Wn9tk r}tjdj|d|||t nXdS(Niunotify {} from {} to {}u samp.mtypeureceiveNotificationu7{} notification from client {} to client {} failed [{}]( RBRRRtRt _retry_methodt ExceptionRRR ( RItsender_private_keytrecipient_public_idRtsender_public_idtrecipient_private_keyt arg_paramstsamp_method_nameR((s/lib/python2.7/site-packages/astropy/samp/hub.pyRs"      cC`sl|j|||jkrPd|kr:tddn|j||}|Stddj|dS(Nu samp.mtypeiusamp.mtype keyword is missingiu"Private-key {} expired or invalid.(RRBR t _notify_all_Rt(RIRRt recipient_ids((s/lib/python2.7/site-packages/astropy/samp/hub.pyRds   c C`sg}tj|d}x|D]}||jkr xg|j|D]U}||krC|j|d}|j||jd|jd|||fqCqCWq q W|S(Nu samp.mtypeiRR(RRRDRBRRRc(RIRRRRRRt _recipient_id((s/lib/python2.7/site-packages/astropy/samp/hub.pyRs   c C`s|j|||jkr|j|j||dtkrctddj||dn|j|d}|j||}|jd|j d|||||f|Stddj|dS( Nu samp.mtypeiu$Client {} not subscribed to MType {}iRRiu"Private-key {} expired or invalid.( RRBRRRR Rtt_get_new_hub_msg_idRt_call_(RIRRtmsg_tagRRtmsg_id((s/lib/python2.7/site-packages/astropy/samp/hub.pyRes    c C`s||jkrdSyntjdj|jdd|||d|j|}|||f}d}|j||||WnRtk r} tj dj|d|jdd||t | | t nXdS(Nucall {} from {} to {} ({})u;;iu samp.mtypeu receiveCallu5{} call {} from client {} to client {} failed [{},{}]( RBRRRtRRRRRRttypeR ( RIRRRRRRRtsamp_methodNameR((s/lib/python2.7/site-packages/astropy/samp/hub.pyRs"  cC`s|j|||jkrpd|krCtddj|n|j|d}|j||||}|Stddj|dS(Nu samp.mtypeiu5samp.mtype keyword is missing in message tagged as {}iiu"Private-key {} expired or invalid.(RRBR Rtt _call_all_(RIRRRRR((s/lib/python2.7/site-packages/astropy/samp/hub.pyRf2s    c C`si}tj|d}x|D]}||jkr x||j|D]j}||krC|j||} |j|d} | || <|jd|jd||| | |fqCqCWq q W|S(Nu samp.mtypeiRR(RRRDRRBRR( RIRRRRRRRRt_msg_idtreceiver_public_id((s/lib/python2.7/site-packages/astropy/samp/hub.pyRAs      cC`s|j|||jkrt|}tj}i}|j||d|}d|j|s     N(uunknowni(Ft__name__t __module__t__doc__RRqRVtpropertyRWRjRnRxR~RRRRRRRRRRR?RRRRRRRRRRRYRRZRR[RR\R]R^R_R`RaRbt staticmethodRRRcRRdRReRRfRRgRhR RRRRRRRRkRlRm(((s/lib/python2.7/site-packages/astropy/samp/hub.pyR#s?    U        *   $  (      )  4   (          7       ' tWebProfileDialogcB`s)eZdZdZdZdZRS(uU A base class to make writing Web Profile GUI consent dialogs easier. The concrete class must: 1) Poll ``handle_queue`` periodically, using the timer services of the GUI's event loop. This function will call ``self.show_dialog`` when a request requires authorization. ``self.show_dialog`` will be given the arguments: - ``samp_name``: The name of the application making the request. - ``details``: A dictionary of details about the client making the request. - ``client``: A hostname, port pair containing the client address. - ``origin``: A string containing the origin of the request. 2) Call ``consent`` or ``reject`` based on the user's response to the dialog. cC`sy|jj}Wn$tjk r)natk r9nQXt|dtrZ|d}n|dd}|j||d|d|ddS(Niu samp.nameii(RzRRRtAttributeErrorRRt show_dialog(RIRt samp_name((s/lib/python2.7/site-packages/astropy/samp/hub.pyt handle_queuejs  cC`s|jjtdS(N(R{RRq(RI((s/lib/python2.7/site-packages/astropy/samp/hub.pytconsentyscC`s|jjtdS(N(R{RR(RI((s/lib/python2.7/site-packages/astropy/samp/hub.pytreject|s(R+R,R-R4R5R6(((s/lib/python2.7/site-packages/astropy/samp/hub.pyR0Os  (.t __future__RRRRRRRR0R4RRRtextern.six.movesRRRRtextern.six.moves.urllib.parseRRRt constantsR R terrorsR R R tutilsRRRtlockfile_helpersRRtstandard_profileRRRRRt__all__t__doctest_skip__tobjectRR0(((s/lib/python2.7/site-packages/astropy/samp/hub.pyts8"          1