diff -r c2e6e24b83d9 -r d59c473c9ad7 batchupdate.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/batchupdate.py Fri Mar 23 14:54:04 2012 +0100 @@ -0,0 +1,32 @@ +#!/usr/bin/env python3.2 + +import time + +from pgtoolkit import toolbase +from pgtoolkit.highlight import highlight + + +class BatchUpdateTool(toolbase.SimpleTool): + def __init__(self): + toolbase.SimpleTool.__init__(self, name='batchupdate', desc='Run an UPDATE query until it does not update any rows.') + self.parser.add_argument('--query', dest='query', type=str, help='Query to run. It should be constructed using subquery with LIMIT.') + self.parser.add_argument('--sleep', dest='sleep', default=0, type=int, help='Sleep SLEEP seconds between subsequent queries.') + self.init() + + def main(self): + # connect DB + with self.pgm.cursor('target') as curs: + rowcount = 1 + while rowcount > 0: + print('query:', self.args.query) + curs.execute(self.args.query, []) + rowcount = curs.rowcount + print('updated', rowcount) + curs.connection.commit() + print('sleep %s seconds' % self.args.sleep) + time.sleep(self.args.sleep) + + +tool = BatchUpdateTool() +tool.main() +