Merge pull request #50 from sbrunner/toml

Use tomllib from Python standard lib, for version that provides it

pyproject.toml

@@ -31,7 +31,7 @@ detect-requirements = 'requirements_detector.run:run'
 python = ">=3.8,<4.0"
 astroid = "^3.0"
 packaging = ">=21.3"
-toml = "^0.10.2"
+toml = {version = "^0.10.2", python = "<3.11"}
 semver = "^3.0.0"
 
 [tool.poetry.dev-dependencies]

requirements_detector/detect.py

@@ -1,4 +1,5 @@
 import re
+import sys
 from pathlib import Path
 from typing import List, Union
 
@@ -9,6 +10,12 @@
 from .poetry_semver import parse_constraint
 from .requirement import DetectedRequirement
 
+_USE_TOMLLIB = sys.version_info.major > 3 or sys.version_info.minor >= 11
+if _USE_TOMLLIB:
+    import tomllib
+else:
+    import toml
+
 __all__ = [
     "find_requirements",
     "from_requirements_txt",
@@ -108,7 +115,11 @@ def from_pyproject_toml(toml_file: P) -> List[DetectedRequirement]:
     if isinstance(toml_file, str):
         toml_file = Path(toml_file)
 
-    parsed = toml.load(toml_file)
+    if _USE_TOMLLIB:
+        with open(toml_file, "rb") as toml_file_open:
+            parsed = tomllib.load(toml_file_open)
+    else:
+        parsed = toml.load(toml_file)
     poetry_section = parsed.get("tool", {}).get("poetry", {})
     dependencies = poetry_section.get("dependencies", {})
     dependencies.update(poetry_section.get("dev-dependencies", {}))