gistfile1.py

create or replace function update_asset_json(text, text, integer) returns text as $$
import simplejson
from datetime import datetime
now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
updated_data = simplejson.loads(args[1])
rs = plpy.execute("select * from assets where id in (%s)" % args[0])
resp = []
for r in rs:
    data = simplejson.loads(r['data'])
    data.update(updated_data) # apply the update(s)
    id = r['id']
    r.update(data)
    r['updated_on'] = now # update the timestamp in the results being returned
    r.pop('data') # Get rid of the data key as data is now in root hash
    resp.append(r)
    plpy.execute("update assets set updated_by=%d, updated_on='%s', data='%s' where id=%d" % (args[2], now, simplejson.dumps(data), id))
return simplejson.dumps(resp)
$$ language plpythonu;