From ef204f1aae0f15c7429971d2e23ef99de8794379 Mon Sep 17 00:00:00 2001 From: kinoshitakenta Date: Wed, 28 May 2025 13:19:31 +0800 Subject: [PATCH] chore: improved compatibility with Python 3.11 and above Uses `tomllib` when available for Python 3.11+ and falls back to `tomli` for older versions. This avoids deprecation warnings and ensures compatibility across different Python versions. --- utils/utils.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index 1c4ce29..13bc9f1 100644 --- a/utils/utils.py +++ b/utils/utils.py @@ -1,8 +1,12 @@ import getpass +import sys import time from pathlib import WindowsPath -import tomli +if sys.version_info >= (3, 11): + import tomllib +else: + import tomli as tomllib from selenium import webdriver from selenium.webdriver.common.by import By from selenium.webdriver.support.ui import Select @@ -11,7 +15,7 @@ from selenium.webdriver.support.ui import Select class LoginInfo(): def __init__(self, config_path: WindowsPath): with open(config_path, mode="rb") as f: - config = tomli.load(f) + config = tomllib.load(f) self.lang = config["login_info"]["lang"] self.login_ID = config["login_info"]["login_ID"]