mirror of https://github.com/python/cpython
21 lines
470 B
Python
21 lines
470 B
Python
import sqlite3
|
|||
|
|||
class MySum:
|
|||
def __init__(self):
|
|||
self.count = 0
|
|||
|
|||
def step(self, value):
|
|||
self.count += value
|
|||
|
|||
def finalize(self):
|
|||
return self.count
|
|||
|
|||
con = sqlite3.connect(":memory:")
|
|||
con.create_aggregate("mysum", 1, MySum)
|
|||
cur = con.cursor()
|
|||
cur.execute("create table test(i)")
|
|||
cur.execute("insert into test(i) values (1)")
|
|||
cur.execute("insert into test(i) values (2)")
|
|||
cur.execute("select mysum(i) from test")
|
|||
print cur.fetchone()[0]
|
