diff options
Diffstat (limited to 'arthur/exts/cloudflare/zones.py')
| -rw-r--r-- | arthur/exts/cloudflare/zones.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/arthur/exts/cloudflare/zones.py b/arthur/exts/cloudflare/zones.py index 657de71..5840d36 100644 --- a/arthur/exts/cloudflare/zones.py +++ b/arthur/exts/cloudflare/zones.py @@ -1,4 +1,5 @@ """The zones cog helps with managing Cloudflare zones.""" +import aiohttp import discord from discord.ext import commands @@ -11,10 +12,11 @@ from arthur.utils import generate_error_message class ZonesView(discord.ui.View): """This view allows users to select and purge the zones specified.""" - def __init__(self, domains: dict[str, str]) -> None: + def __init__(self, domains: dict[str, str], session: aiohttp.ClientSession) -> None: super().__init__() self.domains = domains + self.session = session for domain, zone_id in self.domains.items(): self.children[0].add_option(label=domain, value=domain, description=zone_id, emoji="🌐") @@ -37,7 +39,7 @@ class ZonesView(discord.ui.View): zone_name = dropdown.values[0] required_id = self.domains[zone_name] - purge_attempt_response = await zones.purge_zone(required_id) + purge_attempt_response = await zones.purge_zone(self.session, required_id) if purge_attempt_response["success"]: message = ":white_check_mark:" message += " **Cache cleared!** " @@ -69,9 +71,9 @@ class Zones(commands.Cog): @zones.command(name="purge") async def purge(self, ctx: commands.Context) -> None: """Command to clear the Cloudflare cache of the specified zone.""" - cf_zones = await zones.list_zones() + cf_zones = await zones.list_zones(self.bot.http_session) - view = ZonesView(cf_zones) + view = ZonesView(cf_zones, self.bot.http_session) await ctx.send(":cloud: Pick which zone(s) that should have their cache purged", view=view) |