/Doc/includes/sqlite3/mysumaggr.py

http://unladen-swallow.googlecode.com/ · Python · 20 lines · 16 code · 4 blank · 0 comment · 0 complexity · 2d3c0e1c5d46bf969b13571edb67a8ed MD5 · raw file

  1. import sqlite3
  2. class MySum:
  3. def __init__(self):
  4. self.count = 0
  5. def step(self, value):
  6. self.count += value
  7. def finalize(self):
  8. return self.count
  9. con = sqlite3.connect(":memory:")
  10. con.create_aggregate("mysum", 1, MySum)
  11. cur = con.cursor()
  12. cur.execute("create table test(i)")
  13. cur.execute("insert into test(i) values (1)")
  14. cur.execute("insert into test(i) values (2)")
  15. cur.execute("select mysum(i) from test")
  16. print cur.fetchone()[0]