Source

pmxbot / pmxbot / itertools.py

The default branch has multiple heads

Full commit
Jason R. Coombs a6a9059 






































Jason R. Coombs dd63977 
Jason R. Coombs a6a9059 



Jason R. Coombs dd63977 
Jason R. Coombs a6a9059 

Jason R. Coombs ea65239 
Jason R. Coombs a6a9059 









from __future__ import unicode_literals

import io

def always_iterable(item):
	r"""
	Given an item from a pmxbot handler, always return an iterable.

	If the item is None, return an empty iterable.
	>>> list(always_iterable(None))
	[]

	If the item is a string, return an iterable of the lines in the string.
	>>> list(always_iterable('foo'))
	[u'foo']
	>>> list(always_iterable('foo\nbar'))
	[u'foo\n', u'bar']

	>>> list(always_iterable([1,2,3]))
	[1, 2, 3]
	>>> always_iterable(xrange(10))
	xrange(10)

	And any other non-iterable objects are returned as single-tuples of that
	item.
	>>> list(always_iterable(object()))  # doctest: +ELLIPSIS
	[<object object at ...>]
	"""
	if item is None:
		item = ()

	if isinstance(item, basestring):
		item = io.StringIO(unicode(item))

	if not hasattr(item, '__iter__'):
		item = item,

	return item

def generate_results(function):
	"""
	Take a function, which may return an iterator or a static result
	and convert it to a late-dispatched generator.
	"""
	for item in always_iterable(function()):
		yield item

def trap_exceptions(results, handler, exceptions=Exception):
	"""
	Iterate through the results, but if an exception occurs, stop
	processing the results and instead replace
	the results with the output from the exception handler.
	"""
	try:
		for result in results:
			yield result
	except exceptions as exc:
		for result in always_iterable(handler(exc)):
			yield result