# HG changeset patch
# User Goffi <goffi@goffi.org>
# Date 1304688842 -7200
# Node ID b544bec477dd936a5056ec47d2daa724b56677e5
# Parent  f1f7c94278f25ee5f0d63e90c984a0208916ab62
core: multiple identities with same category/type are now managed with getServerServiceEntities

diff -r f1f7c94278f2 -r b544bec477dd src/tools/memory.py
--- a/src/tools/memory.py	Fri May 06 15:33:07 2011 +0200
+++ b/src/tools/memory.py	Fri May 06 15:34:02 2011 +0200
@@ -553,12 +553,25 @@
         @param profile: which profile is using this server ?"""
         if not self.server_identities.has_key(profile):
             self.server_identities[profile] = {}
-        self.server_identities[profile][(category, type)] = entity
+        if not self.server_identities[profile].has_key((category, type)):
+            self.server_identities[profile][(category, type)]=set()
+        self.server_identities[profile][(category, type)].add(entity)
+
+    def getServerServiceEntities(self, category, type, profile):
+        """Return all available entities for a service"""
+        if self.server_identities.has_key(profile):
+            return self.server_identities[profile].get((category, type), set())
+        else:
+            return None
 
     def getServerServiceEntity(self, category, type, profile):
-        if self.server_identities.has_key(profile):
-            return self.server_identities[profile].get((category, type))
-        return None
+        """Helper method to get first available entity for a service"""
+        entities = self.getServerServiceEntities(category, type, profile)
+        if entities == None:
+            warning(_("Entities not available, maybe they haven't been asked to server yet ?"))
+            return None
+        else:
+            return list(entities)[0] if entities else None
 
     def hasServerFeature(self, feature, profile_key='@DEFAULT@'):
         """Tell if the server of the profile has the required feature"""