better health check

ropc · Mar 2, 2024 · 26d6fcb · 26d6fcb
1 parent 6607159
commit 26d6fcb
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -24,5 +24,6 @@ COPY tests ./tests
 COPY meme_generator ./meme_generator
 COPY memebot ./memebot
 
+HEALTHCHECK CMD discordhealthcheck || exit 1
 ENTRYPOINT [ "python", "-m" ]
 CMD [ "memebot.memebot", "run" ]
diff --git a/memebot/memebot.py b/memebot/memebot.py
@@ -1,6 +1,7 @@
 import os
 import logging
 import discord
+import discordhealthcheck
 from discord.ext import commands
 from discord.ext.commands.view import StringView
 from pygtrie import CharTrie
@@ -55,6 +56,7 @@ async def setup_hook(self):
         await self.add_cog(Whisper(whisper_config_filepath=os.getenv('WHISPER_CONFIG_FILE_PATH')))
         await self.add_cog(Calendar())
         await self.add_cog(ChatGPT(bot_id=self.user.id))
+        self.healthcheck_server = await discordhealthcheck.start(self)
 
     async def on_command(self, context: commands.Context):
         log.debug(f'received command {context.command}')

diff --git a/pdm.lock b/pdm.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -35,6 +35,7 @@ dependencies = [
     "python-dateutil<3.0.0,>=2.8.2",
     "dateparser<2.0.0,>=1.2.0",
     "openai<2.0.0,>=1.12.0",
+    "discordhealthcheck>=0.1.1",
 ]
 requires-python = "<4.0,>=3.12"
 readme = "README.md"