[GRASS-SVN] r33648 - grass/trunk/scripts/db.test

[svn_grass at osgeo.org]

Author: glynn
Date: 2008-10-01 19:06:56 -0400 (Wed, 01 Oct 2008)
New Revision: 33648

Added:
   grass/trunk/scripts/db.test/db.test.py
Log:
Convert db.test to Python


Added: grass/trunk/scripts/db.test/db.test.py
===================================================================
--- grass/trunk/scripts/db.test/db.test.py	                        (rev 0)
+++ grass/trunk/scripts/db.test/db.test.py	2008-10-01 23:06:56 UTC (rev 33648)
@@ -0,0 +1,82 @@
+#!/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
+import subprocess
+import 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 subprocess.call(['diff', result, expected]) != 0:
+		grass.error("RESULT: ******** ERROR ********")
+	    else:
+		grass.message("RESULT: OK")
+
+if __name__ == "__main__":
+    options, flags = grass.parser()
+    main()
+


Property changes on: grass/trunk/scripts/db.test/db.test.py
___________________________________________________________________
Name: svn:executable
   + *