root/controlies/Utils/avahiClient.py @ 93
| 91 | manumora | ##############################################################################
|
|
# -*- coding: utf-8 -*-
|
|||
# Project: ControlIES
|
|||
# Module: avahiClient.py
|
|||
# Purpose: Avahi client to detect ltsp servers
|
|||
# Language: Python 2.5
|
|||
# Date: 27-Feb-2011.
|
|||
# Ver: 27-Feb-2011.
|
|||
# Author: José Luis Redrejo Rodriguez
|
|||
# Copyright: 2011 - José Luis Redrejo Rodriguez <jredrejo @no-spam@ debian.org>
|
|||
# Modified by: Manuel Mora Gordillo <manuito @no-spam@ gmail.com>
|
|||
#
|
|||
# ControlIES is free software: you can redistribute it and/or modify
|
|||
# it under the terms of the GNU General Public License as published by
|
|||
# the Free Software Foundation, either version 3 of the License, or
|
|||
# (at your option) any later version.
|
|||
# ControlIES is distributed in the hope that it will be useful,
|
|||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|||
# GNU General Public License for more details.
|
|||
# You should have received a copy of the GNU General Public License
|
|||
# along with ControlIES. If not, see <http://www.gnu.org/licenses/>.
|
|||
#
|
|||
##############################################################################
|
|||
| 89 | manumora | ||
| 91 | manumora | try:
|
|
import dbus
|
|||
if getattr(dbus, 'version', (0,0,0)) >= (0,41,0):
|
|||
import dbus.glib
|
|||
except ImportError:
|
|||
dbus = None
|
|||
| 89 | manumora | ||
| 91 | manumora | if dbus:
|
|
try:
|
|||
import avahi
|
|||
except ImportError:
|
|||
avahi = None
|
|||
else:
|
|||
avahi = None
|
|||
| 89 | manumora | ||
| 91 | manumora | from twisted.internet import defer, threads
|
|
from twisted.internet import glib2reactor
|
|||
glib2reactor.install()
|
|||
| 89 | manumora | ||
| 91 | manumora | class avahiClient():
|
|
| 90 | manumora | ||
| 93 | manumora | def __init__(self, type):
|
|
| 91 | manumora | self._callbacks = {'new-service': [], 'remove-service': [] }
|
|
# initialize dbus stuff needed for discovery
|
|||
self.bus = dbus.SystemBus()
|
|||
avahi_bus = self.bus.get_object(avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER)
|
|||
self.server = dbus.Interface(avahi_bus, avahi.DBUS_INTERFACE_SERVER)
|
|||
stype = '_workstation._tcp'
|
|||
| 93 | manumora | stype = '_controlaula._tcp'
|
|
stype = type
|
|||
| 91 | manumora | domain = 'local'
|
|
self._plugged = {}
|
|||
| 89 | manumora | ||
| 91 | manumora | avahi_browser = self.server.ServiceBrowserNew(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, stype, domain, dbus.UInt32(0))
|
|
obj = self.bus.get_object(avahi.DBUS_NAME, avahi_browser)
|
|||
| 93 | manumora | self.browser = dbus.Interface(obj, avahi.DBUS_INTERFACE_SERVICE_BROWSER)
|
|
| 89 | manumora | ||
| 91 | manumora | def start(self):
|
|
self.browser.connect_to_signal('ItemNew', self.new_service)
|
|||
self.browser.connect_to_signal('ItemRemove', self.remove_service)
|
|||
def stop(self):
|
|||
self.bus.close()
|
|||
| 89 | manumora | ||
| 91 | manumora | def new_service(self, interface, protocol, name, type, domain, flags):
|
|
def resolve_service_reply(*service):
|
|||
address, port = service[-4:-2]
|
|||
name = unicode(service[2])
|
|||
for cb in self._callbacks['new-service']:
|
|||
self._plugged[name] = (address,port)
|
|||
cb(self,name, address, port)
|
|||
def resolve_service_error(exception):
|
|||
| 93 | manumora | self.warning('could not resolve daap service %s %s: %s' % (name, domain, exception))
|
|
| 89 | manumora | ||
| 91 | manumora | self.server.ResolveService(interface, protocol, name, type, domain,
|
|
avahi.PROTO_UNSPEC, dbus.UInt32(0),
|
|||
reply_handler=resolve_service_reply,
|
|||
error_handler=resolve_service_error)
|
|||
| 90 | manumora | ||
| 91 | manumora | def remove_service(self, interface, protocol, name, type, domain,server):
|
|
address, port = self._plugged[name]
|
|||
for cb in self._callbacks['remove-service']:
|
|||
cb(self,name, address, port)
|
|||
| 89 | manumora | ||
| 91 | manumora | def add_callback(self, sig_name, callback):
|
|
self._callbacks[sig_name].append(callback)
|
|||
| 89 | manumora | ||
| 91 | manumora | def remove_callback(self, sig_name, callback):
|
|
self._callback[sig_name].remove(callback)
|