Geografic-Information-System/scripts/db.test/db.test.py

#!/usr/bin/env python
############################################################################
#
# MODULE:	db.test
# AUTHOR(S):	Radim Blazek
#               Converted to Python by Glynn Clements
# PURPOSE:	Test database driver
# COPYRIGHT:	(C) 2004 by the GRASS Development Team
#
#		This program is free software under the GNU General Public
#		License (version 2). Read the file COPYING that comes with GRASS
#		for details.
#
#############################################################################

#% Module
#%  description: Test database driver, database must exist and set by db.connect.
#%  keywords: database, attribute table
#% End
#% option
#%  key: test
#%  type: string
#%  description: Test name
#%  required : yes
#%  options : test1
#% end

import sys
import os
from grass.script import core as grass

def main():
    test_file = options['test']

    expected = grass.tempfile()
    result = grass.tempfile()

    infile = os.path.join(os.environ['GISBASE'], 'etc', 'db.test', test_file)
    inf = file(infile)

    while True:
	type = inf.readline()
	if not type:
	    break
	type = type.rstrip('\r\n')

	sql = inf.readline().rstrip('\r\n')
	sys.stdout.write(sql + '\n')

	# Copy expected result to temp file

	if type == 'X':
	    r = grass.write_command('db.execute', stdin = sql + '\n')
	else:
	    resf = file(result, 'w')
	    r = grass.write_command('db.select', flags = 'c', stdin = sql + '\n', stdout = resf)
	    resf.close()

	if r != 0:
	    grass.error("EXECUTE: ******** ERROR ********")
	else:
	    grass.message("EXECUTE: OK")

	expf = file(expected, 'w')
	while True:
	    res = inf.readline().rstrip('\r\n')
	    if not res:
		break
	    expf.write(res + '\n')
	expf.close()

	if type == 'S':
	    if grass.call(['diff', result, expected]) != 0:
		grass.error("RESULT: ******** ERROR ********")
	    else:
		grass.message("RESULT: OK")

if __name__ == "__main__":
    options, flags = grass.parser()
    main()