From db81fd831e16f925d5c93d4c7d9b1fcc6a7bbffa Mon Sep 17 00:00:00 2001
From: Johannes Christ <jc@jchri.st>
Date: Wed, 14 Aug 2019 20:35:39 +0200
Subject: Add a database logging handler.

---
 pydis_site/apps/api/dblogger.py | 22 ++++++++++++++++++++++
 pydis_site/settings.py          |  5 ++++-
 2 files changed, 26 insertions(+), 1 deletion(-)
 create mode 100644 pydis_site/apps/api/dblogger.py

diff --git a/pydis_site/apps/api/dblogger.py b/pydis_site/apps/api/dblogger.py
new file mode 100644
index 00000000..f133832a
--- /dev/null
+++ b/pydis_site/apps/api/dblogger.py
@@ -0,0 +1,22 @@
+from logging import LogRecord, StreamHandler
+
+
+class DatabaseLogHandler(StreamHandler):
+    """Logs entries into the database."""
+
+    def emit(self, record: LogRecord):
+        """Write the given `record` into the database."""
+        # This import needs to be deferred due to Django's application
+        # registry instantiation logic loading this handler before the
+        # application is ready.
+        from pydis_site.apps.api.models.log_entry import LogEntry
+
+        entry = LogEntry(
+            application='site',
+            logger_name=record.name,
+            level=record.levelname.lower(),
+            module=record.module,
+            line=record.lineno,
+            message=self.format(record)
+        )
+        entry.save()
diff --git a/pydis_site/settings.py b/pydis_site/settings.py
index 2050c6ab..4a4eb94b 100644
--- a/pydis_site/settings.py
+++ b/pydis_site/settings.py
@@ -227,11 +227,14 @@ LOGGING = {
     'handlers': {
         'console': {
             'class': 'logging.StreamHandler'
+        },
+        'database': {
+            'class': 'pydis_site.apps.api.dblogger.DatabaseLogHandler'
         }
     },
     'loggers': {
         'django': {
-            'handlers': ['console'],
+            'handlers': ['console', 'database'],
             'propagate': True,
             'level': env(
                 'LOG_LEVEL',
-- 
cgit v1.2.3