Initial commit, X events tracking

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()