diff options
Diffstat (limited to 'arthur')
| -rw-r--r-- | arthur/config.py | 2 | ||||
| -rw-r--r-- | arthur/exts/cloudflare/zones.py | 4 | 
2 files changed, 5 insertions, 1 deletions
diff --git a/arthur/config.py b/arthur/config.py index 80e4825..db92ad7 100644 --- a/arthur/config.py +++ b/arthur/config.py @@ -17,7 +17,7 @@ class Config(      token: pydantic.SecretStr      prefixes: tuple[str, ...] = ("arthur ", "M-x ") -    cloudflare_token: pydantic.SecretStr +    cloudflare_token: pydantic.SecretStr | None = None      youtube_api_key: pydantic.SecretStr | None = None      grafana_url: str = "https://grafana.pydis.wtf"      grafana_token: pydantic.SecretStr diff --git a/arthur/exts/cloudflare/zones.py b/arthur/exts/cloudflare/zones.py index 8a21ad1..6735f50 100644 --- a/arthur/exts/cloudflare/zones.py +++ b/arthur/exts/cloudflare/zones.py @@ -7,6 +7,7 @@ from discord.ext import commands  from arthur.apis.cloudflare import zones  from arthur.bot import KingArthur  from arthur.config import CONFIG +from arthur.log import logger  from arthur.utils import generate_error_message @@ -80,4 +81,7 @@ class Zones(commands.Cog):  async def setup(bot: KingArthur) -> None:      """Add the extension to the bot.""" +    if not CONFIG.cloudflare_token: +        logger.warning("Not loading Cloudflare Zone cog as cloudflare_token env var is not set.") +        return      await bot.add_cog(Zones(bot))  |