pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-03-31 18:46:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use tomli and tomllib instead of toml

Browse source
This commit is contained in:
Daniël van Noord 2022-05-24 08:44:13 +02:00 • committed by Anthony Sottile
parent 6a5ae704e5
commit fc88f3fa49
2 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
pre_commit_hooks/check_toml.py
View file

@ -1,9 +1,13 @@
from __future__ import annotations
import argparse
import sys
from typing import Sequence
import toml
if sys.version_info >= (3, 11): # pragma: >=3.11 cover
import tomllib
else: # pragma: <3.11 cover
import tomli as tomllib
def main(argv: Sequence[str] | None = None) -> int:
@ -14,8 +18,9 @@ def main(argv: Sequence[str] | None = None) -> int:
retval = 0
for filename in args.filenames:
try:
toml.load(filename)
except toml.TomlDecodeError as exc:
with open(filename, mode='rb') as fp:
tomllib.load(fp)
except tomllib.TOMLDecodeError as exc:
print(f'{filename}: {exc}')
retval = 1
return retval