feat: update to handle new clustering setup

suggestionsbot · Dec 17, 2023 · 3e1f09f · 3e1f09f
1 parent e3a35aa
commit 3e1f09f
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 16 deletions.
diff --git a/suggestions/cogs/help_guild_cog.py b/suggestions/cogs/help_guild_cog.py
@@ -76,16 +76,26 @@ async def instance_info(
         shard_id = self.bot.get_shard_id(guild_id)
         cluster_id = (
             1
-            if shard_id < 10
+            if shard_id < 5
             else 2
-            if shard_id < 20
+            if shard_id < 10
             else 3
-            if shard_id < 30
+            if shard_id < 15
             else 4
-            if shard_id < 40
+            if shard_id < 20
             else 5
-            if shard_id < 50
+            if shard_id < 25
             else 6
+            if shard_id < 30
+            else 7
+            if shard_id < 35
+            else 8
+            if shard_id < 40
+            else 9
+            if shard_id < 45
+            else 10
+            if shard_id < 50
+            else 11
         )
 
         await interaction.send(
@@ -154,7 +164,7 @@ async def show_bot_status(
             title="Bot infrastructure status",
         )
         down_shards: list[str] = [str(i) for i in range(53)]
-        down_clusters: list[str] = [str(i) for i in range(1, 7)]
+        down_clusters: list[str] = [str(i) for i in range(1, 12)]
         avg_bot_latency: list[float] = []
         async with aiohttp.ClientSession(
             headers={"X-API-KEY": os.environ["GARVEN_API_KEY"]}

diff --git a/suggestions/main.py b/suggestions/main.py
@@ -29,7 +29,7 @@ async def create_bot(database_wrapper=None) -> SuggestionsBot:
         total_shards = 53
         cluster_id = int(os.environ["CLUSTER"])
         offset = cluster_id - 1
-        number_of_shards_per_cluster = 10
+        number_of_shards_per_cluster = 5
         shard_ids = [
             i
             for i in range(

diff --git a/suggestions/zonis_routes.py b/suggestions/zonis_routes.py
@@ -79,12 +79,3 @@ async def share_with_devs(self, title, description, sender):
         )
         embed.set_footer(text=f"Sender: {sender}")
         await channel.send(embed=embed)
-
-    @client.route()
-    async def refresh_premium(self, user_id: int):
-        # TODO Implement the ability to refresh premium states for a user
-        return True
-
-    @client.route()
-    async def shared_guilds(self, guild_ids: list[int]):
-        return [gid for gid in guild_ids if gid in self.bot.guild_ids]