aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar Vivaan Verma <[email protected]>2021-07-19 15:49:41 +0100
committerGravatar Vivaan Verma <[email protected]>2021-07-19 15:49:41 +0100
commitd8fbfffebfac31604c7b6fcf9b5b96a8e945791b (patch)
treee6becef46e0639de9ebece0405f36cefc7570336
parentFinished the CF cache purge command (diff)
Lint all code with black
-rw-r--r--arthur/apis/cloudflare/zones.py15
-rw-r--r--arthur/exts/cloudflare/zones.py5
-rw-r--r--arthur/exts/error_handler/error_handler.py3
-rw-r--r--arthur/exts/kubernetes/deployments.py4
4 files changed, 10 insertions, 17 deletions
diff --git a/arthur/apis/cloudflare/zones.py b/arthur/apis/cloudflare/zones.py
index 93c29d1..e2955f7 100644
--- a/arthur/apis/cloudflare/zones.py
+++ b/arthur/apis/cloudflare/zones.py
@@ -7,12 +7,11 @@ from arthur.config import CONFIG
CF_TOKEN = CONFIG.cloudflare_token
+
async def list_zones(zone_name: Optional[str] = None) -> dict[str, str]:
endpoint = f"https://api.cloudflare.com/client/v4/zones"
- request_headers = {
- "X-Auth-User-Service-Key": CF_TOKEN
- }
+ request_headers = {"X-Auth-User-Service-Key": CF_TOKEN}
if zone_name is not None:
endpoint += f"?name={zone_name}"
@@ -29,16 +28,10 @@ async def list_zones(zone_name: Optional[str] = None) -> dict[str, str]:
async def purge_zone(zone_identifier: str) -> dict:
endpoint = f"https://api.cloudflare.com/client/v4/zones/{zone_identifier}/purge_cache?purge_everything=true"
- request_headers = {
- "X-Auth-User-Service-Key": CF_TOKEN
- }
+ request_headers = {"X-Auth-User-Service-Key": CF_TOKEN}
async with aiohttp.ClientSession() as session:
async with session.post(endpoint, headers=request_headers) as response:
info = await response.json()
-
- return {
- "success": info["success"],
- "errors": info["errors"]
- }
+ return {"success": info["success"], "errors": info["errors"]}
diff --git a/arthur/exts/cloudflare/zones.py b/arthur/exts/cloudflare/zones.py
index 20169bc..3b8b3b7 100644
--- a/arthur/exts/cloudflare/zones.py
+++ b/arthur/exts/cloudflare/zones.py
@@ -6,8 +6,8 @@ from discord.ext import commands
from arthur.apis.cloudflare import zones
from arthur.bot import KingArthur
-class Zones(commands.Cog):
+class Zones(commands.Cog):
def __init__(self, bot: KingArthur) -> None:
self.bot = bot
@@ -38,6 +38,7 @@ class Zones(commands.Cog):
message += f"**Message**: {error.message}\n"
return await ctx.send(message)
+
def setup(bot: KingArthur) -> None:
"""Add the extension to the bot."""
- bot.add_cog(Zones(bot)) \ No newline at end of file
+ bot.add_cog(Zones(bot))
diff --git a/arthur/exts/error_handler/error_handler.py b/arthur/exts/error_handler/error_handler.py
index 2a61f40..673fbc7 100644
--- a/arthur/exts/error_handler/error_handler.py
+++ b/arthur/exts/error_handler/error_handler.py
@@ -49,8 +49,7 @@ class ErrorHandler(Cog):
await ctx.send(
generate_error_message(
description=(
- f"Unknown exception occurred: `{error.__class__.__name__}:"
- f" {error}`"
+ f"Unknown exception occurred: `{error.__class__.__name__}:" f" {error}`"
)
)
)
diff --git a/arthur/exts/kubernetes/deployments.py b/arthur/exts/kubernetes/deployments.py
index dccf676..dc1284f 100644
--- a/arthur/exts/kubernetes/deployments.py
+++ b/arthur/exts/kubernetes/deployments.py
@@ -134,14 +134,14 @@ class Deployments(commands.Cog):
content=generate_error_message(
description="Could not find deployment, check the namespace.",
),
- ephemeral=False
+ ephemeral=False,
)
return await interaction.respond(
content=generate_error_message(
description=f"Unexpected error occurred, error code {e.status}"
),
- ephemeral=False
+ ephemeral=False,
)
else:
description = (