mirror of
https://github.com/kennethreitz-archive/python-convore.git
synced 2026-06-05 23:40:18 +00:00
Kenneth Reitz
72 lines
1.5 KiB
Python
72 lines
1.5 KiB
Python
import requests
|
|
import json
|
|
import models
|
|
|
|
from UserList import UserList
|
|
|
|
API_URL = 'https://convore.com/api/'
|
|
|
|
def login(username, password):
|
|
auth = requests.AuthObject(username, password)
|
|
requests.add_autoauth(API_URL, auth)
|
|
|
|
|
|
class Groups(UserList):
|
|
|
|
def __init__(self):
|
|
self.data = []
|
|
self.sync()
|
|
|
|
def joined(self):
|
|
"""Returns list of """
|
|
|
|
def __getitem__(self, key):
|
|
|
|
if isinstance(key, int):
|
|
key = unicode(key)
|
|
|
|
for group in self.data:
|
|
if key in (group.id, group.slug):
|
|
return group
|
|
|
|
try:
|
|
r = requests.get(API_URL + 'groups/%s.json' % key)
|
|
r.raise_for_status()
|
|
|
|
group = models.Group()
|
|
group.import_from_api(json.loads(r.content)['group'])
|
|
|
|
return group
|
|
|
|
except requests.HTTPError:
|
|
return None
|
|
|
|
|
|
def __contains__(self, key):
|
|
|
|
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)
|
|
group.joined = True
|
|
self.data.append(group)
|