import beagle
import gtk

sent = 0
snippets = 0
closed = 0

c=beagle.Client()

def test2(req, snip):
	print 'Got:', snip.get_snippet()
	global snippets
	snippets += 1
	
def test3(req):
	global closed, snippets, sent
	print 'Got closed:',closed, ' Snippets:', snippets, ' Sent:', sent
	closed += 1
	
def test(query, hits):
	for hit in hits.get_hits():
		if hit.get_type() == "IMLog":
			global sent
			print 'Send snippet req'
			sent += 1
			s=beagle.SnippetRequest()
			s.set_query(query)
			s.set_hit(hit)
			s.connect('response', test2)
			s.connect('closed', test3)
			c.send_request_async(s)
			print 'Done'

def test_sync(query, hits):
	for hit in hits.get_hits():
		if hit.get_type() == "IMLog":
			global sent, closed, snippets
			print 'Send snippet req'
			sent += 1
			s=beagle.SnippetRequest()
			s.set_query(query)
			s.set_hit(hit)
			resp = c.send_request(s)
			print 'Done'
			print 'Got:', resp.get_snippet()
			closed += 1
			snippets += 1
			print 'Got closed:',closed, ' Snippets:', snippets, ' Sent:', sent
			
r=beagle.Query()
r.add_text("le")
r.connect("hits-added", test_sync)
c.send_request_async(r)

gtk.main()