From 4c7be2154b24db6fada38116ce6db2ccc3200f12 Mon Sep 17 00:00:00 2001 From: Chris Lovering Date: Mon, 21 Aug 2023 15:54:35 +0100 Subject: Use pydantic-settings for config --- arthur/config.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'arthur/config.py') diff --git a/arthur/config.py b/arthur/config.py index 13ad241..ddfa479 100644 --- a/arthur/config.py +++ b/arthur/config.py @@ -1,16 +1,20 @@ """Utilities for interacting with the config for King Arthur.""" +from pydantic_settings import BaseSettings -from pydantic import BaseSettings - -class Config(BaseSettings): +class Config( + BaseSettings, + env_file=".env", + env_prefix="KING_ARTHUR_", + extra="ignore", +): """Configuration for King Arthur.""" # Discord bot token token: str # Discord bot prefix - prefixes: list[str] = ["arthur ", "M-x "] + prefixes: tuple[str] = ("arthur ", "M-x ") # Authorised role ID for usage devops_role: int = 409416496733880320 @@ -24,9 +28,5 @@ class Config(BaseSettings): # Token for authorising with the Notion API notion_api_token: str | None = None - class Config: # noqa: D106 - env_file = ".env" - env_prefix = "KING_ARTHUR_" - CONFIG = Config() -- cgit v1.2.3