From cfd1e5e61afdfa3415fa8dc47397e6f690fad985 Mon Sep 17 00:00:00 2001 From: kinoshitakenta Date: Wed, 28 May 2025 15:19:55 +0800 Subject: [PATCH] feat: validates required fields in login configuration Adds validation to ensure that all required fields ("lang", "login_ID", "login_passwd", "company_ID") are present in the login_info section of the configuration file. Raises a ValueError if any required field is missing, improving error handling and preventing potential issues due to incomplete configuration. --- utils/utils.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/utils/utils.py b/utils/utils.py index 13bc9f1..c3c4a01 100644 --- a/utils/utils.py +++ b/utils/utils.py @@ -17,6 +17,11 @@ class LoginInfo(): with open(config_path, mode="rb") as f: config = tomllib.load(f) + required_fields = ["lang", "login_ID", "login_passwd", "company_ID"] + for field in required_fields: + if field not in config.get("login_info", {}): + raise ValueError(f"Missing required field: {field} in login_info section") + self.lang = config["login_info"]["lang"] self.login_ID = config["login_info"]["login_ID"] self.login_passwd = config["login_info"]["login_passwd"]