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