From 850928ae382b33698cda07c3612b8cc6ee0d134b Mon Sep 17 00:00:00 2001 From: Chris Lovering Date: Sun, 21 Aug 2022 22:16:59 +0100 Subject: Use the bot's http_session when calling cloudflare's API --- arthur/exts/cloudflare/zones.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'arthur/exts/cloudflare/zones.py') 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) -- cgit v1.2.3