@db.register_function decorator, closes #162

This commit is contained in:
Simon Willison 2020-09-21 17:31:43 -07:00
commit 482477585a
3 changed files with 44 additions and 0 deletions

View file

@ -4,6 +4,7 @@ import contextlib
import datetime
import decimal
import hashlib
import inspect
import itertools
import json
import os
@ -143,6 +144,11 @@ class Database:
def __repr__(self):
return "<Database {}>".format(self.conn)
def register_function(self, fn):
name = fn.__name__
arity = len(inspect.signature(fn).parameters)
self.conn.create_function(name, arity, fn)
def execute(self, sql, parameters=None):
if self._tracer:
self._tracer(sql, parameters)