From 6bbf5e7230715f5d26289354d76221f276fa15fa Mon Sep 17 00:00:00 2001
From: MagicTheDev <matthewanderson3607@gmail.com>
Date: Tue, 17 Dec 2024 07:23:03 -0600
Subject: [PATCH] fix button click

---
 background/logs/war.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/background/logs/war.py b/background/logs/war.py
index a62a2fce..ababbfc6 100644
--- a/background/logs/war.py
+++ b/background/logs/war.py
@@ -247,12 +247,14 @@ async def on_button_click(self, ctx: disnake.MessageInteraction):
             await ctx.response.defer(ephemeral=True)
             war_id = (str(ctx.data.custom_id).split('_'))[-1]
             clan_tag = (str(ctx.data.custom_id).split('_'))[1]
-            war_data = await self.bot.clan_wars.find_one({'war_id': war_id})
+            war_data = await self.bot.clan_wars.find_one({"$and" : [{'war_id': war_id}, {"data" : {"$ne" : None}}]})
             war = None
             if war_data is not None:
                 war = coc.ClanWar(data=war_data['data'], client=self.bot.coc_client, clan_tag=clan_tag)
             if war is None:
                 war = await self.bot.get_clanwar(clanTag=clan_tag)
+                if war_id != f'{war.clan.tag}v{war.opponent.tag}-{int(war.preparation_start_time.time.timestamp())}':
+                    war = None
             if war is None:
                 return await ctx.send(content='No War Found', ephemeral=True)
             attack_embed: disnake.Embed = await attacks_embed(bot=self.bot, war=war)
@@ -262,12 +264,14 @@ async def on_button_click(self, ctx: disnake.MessageInteraction):
             await ctx.response.defer(ephemeral=True)
             war_id = (str(ctx.data.custom_id).split('_'))[-1]
             clan_tag = (str(ctx.data.custom_id).split('_'))[1]
-            war_data = await self.bot.clan_wars.find_one({'war_id': war_id})
+            war_data = await self.bot.clan_wars.find_one({"$and" : [{'war_id': war_id}, {"data" : {"$ne" : None}}]})
             war = None
             if war_data is not None:
                 war = coc.ClanWar(data=war_data['data'], client=self.bot.coc_client, clan_tag=clan_tag)
             if war is None:
                 war = await self.bot.get_clanwar(clanTag=clan_tag)
+                if war_id != f'{war.clan.tag}v{war.opponent.tag}-{int(war.preparation_start_time.time.timestamp())}':
+                    war = None
             if war is None:
                 return await ctx.send(content='No War Found', ephemeral=True)
             attack_embed = await defenses_embed(bot=self.bot, war=war)
@@ -277,12 +281,14 @@ async def on_button_click(self, ctx: disnake.MessageInteraction):
             await ctx.response.defer()
             war_id = (str(ctx.data.custom_id).split('_'))[-1]
             clan_tag = (str(ctx.data.custom_id).split('_'))[1]
-            war_data = await self.bot.clan_wars.find_one({'war_id': war_id})
+            war_data = await self.bot.clan_wars.find_one({"$and" : [{'war_id': war_id}, {"data" : {"$ne" : None}}]})
             war = None
             if war_data is not None:
                 war = coc.ClanWar(data=war_data['data'], client=self.bot.coc_client, clan_tag=clan_tag)
             if war is None:
                 war = await self.bot.get_clanwar(clanTag=clan_tag)
+                if war_id != f'{war.clan.tag}v{war.opponent.tag}-{int(war.preparation_start_time.time.timestamp())}':
+                    war = None
             if war is None:
                 return await ctx.send(content='No War Found', ephemeral=True)
             embed = await main_war_page(bot=self.bot, war=war)
@@ -293,12 +299,14 @@ async def on_button_click(self, ctx: disnake.MessageInteraction):
             await ctx.response.defer(ephemeral=True)
             war_id = (str(ctx.data.custom_id).split('_'))[-1]
             clan_tag = (str(ctx.data.custom_id).split('_'))[1]
-            war_data = await self.bot.clan_wars.find_one({'war_id': war_id})
+            war_data = await self.bot.clan_wars.find_one({"$and" : [{'war_id': war_id}, {"data" : {"$ne" : None}}]})
             war = None
             if war_data is not None:
                 war = coc.ClanWar(data=war_data['data'], client=self.bot.coc_client, clan_tag=clan_tag)
             if war is None:
                 war = await self.bot.get_clanwar(clanTag=clan_tag)
+                if war_id != f'{war.clan.tag}v{war.opponent.tag}-{int(war.preparation_start_time.time.timestamp())}':
+                    war = None
             if war is None:
                 return await ctx.send(content='No War Found', ephemeral=True)