check_toml.py 560 字节
Newer Older
Semyon Maryasin's avatar
Semyon Maryasin 已提交
1
2
3
4
import argparse
from typing import Optional
from typing import Sequence

5
import toml
Semyon Maryasin's avatar
Semyon Maryasin 已提交
6
7


8
def main(argv: Optional[Sequence[str]] = None) -> int:
Semyon Maryasin's avatar
Semyon Maryasin 已提交
9
10
11
12
13
14
15
    parser = argparse.ArgumentParser()
    parser.add_argument('filenames', nargs='*', help='Filenames to check.')
    args = parser.parse_args(argv)

    retval = 0
    for filename in args.filenames:
        try:
16
            toml.load(filename)
17
        except toml.TomlDecodeError as exc:
18
            print(f'{filename}: {exc}')
Semyon Maryasin's avatar
Semyon Maryasin 已提交
19
20
21
22
23
            retval = 1
    return retval


if __name__ == '__main__':
24
    exit(main())