Return groups after comptia.fetch()

1 files changed, 5 insertions, 1 deletions

diff --git a/gsxws.py b/gsxws.py
index 111af5b..888838d 100755
--- a/gsxws.py
+++ b/gsxws.py
@@ -223,6 +223,8 @@ class CompTIA:
         
     def fetch(self):
         '''
+        The CompTIA Codes Lookup API retrieves a list of CompTIA groups and modifiers. 
+
         Here we must resort to raw XML parsing since SUDS throws this:
         suds.TypeNotFound: Type not found: 'comptiaDescription'
         when calling CompTIACodes()...
@@ -245,13 +247,15 @@ class CompTIA:
 
             for ci in el.findall('comptiaCodeInfo'):
                 group['codes'][ci[0].text] = ci[1].text
-                
+
             self.groups[comp_id] = group
             
         for el in root.findall('.//comptiaModifier'):
             descr, code = list(el)
             self.modifiers[code.text] = descr.text
 
+        return self.groups
+
     def symptoms(self, component=None):
         symptoms = self.data['symptoms']