From 59ab4b440dbc6972215e948c23f7004311d018b3 Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.com>
Date: Sat, 19 Feb 2011 13:23:11 -0500
Subject: [PATCH] UserList

---
 convore/api.py | 60 +++++++++++++++++++++++++++++++-------------------
 1 file changed, 37 insertions(+), 23 deletions(-)

diff --git a/convore/api.py b/convore/api.py
index 00bb3c2..c828288 100644
--- a/convore/api.py
+++ b/convore/api.py
@@ -2,6 +2,8 @@ import requests
 import json
 import models
 
+from UserList import UserList
+
 API_URL = 'https://convore.com/api/'
 
 def login(username, password):
@@ -9,32 +11,44 @@ def login(username, password):
     requests.add_autoauth(API_URL, auth)
     
 
-class Groups(object):
-    
+class Groups(UserList):
     
     def __init__(self):
-        pass
-        
-        
-    def __iter__(self):
-        r = requests.get(API_URL + 'groups.json')
-        groups = json.loads(r.content)['groups']
-        
-        for group in groups:
-            _group = models.Group()
-            _group.import_from_api(group)
-            yield _group
+        self.data = []
+        self.sync()
+
 
     def __getitem__(self, key):
-        r = requests.get(API_URL + 'groups/%s.json' % key)
-        group = models.Group()
-        _group = json.loads(r.content)['group']
-        group.import_from_api(_group)
 
-        return _group
+        if isinstance(key, int):
+            key = unicode(key)
+
+        for group in self.data:
+            if key in (group.id, group.slug):
+               return group
+
+    def __contains__(self, key):
         
-    def iterkeys():
-        return []
-        
-    def iteritems():
-        return
+        if isinstance(key, int):
+            key = unicode(key)
+
+        for group in self.data:
+            if key in (group.id, group.slug):
+               return True
+
+        return False
+
+    def __iter__(self):
+        for group in self.data:
+            yield group
+
+    def sync(self):
+
+        self.data = []
+
+        r = requests.get(API_URL + 'groups.json')
+        for _group in json.loads(r.content)['groups']:
+
+            group = models.Group()
+            group.import_from_api(_group)
+            self.data.append(group)