From 4cb11c30b9794d20ca7613ca7ea3d919a0a87f09 Mon Sep 17 00:00:00 2001
From: Sebastien Estienne <sebastien.estienne@gmail.com>
Date: Fri, 7 Oct 2005 18:32:57 +0000
Subject:  * quick conversion of avahi-browse in python object

git-svn-id: file:///home/lennart/svn/public/avahi/branches/py-simple-api@706 941a03a8-eaeb-0310-b9a0-b1bbd8fe43fe
---
 avahi-utils/avahi/SimpleAvahiApp.py | 98 +++++++++++++++++++++++++++++++++++++
 1 file changed, 98 insertions(+)

diff --git a/avahi-utils/avahi/SimpleAvahiApp.py b/avahi-utils/avahi/SimpleAvahiApp.py
index f81d189..84f942a 100755
--- a/avahi-utils/avahi/SimpleAvahiApp.py
+++ b/avahi-utils/avahi/SimpleAvahiApp.py
@@ -19,7 +19,105 @@
 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
 # USA.
 
+import sys, getopt
+import avahi, gobject, dbus, avahi.ServiceTypeDatabase
+
+try:
+    import dbus.glib
+except ImportError, e:
+    pass
+
+class SimpleAvahiApp:
+    def __init__(self):
+        self.domain = None
+        self.stype = "_ssh._tcp"
+        self.service_type_browsers = {}
+        self.service_browsers = {}
+        self.service_type_db = avahi.ServiceTypeDatabase.ServiceTypeDatabase()
+        self.bus = dbus.SystemBus()
+        self.server = dbus.Interface(self.bus.get_object(avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER), avahi.DBUS_INTERFACE_SERVER)
+
+        if self.domain is None:
+            # Explicitly browse .local
+            self.browse_domain(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, "local")
+
+            # Browse for other browsable domains
+            db = dbus.Interface(self.bus.get_object(avahi.DBUS_NAME, self.server.DomainBrowserNew(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, "", avahi.DOMAIN_BROWSER_BROWSE)), avahi.DBUS_INTERFACE_DOMAIN_BROWSER)
+            db.connect_to_signal('ItemNew', self.new_domain)
+
+        else:
+            # Just browse the domain the user wants us to browse
+            self.browse_domain(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, self.domain)
+
+        
+    def siocgifname(self, interface):
+         if interface <= 0:
+             return "any"
+         else:
+             return self.server.GetNetworkInterfaceNameByIndex(interface)
+
+    def lookup_service_type(self, stype):
+        try:
+            return self.service_type_db[stype]
+        except KeyError:
+            return "n/a"
+
+    def service_resolved(self, interface, protocol, name, stype, domain, host, aprotocol, address, port, txt):
+        print "Service data for service '%s' of type '%s' (%s) in domain '%s' on %s.%i:" % (name, stype, self.lookup_service_type(stype), domain, self.siocgifname(interface), protocol)
+        print "\tHost %s (%s), port %i, TXT data: %s" % (host, address, port, avahi.txt_array_to_string_array(txt))
+
+    def print_error(self, err):
+        print "Error:", str(err)
+
+    def new_service(self, interface, protocol, name, stype, domain):
+        print "Found service '%s' of type '%s' (%s) in domain '%s' on %s.%i." % (name, stype, self.lookup_service_type(stype), domain, self.siocgifname(interface), protocol)
+        
+        # Asynchronous resolving
+        self.server.ResolveService(interface, protocol, name, stype, domain, avahi.PROTO_UNSPEC, reply_handler=self.service_resolved, error_handler=self.print_error)
+
+    def remove_service(self, interface, protocol, name, stype, domain):
+        print "Service '%s' of type '%s' (%s) in domain '%s' on %s.%i disappeared." % (name, stype, self.lookup_service_type(stype), domain, self.siocgifname(interface), protocol)
+ 
+    def new_service_type(self, interface, protocol, stype, domain):
+        # Are we already browsing this domain for this type? 
+        if self.service_browsers.has_key((interface, protocol, stype, domain)):
+            return
+        
+        print "Browsing for services of type '%s' (%s) in domain '%s' on %s.%i ..." % (stype, self.lookup_service_type(stype), domain, self.siocgifname(interface), protocol)
+        
+        b = dbus.Interface(self.bus.get_object(avahi.DBUS_NAME, self.server.ServiceBrowserNew(interface, protocol, stype, domain)), avahi.DBUS_INTERFACE_SERVICE_BROWSER)
+        b.connect_to_signal('ItemNew', self.new_service)
+        b.connect_to_signal('ItemRemove', self.remove_service)
+        
+        self.service_browsers[(interface, protocol, stype, domain)] = b
+
+    def browse_domain(self, interface, protocol, domain):
+        print "browse_domain"
+        # Are we already browsing this domain?
+        if self.service_type_browsers.has_key((interface, protocol, domain)):
+            return
+        
+        if self.stype is None:
+            print "Browsing domain '%s' on %s.%i ..." % (domain, self.siocgifname(interface), protocol)
+            
+            b = dbus.Interface(self.bus.get_object(avahi.DBUS_NAME, self.server.ServiceTypeBrowserNew(interface, protocol, domain)), avahi.DBUS_INTERFACE_SERVICE_TYPE_BROWSER)
+            b.connect_to_signal('ItemNew', self.new_service_type)
+            
+            self.service_type_browsers[(interface, protocol, domain)] = b
+        else:
+            self.new_service_type(interface, protocol, self.stype, domain)
+
+    def new_domain(self, interface, protocol, domain):
+        # We browse for .local anyway...
+        if domain != "local":
+            self.browse_domain(interface, protocol, domain)
+
 if __name__ == "__main__":
     print "__main__"
+    sb = SimpleAvahiApp()
 
+    try:
+        gobject.MainLoop().run()
+    except KeyboardInterrupt, k:
+        pass
 
-- 
cgit