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"]