Use tomllib on Python 3.11

Author: jwodder

CHANGELOG.md

@@ -2,6 +2,7 @@ v0.4.0 (in development)
 -----------------------
 - Drop support for Python 3.6
 - Support Python 3.11
+- Use `tomllib` on Python 3.11
 
 v0.3.4 (2022-01-02)
 -------------------

setup.cfg

@@ -47,7 +47,7 @@ install_requires =
     packaging
     backports.cached-property; python_version < "3.8"
     pydantic         ~= 1.7
-    tomli            >= 1.2, < 3.0
+    tomli            >= 1.2, < 3.0; python_version < "3.11"
     wheel-filename   ~= 1.1
 
 [options.packages.find]

src/check_wheel_contents/config.py

@@ -2,14 +2,19 @@
 from collections.abc import Sequence
 from configparser import ConfigParser
 from pathlib import Path
+import sys
 from typing import Any, List, Optional, Set
 from pydantic import BaseModel, Field, ValidationError, validator
-import tomli
 from .checks import Check, parse_check_prefix
 from .errors import UserInputError
 from .filetree import Directory
 from .util import comma_split
 
+if sys.version_info[:2] >= (3, 11):
+    from tomllib import load as toml_load
+else:
+    from tomli import load as toml_load
+
 #: The filenames that configuration is read from by default, in descending
 #: order of preference
 CONFIG_FILES = [
@@ -167,7 +172,7 @@ def from_file(cls, path: Path) -> Optional["Configuration"]:
         """
         if path.suffix == ".toml":
             with path.open("rb") as fb:
-                tdata = tomli.load(fb)
+                tdata = toml_load(fb)
             tool = tdata.get("tool")
             if not isinstance(tool, dict):
                 return None