o va'@sdZddlZddlZddlmZddlmZzddlmZ m Z m Z m Z dZ Wn ey1dZ Ynwe rXdZed d e d DZed e eekrXed edddZGdddeZGdddeZGdddeZGdddeZdS)zBManage autodiscover Glances server (thk to the ZeroConf protocol).N)BSD)logger) __version__ServiceBrowser ServiceInfoZeroconfTF)rrcCsg|]}t|qS)int).0Znumr r 6/usr/lib/python3/dist-packages/glances/autodiscover.py *sr .zZeroconf version {} detected.z'Please install zeroconf 0.17 or higher.z_%s._tcp.local.glancesc@8eZdZdZddZddZddZdd Zd d Zd S) AutoDiscoveredz1Class to manage the auto discovered servers dict.cCs g|_dSN _server_listselfr r r __init__:s zAutoDiscovered.__init__cCs|jSz.Return the current server list (list of dict).rrr r r get_servers_list?szAutoDiscovered.get_servers_listcCs||j||<dSzCSet the key to the value for the server_pos (position in the list).NrrZ server_poskeyvaluer r r set_serverCszAutoDiscovered.set_serverc CsJ||dd||ddddd}|j|tdt|j|jfd S) zAdd a new server to the list.:rrZUNKNOWNZDYNAMIC)rnameipportZusernameZpasswordstatustype%Updated servers list (%s servers): %sN)splitrappendrdebuglen)rr"r#r$Z new_serverr r r add_serverGs   zAutoDiscovered.add_serverc Csv|jD]5}|d|kr8z|j|td|tdt|j|jfWqty7td|YqwqdS)zRemove a server from the dict.rzRemove server %s from the listr'z%Cannot remove server %s from the listN)rremoverr*r+ ValueErrorerror)rr"ir r r remove_serverVs     zAutoDiscovered.remove_serverN) __name__ __module__ __qualname____doc__rrrr,r1r r r r r6s rc@r) GlancesAutoDiscoverListenerz%Zeroconf listener for Glances server.cCs t|_dSr)rserversrr r r rhs z$GlancesAutoDiscoverListener.__init__cCs |jSr)r7rrr r r rls z,GlancesAutoDiscoverListener.get_servers_listcCs|j|||dSr)r7rrr r r rpsz&GlancesAutoDiscoverListener.set_servercCsv|tkrdStd||f|||}|r4t|j}|j}|j |||t d|||fdSt ddS)zMethod called when a new Zeroconf client is detected. Return True if the zeroconf client is a Glances server Note: the return code will never be used Fz"Check new Zeroconf server: %s / %sz+New Glances server detected (%s from %s:%s)zCNew Glances server detected, but Zeroconf info failed to be grabbedT) zeroconf_typerr*Zget_service_infosocketZ inet_ntoaaddressr$r7r,infoZwarning)rzeroconfsrv_typesrv_namer;Z new_server_ipZnew_server_portr r r add_servicets$  z'GlancesAutoDiscoverListener.add_servicecCs|j|td|dS)z Remove the server from the list.z2Glances server %s removed from the autodetect listN)r7r1rr;)rr<r=r>r r r remove_services z*GlancesAutoDiscoverListener.remove_serviceN) r2r3r4r5rrrr?r@r r r r r6ds r6c@s2eZdZdZd ddZddZddZd d ZdS) GlancesAutoDiscoverServerzMImplementation of the Zeroconf protocol (server side for the Glances client).Nc Cstr>tdzt|_Wntjy+}ztd|d|_WYd}~dSd}~wwt|_ t |jt |j |_ d|_dStdd|_dS)Nz*Init autodiscover mode (Zeroconf protocol)zCannot start Zeroconf (%s)FTz>Cannot start autodiscover mode (Zeroconf lib is not installed)) zeroconf_tagrr;rr<r9r/zeroconf_enable_tagr6listenerrr8Zbrowser)rargser r r rs       z"GlancesAutoDiscoverServer.__init__cCstr |jr |jSgS)z.Return the current server list (dict of dict).)rBrCrDrrr r r rs  z*GlancesAutoDiscoverServer.get_servers_listcCs&tr|jr|j|||dSdSdSr)rBrCrDrrr r r rs z$GlancesAutoDiscoverServer.set_servercCs tr |jr|jdSdSdSr)rBrCr<closerr r r rGs zGlancesAutoDiscoverServer.closer)r2r3r4r5rrrrGr r r r rAs   rAc@s.eZdZdZd ddZeddZddZdS) GlancesAutoDiscoverClientzMImplementation of the zeroconf protocol (client side for the Glances server).Nc Csttr|j}zt|_Wntjy&}ztd|WYd}~nd}~wwts=z |dkr2| }Wn t y<Ynwt ||j dd}zt td||j tt|||j ddi|d|_Wn"tyt td||j tt||g|j ddi|d|_Ynwz |j|jWnty}ztd|WYd}~dSd}~wwtd|dStd dS) NzCannot start zeroconf: {}z0.0.0.0rz{}:{}.{})r:r$weightpriority propertiesserver)r"Z addressesr$rIrJrKrLz)Error while announcing Glances server: {}zs.     ./&