summaryrefslogtreecommitdiff
path: root/tests/londiste/loadgen.py
blob: 9efea50a1ba662da26b862ea80f701acd0c52487 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#! /usr/bin/env python

import sys

import pkgloader
pkgloader.require('skytools', '3.0')
import skytools

class LoadGen(skytools.DBScript):
    seq = 1
    def work(self):
        db = self.get_database('db', autocommit = 1)
        curs = db.cursor()
        data = 'data %d' % self.seq
        curs.execute('insert into mytable (data) values (%s)', [data])
        self.seq += 1

if __name__ == '__main__':
    script = LoadGen('loadgen', sys.argv[1:])
    script.start()