From 33367e2be6dc6f08dc89a64b3e9adf35fb0f85e4 Mon Sep 17 00:00:00 2001
From: Isaac Beh <isaacwbeh@gmail.com>
Date: Thu, 7 Mar 2024 23:20:52 +1000
Subject: [PATCH] Added typing for member counter

---
 pyproject.toml            |  1 -
 uqcsbot/member_counter.py | 26 ++++++++++++++++++++++----
 2 files changed, 22 insertions(+), 5 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index f9548578..91963233 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -45,7 +45,6 @@ exclude = [
     "**/bot.py",
     "**/events.py",
     "**/gaming.py",
-    "**/member_counter.py",
     "**/remindme.py",
     "**/snailrace.py",
     "**/starboard.py",
diff --git a/uqcsbot/member_counter.py b/uqcsbot/member_counter.py
index a6cda54d..2e6939f2 100644
--- a/uqcsbot/member_counter.py
+++ b/uqcsbot/member_counter.py
@@ -7,13 +7,15 @@
 from discord import app_commands
 from discord.ext import commands
 
+from uqcsbot.bot import UQCSBot
+
 
 class MemberCounter(commands.Cog):
     MEMBER_COUNT_PREFIX = "Member Count: "
     RATE_LIMIT = timedelta(minutes=5)
     NEW_MEMBER_TIME = timedelta(days=7)
 
-    def __init__(self, bot: commands.Bot):
+    def __init__(self, bot: UQCSBot):
         self.bot = bot
         self.last_rename_time = datetime.now()
         self.waiting_for_rename = False
@@ -43,7 +45,13 @@ async def on_ready(self):
             )
             return
 
+        if self.bot.user is None:
+            logging.warning("Could not find the bot's own id.")
+            return
         bot_member = self.bot.uqcs_server.get_member(self.bot.user.id)
+        if bot_member is None:
+            logging.warning("Could not find the bot's user.")
+            return
         permissions = self.member_count_channel.permissions_for(bot_member)
         if not permissions.manage_channels:
             logging.warning(
@@ -55,17 +63,27 @@ async def on_ready(self):
     @app_commands.command(name="membercount")
     async def member_count(self, interaction: discord.Interaction, force: bool = False):
         """Display the number of members"""
+        if interaction.guild is None:
+            logging.warning(
+                "Could not update member count as could not access members."
+            )
+            return
         new_members = [
             member
             for member in interaction.guild.members
-            if member.joined_at
+            if member.joined_at is not None
+            and member.joined_at
             > datetime.now(tz=ZoneInfo("Australia/Brisbane")) - self.NEW_MEMBER_TIME
         ]
         await interaction.response.send_message(
             f"There are currently {interaction.guild.member_count} members in the UQ Computing Society discord server, with {len(new_members)} joining in the last 7 days."
         )
 
-        if interaction.user.guild_permissions.manage_guild and force:
+        if (
+            isinstance(interaction.user, discord.Member)
+            and interaction.user.guild_permissions.manage_guild
+            and force
+        ):
             # this is dodgy, but the alternative is to restart the bot
             # if it gets caught in a loop of waiting for a broken rename
             self.waiting_for_rename = False
@@ -107,5 +125,5 @@ async def _update_member_count_channel_name(self):
         self.waiting_for_rename = False
 
 
-async def setup(bot: commands.Bot):
+async def setup(bot: UQCSBot):
     await bot.add_cog(MemberCounter(bot))