Python:Mwclient/ init .py

import page, errors
 
class Categories(object):
	def __init__(self, mediawiki):
		self.mediawiki = mediawiki
	def __getitem__(self, key):
		return Category(self.mediawiki, key)
 
class Category(object):
	def __init__(self, mediawiki, name):
		self.mediawiki = mediawiki
		self.name = name
 
		self.key = ''
		self.load_chunck()
 
	def load_chunk(self):
		info = self.mediawiki.query('category|permissions', cptitle = self.name, 
			cplimit = '500', cpfrom = self.key)
		if 'error' in info: raise RuntimeError, info['error']
 
		self._iterator = info.get('pages', {}).itervalues()
		self.key = info.get('query', {}).get('category', {}).get('next')
 
	def __iter__(self):
		return self
 
	def next(self):
		try:
			item = self._iterator.next()
			return page.Page(self.mediawiki, item['title'], (item, ))
		except StopIteration:
			if self.key is None: raise StopIteration		
			self.load_chunck()
			return self.next()