wi1-bot 2.0.2__py3-none-any.whl → 2.0.4__py3-none-any.whl

wi1_bot/discord/bot.py

@@ -1,7 +1,6 @@
 import asyncio
 import logging
 import traceback
-from typing import Any
 
 import discord
 from discord.ext import commands
@@ -21,12 +20,14 @@ sonarr = Sonarr(config["sonarr"]["url"], config["sonarr"]["api_key"])
 
 
 @bot.check
-async def check_channel(ctx: commands.Context[Any]) -> bool:
+async def check_channel(ctx: commands.Context[commands.Bot]) -> bool:
     return ctx.channel.id == config["discord"]["channel_id"]
 
 
 @bot.event
-async def on_command_error(ctx: commands.Context[Any], error: commands.CommandError) -> None:
+async def on_command_error(
+    ctx: commands.Context[commands.Bot], error: commands.CommandError
+) -> None:
     match error:
         case commands.CommandNotFound() | commands.CheckFailure():
             pass
@@ -63,13 +64,13 @@ async def on_ready() -> None:
 
 
 @bot.before_invoke
-async def before_invoke(ctx: commands.Context[Any]) -> None:
+async def before_invoke(ctx: commands.Context[commands.Bot]) -> None:
     logger.info(f"got command from {ctx.message.author}: {ctx.message.content}")
 
 
 @bot.command(name="downloads", aliases=["queue", "q"], help="see the status of movie downloads")
 @commands.cooldown(1, 10)
-async def downloads_cmd(ctx: commands.Context[Any]) -> None:
+async def downloads_cmd(ctx: commands.Context[commands.Bot]) -> None:
     async with ctx.typing():
         queue = radarr.get_downloads() + sonarr.get_downloads()
 
@@ -84,7 +85,7 @@ async def downloads_cmd(ctx: commands.Context[Any]) -> None:
 
 @bot.command(name="quota", help="see your used space on the plex")
 @commands.cooldown(1, 60, commands.BucketType.user)
-async def quota_cmd(ctx: commands.Context[Any]) -> None:
+async def quota_cmd(ctx: commands.Context[commands.Bot]) -> None:
     async with ctx.typing():
         used = (
             radarr.get_quota_amount(ctx.message.author.id)
@@ -98,17 +99,14 @@ async def quota_cmd(ctx: commands.Context[Any]) -> None:
 
         pct = used / maximum * 100 if maximum != 0 else 100
 
-        msg = (
-            f"you have added {used:.2f}/{maximum:.2f} GB ({pct:.1f}%) of useless crap"
-            " to the plex"
-        )
+        msg = f"you have added {used:.2f}/{maximum:.2f} GB ({pct:.1f}%) of useless crap to the plex"
 
     await reply(ctx.message, msg)
 
 
 @bot.command(name="quotas", help="see everyone's used space on the plex")
 @commands.cooldown(1, 60)
-async def quotas_cmd(ctx: commands.Context[Any]) -> None:
+async def quotas_cmd(ctx: commands.Context[commands.Bot]) -> None:
     if "quotas" not in config["discord"]:
         await reply(ctx.message, "quotas are not implemented here")
         return
@@ -139,7 +137,7 @@ async def quotas_cmd(ctx: commands.Context[Any]) -> None:
 
 @bot.command(name="addtag", help="add a user tag")
 @commands.has_role("plex-admin")
-async def addtag_cmd(ctx: commands.Context[Any], name: str, user: discord.Member) -> None:
+async def addtag_cmd(ctx: commands.Context[commands.Bot], name: str, user: discord.Member) -> None:
     tag = f"{name}: {user.id}"
 
     radarr.create_tag(tag)

wi1_bot/discord/cogs/movie.py

@@ -1,6 +1,5 @@
 import asyncio
 import logging
-from typing import Any
 
 from discord.ext import commands
 
@@ -18,7 +17,7 @@ class MovieCog(commands.Cog):
         self.radarr = Radarr(config["radarr"]["url"], config["radarr"]["api_key"])
 
     @commands.command(name="addmovie", help="add a movie to the plex")
-    async def addmovie_cmd(self, ctx: commands.Context[Any], *, query: str = "") -> None:
+    async def addmovie_cmd(self, ctx: commands.Context[commands.Bot], *, query: str = "") -> None:
         if not query:
             await reply(ctx.message, "usage: !addmovie KEYWORDS...")
             return
@@ -71,7 +70,7 @@ class MovieCog(commands.Cog):
             await ctx.send(f"hey <@!{config['discord']['admin_id']}> get this guy a tag")
 
     @commands.command(name="delmovie", help="delete a movie from the plex")
-    async def delmovie_cmd(self, ctx: commands.Context[Any], *, query: str = "") -> None:
+    async def delmovie_cmd(self, ctx: commands.Context[commands.Bot], *, query: str = "") -> None:
         if not query:
             await reply(ctx.message, "usage: !delmovie KEYWORDS...")
             return

wi1_bot/discord/cogs/series.py

@@ -1,6 +1,5 @@
 import asyncio
 import logging
-from typing import Any
 
 from discord.ext import commands
 
@@ -19,7 +18,7 @@ class SeriesCog(commands.Cog):
 
     @commands.command(name="addshow", help="add a show to the plex")
     @commands.has_any_role("plex-admin", "plex-shows")
-    async def addshow_cmd(self, ctx: commands.Context[Any], *, query: str = "") -> None:
+    async def addshow_cmd(self, ctx: commands.Context[commands.Bot], *, query: str = "") -> None:
         if not query:
             await reply(ctx.message, "usage: !addshow KEYWORDS...")
             return
@@ -91,7 +90,9 @@ class SeriesCog(commands.Cog):
 
     @commands.command(name="delshow", help="delete a show from the plex")
     @commands.has_any_role("plex-admin", "plex-shows")
-    async def delshow_command(self, ctx: commands.Context[Any], *, query: str = "") -> None:
+    async def delshow_command(
+        self, ctx: commands.Context[commands.Bot], *, query: str = ""
+    ) -> None:
         if not query:
             await reply(ctx.message, "usage: !delshow KEYWORDS...")
             return

wi1_bot/transcoder/transcode_queue.py

@@ -1,3 +1,4 @@
+import os
 from typing import Any, cast
 
 from mongoengine import Document, connect
@@ -23,7 +24,7 @@ class TranscodeItem(Document):
 
 class TranscodeQueue:
     def __init__(self) -> None:
-        connect("wi1_bot", connect=False)
+        connect("wi1_bot", connect=True, host=os.environ.get("MONGODB_CONNECTION_STRING", None))
 
     def add(
         self,

wi1_bot/webhook.py

@@ -103,6 +103,11 @@ def index() -> Any:
     return "", 200
 
 
+@app.route("/health", methods=["GET"])
+def health() -> Any:
+    return "OK", 200
+
+
 def start() -> None:
     logger.info("starting webhook listener")
 

{wi1_bot-2.0.2.dist-info → wi1_bot-2.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wi1-bot
-Version: 2.0.2
+Version: 2.0.4
 Summary: Discord bot for Radarr/Sonarr integration
 Project-URL: Homepage, https://github.com/wthueb/wi1-bot
 Author-email: William Huebner <wilhueb@gmail.com>

{wi1_bot-2.0.2.dist-info → wi1_bot-2.0.4.dist-info}/RECORD

@@ -1,7 +1,7 @@
 wi1_bot/__init__.py,sha256=dYlvBffClqwInBRSxSAoJHuujuh4yuPderFR4n7Odko,161
 wi1_bot/config.py,sha256=ndL3BxiOpXiDQGYNEIPtuusekS25miPFQrEphINxnsQ,2226
 wi1_bot/push.py,sha256=6EwQ1eXcJnQimAi0dOm1_t7Am056zob-1-EOkytGZWg,783
-wi1_bot/webhook.py,sha256=xTN_-Ivdt1ANJFWkhF4W8csyqwfBNH33d-eWbsWOPS4,3482
+wi1_bot/webhook.py,sha256=VtTInHBmmLHtCCXViQylyY0LuxXVdb34dmsBkipXZ-k,3565
 wi1_bot/arr/__init__.py,sha256=-6UE81yY6OhD6-nbLindChE_HlwsNRuL8JijUmo_Q6k,149
 wi1_bot/arr/download.py,sha256=02AYFglnFdWSG8xj_TaJc6l2wjybyhUW6F97CnoyUFw,1381
 wi1_bot/arr/episode.py,sha256=0K_auUYwkBBd_Ludc4-4z7lSW8HRQ3mchFQ7FR10JSU,1041
@@ -10,21 +10,21 @@ wi1_bot/arr/radarr.py,sha256=jxgNuCZY7aGMWK3nTLt8KLntiYcLaAF7g9dJoThsCvc,5721
 wi1_bot/arr/sonarr.py,sha256=OouuorRHBMFAOy5oDeJZ5H6WZ-95-N01EHtpOTVMTCo,5970
 wi1_bot/arr/util.py,sha256=y4_Dotm-pSJJT8xvPWy7ZoGMGVI-YOU4zK2LNTSoTB0,1025
 wi1_bot/discord/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wi1_bot/discord/bot.py,sha256=VEoe7ST9X7PYnFL5yj1x9hodeAYLMGcjWlPkZSE-ujI,4924
+wi1_bot/discord/bot.py,sha256=qLM0GK4oaBzui_OBTA3X3XcC55gbng3qjU7GFz3Y6xo,4931
 wi1_bot/discord/helpers.py,sha256=7Qr5Kr8H2CQoLiGG7Oqh388jqUCMJH0EBvF8fC-IbJo,2300
 wi1_bot/discord/cogs/__init__.py,sha256=9nA47jEyuGG7s1UJYz5SJASJcs0Xa3M1rNz9BiyCico,95
-wi1_bot/discord/cogs/movie.py,sha256=Gr8xbsdNDlKkOb7boiTE5n3YHz7FR0ajSRANV8EM3XY,3950
-wi1_bot/discord/cogs/series.py,sha256=bPA3LtuK78qkYvyN38ZdNQ_6I3j4qVN9L5JHV8KX_GA,4597
+wi1_bot/discord/cogs/movie.py,sha256=YCDEjswEI35P1dBuuCDGikgR6yPagIIZkx50MzTbRFU,3945
+wi1_bot/discord/cogs/series.py,sha256=k0fiC9cluSiedlRkpNdzIuBo59ZmfvbNWPqViS5PCL8,4606
 wi1_bot/scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wi1_bot/scripts/add_tag.py,sha256=mWwo8egk2Y5XRiQCpfkA11-3rcxZoD0JOJKxV0LguLk,586
 wi1_bot/scripts/rescan.py,sha256=97oGK1Gr3jBjfX-Eil4C0hDIwtpZ34E_rwTYCI8Z7r4,1474
 wi1_bot/scripts/start.py,sha256=TfEKCxS_95L5ZTi8OY45ywsgNnKlsqLDN6YesowifvM,2527
 wi1_bot/scripts/transcode_item.py,sha256=21MeeIZ9wIRhAY-FOEgOdPsg6YOLlSUj59r8NkTfixw,1155
 wi1_bot/transcoder/__init__.py,sha256=B4xr82UtIFc3tyy_MEZdZKMukYW0yejPnfsGowaTIM0,105
-wi1_bot/transcoder/transcode_queue.py,sha256=W7r2I7OD8QuxseCEb7_ddOvYXiBBLmKLvCs2IgJJ92I,1856
+wi1_bot/transcoder/transcode_queue.py,sha256=ixENVEfHeXqKFHIJaNHFZ_i5ctqnSQgqDwimi4rec_Y,1921
 wi1_bot/transcoder/transcoder.py,sha256=bU2oKQckBWWppJOH9W8vk_ql2uz3qs8O63wAYKH1bbk,9978
-wi1_bot-2.0.2.dist-info/METADATA,sha256=Oy614tRy_DoW7yAi7dKmIFtMFXP714GZ1iYm_Vr-P6U,3699
-wi1_bot-2.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-wi1_bot-2.0.2.dist-info/entry_points.txt,sha256=pF5EawAQsUNMHs5exynpNdRq9g4dODg-RaPkx2MyYLU,184
-wi1_bot-2.0.2.dist-info/licenses/LICENSE,sha256=6V4_mQoPoLJl77_WMsQRQMDG2mnIhDUdfCmMkqM9Qwc,1072
-wi1_bot-2.0.2.dist-info/RECORD,,
+wi1_bot-2.0.4.dist-info/METADATA,sha256=glPVd0NDc2IpVRC3IP3X1aAZMRFYOkOYSt3U7Vy9DCk,3699
+wi1_bot-2.0.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+wi1_bot-2.0.4.dist-info/entry_points.txt,sha256=pF5EawAQsUNMHs5exynpNdRq9g4dODg-RaPkx2MyYLU,184
+wi1_bot-2.0.4.dist-info/licenses/LICENSE,sha256=6V4_mQoPoLJl77_WMsQRQMDG2mnIhDUdfCmMkqM9Qwc,1072
+wi1_bot-2.0.4.dist-info/RECORD,,