abravalheri / validate-pyproject / 5226968556240896

import json

import logging

import sys

from functools import partial, wraps

from pathlib import Path

from types import MappingProxyType

from typing import Any, Dict, List, Mapping, NamedTuple, Sequence

from .. import cli

from ..plugins import PluginWrapper

from ..plugins import list_from_entry_points as list_plugins_from_entry_points

from . import pre_compile

_logger = logging.getLogger(__package__)

def JSON_dict(name: str, value: str):

    try:

        return ensure_dict(name, json.loads(value))

    except json.JSONDecodeError as ex:

        raise ValueError(f"Invalid JSON: {value}") from ex

META: Dict[str, dict] = {

def ensure_dict(name: str, value: Any) -> dict:

    if not isinstance(value, dict):

        msg = f"`{value.__class__.__name__}` given (value = {value!r})."

        raise ValueError(f"`{name}` should be a dict. {msg}")

    return value

class CliParams(NamedTuple):

    plugins: List[PluginWrapper]

    output_dir: Path = Path(".")

    main_file: str = "__init__.py"

    replacements: Mapping[str, str] = MappingProxyType({})

    loglevel: int = logging.WARNING

def parser_spec(plugins: Sequence[PluginWrapper]) -> Dict[str, dict]:

    common = ("version", "enable", "disable", "verbose", "very_verbose")

    cli_spec = cli.__meta__(plugins)

    meta = {k: v.copy() for k, v in META.items()}

    meta.update({k: cli_spec[k].copy() for k in common})

    return meta

def run(args: Sequence[str] = ()):

    args = args if args else sys.argv[1:]

    cmd = f"python -m {__package__} " + arg_join(args)

    plugins = list_plugins_from_entry_points()

    desc = 'Generate files for "pre-compiling" `validate-pyproject`'

    prms = cli.parse_args(args, plugins, desc, parser_spec, CliParams)

    cli.setup_logging(prms.loglevel)

    pre_compile(prms.output_dir, prms.main_file, cmd, prms.plugins, prms.replacements)

    return 0

main = cli.exceptions2exit()(run)