Got successful logging using webhooks and logging library

Author: MEhrn00

.env-example

@@ -3,3 +3,5 @@ DISCORD_SERVER_CHANNEL_ID=
 BOT_PREFIX_CHARACTER=
 RECAPTCHA_PUBLIC_KEY=
 RECAPTCHA_PRIVATE_KEY=
+LOGGING_WEBHOOK_CHANNEL=
+LOGGING_WEBHOOK_TOKEN=

JHDBot/bot.py

@@ -6,6 +6,8 @@
 import os
 import random
 import helpembed
+import logging
+from webhook_logging.discordHook import DiscordHandler
 #import configfile
 
 load_dotenv()
@@ -216,8 +218,22 @@ async def attach_embed_info(ctx=None, embed=None):
     embed.set_footer(text='by: JHD Moderation team ')
     return embed
 
-# Token
-TOKEN = os.getenv("DISCORD_API_TOKEN")
-bot.run(TOKEN)  # token
-print("Bot started press ctrl+c to exit....")
-#bot.run(configfile.bot_token)
+if __name__ == '__main__':
+    logger = logging.getLogger('Bot')
+    logger.setLevel(logging.INFO)
+
+    hookToken = os.getenv("LOGGING_WEBHOOK_TOKEN")
+    hookChannel = os.getenv("LOGGING_WEBHOOK_CHANNEL")
+    discordHandler = DiscordHandler(f'https://discordapp.com/api/webhooks/{hookChannel}/{hookToken}')
+    discordHandler.setLevel(logging.INFO)
+
+    formatter = logging.Formatter('[%(name)s] (%(levelname)s): %(message)s')
+    discordHandler.setFormatter(formatter)
+
+    logger.addHandler(discordHandler)
+
+    logger.info("Bot Started")
+
+    TOKEN = os.getenv("DISCORD_API_TOKEN")
+
+    bot.run(TOKEN)  # token

discord-webhooks/discord-webhooks/webhook.py

@@ -19,6 +19,8 @@ services:
     environment:
       - DISCORD_API_TOKEN=${DISCORD_API_TOKEN}
       - BOT_PREFIX_CHARACTER=${BOT_PREFIX_CHARACTER}
+      - LOGGING_WEBHOOK_CHANNEL=${LOGGING_WEBHOOK_CHANNEL}
+      - LOGGING_WEBHOOK_TOKEN=${LOGGING_WEBHOOK_TOKEN}
 
   webapp1:
     build: verify-web/.
@@ -29,6 +31,8 @@ services:
       - RECAPTCHA_PRIVATE_KEY=${RECAPTCHA_PRIVATE_KEY}
       - DISCORD_API_TOKEN=${DISCORD_API_TOKEN}
       - DISCORD_SERVER_CHANNEL_ID=${DISCORD_SERVER_CHANNEL_ID}
+      - LOGGING_WEBHOOK_CHANNEL=${LOGGING_WEBHOOK_CHANNEL}
+      - LOGGING_WEBHOOK_TOKEN=${LOGGING_WEBHOOK_TOKEN}
 
   webapp2:
     build: verify-web/.
@@ -39,3 +43,5 @@ services:
       - RECAPTCHA_PRIVATE_KEY=${RECAPTCHA_PRIVATE_KEY}
       - DISCORD_API_TOKEN=${DISCORD_API_TOKEN}
       - DISCORD_SERVER_CHANNEL_ID=${DISCORD_SERVER_CHANNEL_ID}
+      - LOGGING_WEBHOOK_CHANNEL=${LOGGING_WEBHOOK_CHANNEL}
+      - LOGGING_WEBHOOK_TOKEN=${LOGGING_WEBHOOK_TOKEN}

discord-webhooks/setup.py

@@ -9,5 +9,5 @@ Will eventually get uploaded to pypi but as of now requires a local install.
 
 Use:
 ```python
-import discord-webhooks
+import discord-logging
 ```

docker-compose.yml

@@ -0,0 +1,16 @@
+#!/usr/bin/env python3
+from setuptools import setup
+
+setup(
+    name='webhook_logging',
+    version='0.0.4',
+    description='Simple library for sending logs from python logging to webhooks',
+    url='https:/github.com/fumenoid/JHDBot/tree/webhook_logging/webhook_logging',
+    author='JHD dev team',
+    author_email='[email protected]',
+    packages=['webhook_logging'],
+    install_requires=[
+        'requests',
+        ],
+    classifiers=[],
+)

discord-webhooks/README.md webhook_logging/README.md

@@ -4,5 +4,5 @@
 Simple library to send discord webhooks
 '''
 
-__version__ = '0.0.1'
+__version__ = '0.0.3'
 __author__ = 'JHD Dev Team'

webhook_logging/setup.py

@@ -0,0 +1,42 @@
+#!/usr/bin/env python3
+import logging
+import requests
+import json
+
+
+class DiscordHandler(logging.Handler):
+    '''
+    Class for sending logging records to discord webhooks.
+    '''
+    def __init__(self, url: str) -> None:
+        """
+        Override the constructor because all of those stupid parameters aren't needed.
+        """
+        logging.Handler.__init__(self)
+
+        self.url = url
+        self.__testConnection(url)
+
+    def __testConnection(self, url: str) -> None:
+        r = requests.get(url)
+        if r.status_code != 200:
+            raise Exception(f"Could not establish connection for webhook logging. URL returned a {r.status_code}")
+        return
+
+    def mapLogRecord(self, record):
+        return self.format(record)
+
+    def emit(self, record):
+        """
+        Emit a record.
+
+        Send the record to the Web server as a percent-encoded dictionary
+        """
+        try:
+            url = self.url
+            logMsg = self.mapLogRecord(record)
+            data = json.dumps({"content": str(logMsg)})
+            headers = {"content-type": "application/json"}
+            r = requests.post(url, headers=headers, data=data)
+        except Exception:
+            self.handleError(record)