[GRASS-SVN] r33570 - grass/trunk/scripts/v.db.update

[svn_grass at osgeo.org]

Author: glynn
Date: 2008-09-27 15:15:30 -0400 (Sat, 27 Sep 2008)
New Revision: 33570

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


Added: grass/trunk/scripts/v.db.update/v.db.update.py
===================================================================
--- grass/trunk/scripts/v.db.update/v.db.update.py	                        (rev 0)
+++ grass/trunk/scripts/v.db.update/v.db.update.py	2008-09-27 19:15:30 UTC (rev 33570)
@@ -0,0 +1,124 @@
+#!/usr/bin/env python
+#
+############################################################################
+#
+# MODULE:       v.db.update
+# AUTHOR(S):    Moritz Lennert
+#               extensions by Markus Neteler
+#               Converted to Python by Glynn Clements
+# PURPOSE:      interface to db.execute to update a column in the attribute table connected to a given map
+# COPYRIGHT:    (C) 2005,2007,2008 by the GRASS Development Team
+#
+#               This program is free software under the GNU General Public
+#               License (>=v2). Read the file COPYING that comes with GRASS
+#               for details.
+#
+#############################################################################
+
+#%Module
+#%  description: Allows to update a column in the attribute table connected to a vector map.
+#%  keywords: vector, database, attribute table
+#%End
+#%option
+#% key: map
+#% type: string
+#% gisprompt: old,vector,vector
+#% description: Vector map to edit the attribute table for
+#% required : yes
+#%end
+#%option
+#% key: layer
+#% type: integer
+#% description: Layer to which the table to be changed is connected
+#% answer: 1
+#% required : no
+#%end
+#%option
+#% key: column
+#% type: string
+#% description: Column to update
+#% required : yes
+#%end
+#%option
+#% key: value
+#% type: string
+#% description: Value to update the column with, can be (combination of) other column(s)
+#% required : no
+#%end
+#%option
+#% key: qcolumn
+#% type: string
+#% description: Column to query
+#% required : no
+#%end
+#%option
+#% key: where
+#% type: string
+#% description: WHERE conditions for update, without 'where' keyword (e.g. cat=1 or col1/col2>1)
+#% required : no
+#%end
+
+import sys
+import os
+import grass
+
+def main():
+    map = options['map']
+    layer = options['layer']
+    column = options['column']
+    value = options['value']
+    qcolumn = options['qcolumn']
+    where = options['where']
+
+    mapset = grass.gisenv()['MAPSET']
+
+    # does map exist in CURRENT mapset?
+    if not grass.find_file(map, element = 'vector', mapset = mapset)['file']:
+	grass.fatal("Vector map '$GIS_OPT_MAP' not found in current mapset")
+
+    s = grass.read_command('v.db.connect', flags = 'g', map = map, layer = layer)
+    f = s.rstrip('\r\n').split(' ')
+    if len(s) < 2:
+	grass.fatal('There is no table connected to this map. Run v.db.connect or v.db.addtable first.')
+
+    table = f[1]
+    database = f[3]
+    driver = f[4]
+
+    # checking column types
+    coltype = None
+    s = grass.read_command('v.info', flags = 'c', map = map, quiet = True)
+    for l in s.splitlines():
+	f = l.split('|')
+	if len(f) < 2:
+	    continue
+	if f[1] == column:
+	    coltype = f[0]
+
+    if not coltype:
+	grass.fatal('column <%s> not found' % column)
+
+    if qcolumn:
+	if value:
+	    grass.fatal('value= and qcolumn= are mutually exclusive')
+	# special case: we copy from another column
+	value = qcolumn
+    else:
+	if not value:
+	    grass.fatal('Either value= or qcolumn= must be given')
+	# we insert a value
+	if coltype.upper() not in ["INTEGER", "DOUBLE PRECISION"]:
+	    value = "'%s'" % value
+
+    cmd = "UPDATE %s SET %s=%s" % (table, column, value)
+    if where:
+	cmd += " WHERE " + where
+
+    grass.write_command('db.execute', database = database, driver = driver, stdin = cmd)
+
+    # write cmd history:
+    grass.run_command('v.support', map = map, cmdhist = os.environ['CMDLINE'])
+
+if __name__ == "__main__":
+    options, flags = grass.parser()
+    main()


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