-
Notifications
You must be signed in to change notification settings - Fork 0
/
logging_db.py
48 lines (38 loc) · 1.28 KB
/
logging_db.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
# Original source by giumas
# https://gist.github.com/giumas/994e48d3c1cff45fbe93
import logging
import time
import emoji
insertion_sql = (
"INSERT INTO log (timestamp, log_level, user_status, user_key, type, content, response, func_name)"
"VALUES (%(dbtime)s, %(levelno)s, %(user_status)s, %(user_key)s, %(type)s, %(content)s, %(msg)s, %(funcName)s)"
)
class MySQLHandler(logging.Handler):
"""
Thread-safe logging handler for SQLite.
"""
def __init__(self, conn):
logging.Handler.__init__(self)
self.conn = conn
self.cur = self.conn.cursor()
def format_time(self, record):
"""
Create a time stamp
"""
record.dbtime = int(time.time())
def emit(self, record):
self.format(record)
self.format_time(record)
record.content = emoji.demojize(record.content)
if len(record.content) > 255:
record.content = record.content[:254]
if record.exc_info: # for exceptions
record.exc_text = logging._defaultFormatter.formatException(record.exc_info)
else:
record.exc_text = ""
# Insert the log record
self.cur.execute(insertion_sql, record.__dict__)
self.conn.commit()
if __name__ == '__main__':
# main()
pass