def get_local_services_list():
'''Get the list of services running on this node from the databases.'''
# Define the location of the service database.
servicedb = conf.servicedb
service_list = []
# Set up a connection to the database.
_utils.debug("DEBUG: Opening service database.",5)
connection = sqlite3.connect(servicedb)
cursor = connection.cursor()
# Pull a list of running web apps on the node.
_utils.debug("DEBUG: Getting list of running webapps from database.",5)
cursor.execute("SELECT name FROM webapps WHERE status='active';")
results = cursor.fetchall()
for service in results:
service_list += [{'name':service[0],'path':'/'+service[0],'description':''}]
# Pull a list of daemons running on the node. This means that most of the web apps users will access will be displayed.
_utils.debug("DEBUG: Getting list of running servers from database.",5)
cursor.execute("SELECT name FROM daemons WHERE status='active' AND showtouser='******';")
results = cursor.fetchall()
for service in results:
_utils.debug("DEBUG: Value of service: %s" % str(service))
if service[0] in conf.service_info:
path = conf.service_info[service[0]]
else:
path = '/'+service[0]+'/'
service_list += [{'name':service[0],'path':path,'description':''}]
# Clean up after ourselves.
_utils.debug("DEBUG: Closing service database.",5)
cursor.close()
return service_list
def update_services_cache(service,action = 'add'):
print(service)
print('updating cache')
if os.path.exists(conf.services_cache):
services = json.loads(_utils.file2str(conf.services_cache))
else:
services = {}
if action.lower() == 'add':
services.update(service)
_utils.debug('Service added')
elif action.lower() == 'del' and service in services:
del services[service]
_utils.debug('Service removed')
_utils.str2file(json.dumps(services),conf.services_cache)
def browse_callback(sdRef, flags, interfaceIndex, errorCode, serviceName, regtype, replyDomain):
print(serviceName)
if errorCode != pybonjour.kDNSServiceErr_NoError:
print
return
if not (flags & pybonjour.kDNSServiceFlagsAdd):
update_services_cache(serviceName+'.'+regtype+replyDomain,'del')
return
resolve_sdRef = pybonjour.DNSServiceResolve(0, interfaceIndex, serviceName, regtype, replyDomain, resolve_callback)
try:
while not resolved:
ready = select.select([resolve_sdRef], [], [], timeout)
if resolve_sdRef not in ready[0]:
_utils.debug('Resolve timed out',1)
break
pybonjour.DNSServiceProcessResult(resolve_sdRef)
else:
resolved.pop()
finally:
resolve_sdRef.close()
import re
service_list = []
srvcdict = file2json(conf.services_cache)
if not srvcdict: return service_list
for name, vals in srvcdict.items():
if re.search('\.__byz__\._[tucdp]{3}',name):
description = ''
path = vals['host']
if vals['port']: path += str(vals['port'])
if vals['text'] not in ('','\x00'):
for entry in vals['text'].split('\n'):
key,val = (entry+'=').split('=')
v = list(val)
v.pop(-1)
val = ''.join(v)
if key == conf.uri_post_port_string_key:
path += val
elif key == conf.service_description_key:
description += val
name = re.sub('\.__byz__\._[udtcp]{3}.*','',name)
service_list = [{'name':name,'path':path,'description':description}]
return service_list
def get_services_list():
local_srvc = get_local_services_list()
remote_srvc = get_remote_services_list()
return local_srvc + remote_srvc
if __name__ == '__main__':
_utils.debug(get_services_list(),0)