DiscoInfo: Add has_category() method

1 files changed, 6 insertions, 0 deletions

diff --git a/nbxmpp/structs.py b/nbxmpp/structs.py
index ea0c58e..64ce391 100644
--- a/nbxmpp/structs.py
+++ b/nbxmpp/structs.py
@@ -372,6 +372,12 @@ class DiscoInfo(namedtuple('DiscoInfo', 'stanza identities features dataforms ti
                 return identity.type
         return None
 
+    def has_category(self, category):
+        for identity in self.identities:
+            if identity.category == category:
+                return True
+        return False
+
     @property
     def httpupload_max_file_size(self):
         size = self.get_field_value(NS_HTTPUPLOAD_0, 'max-file-size')