Use tomllib on Python 3.11

flake8_black.py

@@ -4,11 +4,16 @@
 source code using the tool black.
 """
 
+import sys
 from os import path
 from pathlib import Path
 
+if sys.version_info >= (3, 11):
+    import tomllib
+else:
+    import tomli as tomllib
+
 import black
-import tomli
 
 from flake8 import utils as stdin_utils
 from flake8 import LOG
@@ -58,7 +63,7 @@ def load_black_mode(toml_filename=None):
     LOG.info("flake8-black: loading black settings from %s", toml_filename)
     try:
         with toml_filename.open(mode="rb") as toml_file:
-            pyproject_toml = tomli.load(toml_file)
+            pyproject_toml = tomllib.load(toml_file)
     except ValueError:
         LOG.info("flake8-black: invalid TOML file %s", toml_filename)
         raise BadBlackConfig(path.relpath(toml_filename))

pyproject.toml

@@ -29,7 +29,7 @@ requires-python = '>=3.7'
 dependencies = [
   'flake8>=3',
   'black>=22.1.0',
-  'tomli',
+  'tomli ; python_version < "3.11"',
 ]
 dynamic = ['version']
 [project.entry-points]

requirements.txt

@@ -4,4 +4,4 @@ flake8 >= 3.0.0
 
 # We need black, which in turn needs Python 3.6+
 black
-tomli
+tomli ; python_version < "3.11"