tko/compose_query.cgi

#!/usr/bin/python

"""
Selects all rows and columns that satisfy the condition specified
and draws the matrix. There is a seperate SQL query made for every (x,y)
in the matrix.
"""


print "Content-type: text/html\n"
import cgi, cgitb, re
import sys, os

tko = os.path.dirname(os.path.realpath(os.path.abspath(sys.argv[0])))
sys.path.insert(0, tko)

import display, frontend, db, query_lib

cgitb.enable()
db = db.db()

def main():

	# parse the fields from the form.
	form = cgi.FieldStorage()
	columns = 'kernel'
	rows = 'test'
	condition = None
	for field in form:
		value = form[field].value
		if field == 'columns':
			columns = value
		elif field == 'rows':
			rows = value
		elif field == 'condition':
			condition = value

	# parse the conditions into sql query and value list.
	condition_sql = ""
	condition_value = []
	if condition:
		condition_list = query_lib.parse_condition(condition)
		condition_sql, condition_value =   \
		 query_lib.generate_sql_condition(condition_list)

	# get all possible column values.
	column_groups = frontend.anygroup.selectunique(db, columns)

	# get all possible row values.
	row_groups = frontend.anygroup.selectunique(db,rows)
	# keep only those values in rows/columns that have a test
	# corresponding to it.
	row_groups = query_lib.prune_list(row_groups, condition_sql,  \
					  condition_value)
	column_groups = query_lib.prune_list(column_groups, condition_sql, \
					     condition_value)

	# prepare the header for the table.
	headers = [g.name for g in column_groups]

	header_row = [display.box(x, header=True) for x in headers]
	header_row.insert(0, display.box("", header=True))

	matrix = [header_row]

	# get all the tests that satify the given condition.
	tests = query_lib.get_tests(condition_sql, condition_value)

	for r_group in row_groups:
		row = [display.box(r_group.name)]

		# build the row sql for this row.
		row_expr = [ " %s = %%s " % r_group.idx_name for val in r_group.idx_value]
		row_sql = " (%s) " % " or ".join(row_expr)

		# get individual unit values
		for c_group in column_groups:
			# get the list of tests that belong to this x,y in the matrix.
			xy_test = [test for test in tests
				   if query_lib.get_value(test, r_group.idx_name) \
				   in r_group.idx_value \
				   and query_lib.get_value(test,c_group.idx_name) \
				   in c_group.idx_value]

			# build the column sql
			column_expr = [ " %s = %%s " % c_group.idx_name for val in c_group.idx_value]
			column_sql = " (%s) " % " or ".join(column_expr)

			sql = "t where %s and %s " % (row_sql, column_sql)

			# add the corresponding values of the fields to
			# the value list.

			value = []
			value.extend(r_group.idx_value)
			value.extend(c_group.idx_value)

			# append the condition sql and the values to the
			# sql/list respectively.
			if condition_sql:
				sql += " and "
				sql += condition_sql
				value.extend(condition_value)

			value_str = [str(val) for val in value]
			link = 'test.cgi?sql=%s&values=%s' % \
				(sql, ','.join(value_str))
			row.append(display.status_count_box(db, xy_test, link))
		matrix.append(row)
	display.print_table(matrix)


main()