diff --git a/pylama/config_toml.py b/pylama/config_toml.py index 2af02a5..ea6e17a 100644 --- a/pylama/config_toml.py +++ b/pylama/config_toml.py @@ -1,16 +1,21 @@ """Pylama TOML configuration.""" -import toml +import sys from pylama.libs.inirama import Namespace as _Namespace +if sys.version_info >= (3, 11): + import tomllib +else: + import tomli as tomllib + class Namespace(_Namespace): """Inirama-style wrapper for TOML config.""" def parse(self, source: str, update: bool = True, **params): """Parse TOML source as string.""" - content = toml.loads(source) + content = tomllib.loads(source) tool = content.get("tool", {}) pylama = tool.get("pylama", {}) linters = pylama.pop("linter", {}) diff --git a/requirements/requirements-tests.txt b/requirements/requirements-tests.txt index d786f1f..e62ccae 100644 --- a/requirements/requirements-tests.txt +++ b/requirements/requirements-tests.txt @@ -5,8 +5,7 @@ radon >= 5.1.0 mypy pylint >= 2.11.1 pylama-quotes -toml +tomli >= 1.2.3 ; python_version < "3.11" vulture types-setuptools -types-toml diff --git a/setup.py b/setup.py index 911aea6..6d0222b 100644 --- a/setup.py +++ b/setup.py @@ -21,6 +21,6 @@ def parse_requirements(path: str) -> "list[str]": extras_require=dict( tests=parse_requirements("requirements/requirements-tests.txt"), all=OPTIONAL_LINTERS, **{linter: [linter] for linter in OPTIONAL_LINTERS}, - toml="toml>=0.10.2", + toml="tomli>=1.2.3; python_version < '3.11'", ), )