import itertools

from _init_environment import SetHostMTurkConnection

def description_filter(substring):
	return lambda hit: substring in hit.Title

def disable_hit(hit):
	return conn.disable_hit(hit.HITId)

def dispose_hit(hit):
	# assignments must be first approved or rejected
	for assignment in conn.get_assignments(hit.HITId):
		if assignment.AssignmentStatus == 'Submitted':
			conn.approve_assignment(assignment.AssignmentId)
	return conn.dispose_hit(hit.HITId)

def cleanup():
	"""Remove any boto test related HIT's"""

	global conn
	
	conn = SetHostMTurkConnection()


	is_boto = description_filter('Boto')
	print 'getting hits...'
	all_hits = list(conn.get_all_hits())
	is_reviewable = lambda hit: hit.HITStatus == 'Reviewable'
	is_not_reviewable = lambda hit: not is_reviewable(hit)
	hits_to_process = filter(is_boto, all_hits)
	hits_to_disable = filter(is_not_reviewable, hits_to_process)
	hits_to_dispose = filter(is_reviewable, hits_to_process)
	print 'disabling/disposing %d/%d hits' % (len(hits_to_disable), len(hits_to_dispose))
	map(disable_hit, hits_to_disable)
	map(dispose_hit, hits_to_dispose)

	total_hits = len(all_hits)
	hits_processed = len(hits_to_process)
	skipped = total_hits - hits_processed
	fmt = 'Processed: %(total_hits)d HITs, disabled/disposed: %(hits_processed)d, skipped: %(skipped)d'
	print fmt % vars()

if __name__ == '__main__':
	cleanup()
