# Copyright 2012 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2
 
from .AsyncScheduler import AsyncScheduler
 
class TaskScheduler(AsyncScheduler):
 
	"""
	A simple way to handle scheduling of AbstractPollTask instances. Simply
	pass a task iterator into the constructor and call start(). Use the
	poll, wait, or addExitListener methods to be notified when all of the
	tasks have completed.
	"""
 
	def __init__(self, task_iter, **kwargs):
		AsyncScheduler.__init__(self, **kwargs)
		self._task_iter = task_iter
 
	def _next_task(self):
		return next(self._task_iter)