summaryrefslogtreecommitdiffstats
path: root/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'utils.py')
-rw-r--r--utils.py19
1 files changed, 19 insertions, 0 deletions
diff --git a/utils.py b/utils.py
new file mode 100644
index 0000000..9fa3476
--- /dev/null
+++ b/utils.py
@@ -0,0 +1,19 @@
+from logging import Handler
+
+
+class SqlHandler(Handler):
+
+ def __init__(self, session):
+ Handler.__init__(self)
+ self.session = session
+ self.count = 0
+
+ def emit(self, record):
+ self.session.add(record.msg)
+ self.count += 1
+ if self.count >= 100:
+ self.session.commit()
+ self.count = 0
+
+ def close(self):
+ self.session.commit()