from subprocess import CalledProcessError
from sys import version_info
from tempfile import TemporaryDirectory
-from typing import Any, Callable, Dict, NamedTuple, Optional, Sequence, Tuple
+from typing import (
+ Any,
+ Callable,
+ Dict,
+ List,
+ NamedTuple,
+ Optional,
+ Sequence,
+ Tuple,
+ Union,
+)
from urllib.parse import urlparse
import click
+TEN_MINUTES_SECONDS = 600
WINDOWS = system() == "Windows"
BLACK_BINARY = "black.exe" if WINDOWS else "black"
GIT_BINARY = "git.exe" if WINDOWS else "git"
async def _gen_check_output(
cmd: Sequence[str],
- timeout: float = 600,
+ timeout: float = TEN_MINUTES_SECONDS,
env: Optional[Dict[str, str]] = None,
cwd: Optional[Path] = None,
stdin: Optional[bytes] = None,
return results.stats["failed"]
+def _flatten_cli_args(cli_args: List[Union[Sequence[str], str]]) -> List[str]:
+ """Allow a user to put long arguments into a list of strs
+ to make the JSON human readable"""
+ flat_args = []
+ for arg in cli_args:
+ if isinstance(arg, str):
+ flat_args.append(arg)
+ continue
+
+ args_as_str = "".join(arg)
+ flat_args.append(args_as_str)
+
+ return flat_args
+
+
async def black_run(
project_name: str,
repo_path: Optional[Path],
stdin_test = project_name.upper() == "STDIN"
cmd = [str(which(BLACK_BINARY))]
if "cli_arguments" in project_config and project_config["cli_arguments"]:
- cmd.extend(project_config["cli_arguments"])
+ cmd.extend(_flatten_cli_args(project_config["cli_arguments"]))
cmd.append("--check")
if not no_diff:
cmd.append("--diff")
if stdin_test:
cmd.append("-")
stdin = repo_path.read_bytes()
+ elif "base_path" in project_config:
+ cmd.append(project_config["base_path"])
else:
cmd.append(".")
+ timeout = (
+ project_config["timeout_seconds"]
+ if "timeout_seconds" in project_config
+ else TEN_MINUTES_SECONDS
+ )
with TemporaryDirectory() as tmp_path:
# Prevent reading top-level user configs by manipulating environment variables
env = {
cwd_path = repo_path.parent if stdin_test else repo_path
try:
+ LOG.debug(f"Running black for {project_name}: {' '.join(cmd)}")
_stdout, _stderr = await _gen_check_output(
- cmd, cwd=cwd_path, env=env, stdin=stdin
+ cmd, cwd=cwd_path, env=env, stdin=stdin, timeout=timeout
)
except asyncio.TimeoutError:
results.stats["failed"] += 1
{
- "configuration_format_version": 20200509,
+ "configuration_format_version": 20210815,
"projects": {
"STDIN": {
"cli_arguments": ["--experimental-string-processing"],
"long_checkout": false,
"py_versions": ["all"]
},
+ "cpython": {
+ "disabled": true,
+ "disabled_reason": "To big / slow for GitHub Actions but handy to keep config to use manually or in some other CI in the future",
+ "base_path": "Lib",
+ "cli_arguments": [
+ "--experimental-string-processing",
+ "--extend-exclude",
+ [
+ "Lib/lib2to3/tests/data/different_encoding.py",
+ "|Lib/lib2to3/tests/data/false_encoding.py",
+ "|Lib/lib2to3/tests/data/py2_test_grammar.py",
+ "|Lib/test/bad_coding.py",
+ "|Lib/test/bad_coding2.py",
+ "|Lib/test/badsyntax_3131.py",
+ "|Lib/test/badsyntax_pep3120.py",
+ "|Lib/test/test_base64.py",
+ "|Lib/test/test_exceptions.py",
+ "|Lib/test/test_grammar.py",
+ "|Lib/test/test_named_expressions.py",
+ "|Lib/test/test_patma.py",
+ "|Lib/test/test_tokenize.py",
+ "|Lib/test/test_xml_etree.py",
+ "|Lib/traceback.py"
+ ]
+ ],
+ "expect_formatting_changes": true,
+ "git_clone_url": "https://github.com/python/cpython.git",
+ "long_checkout": false,
+ "py_versions": ["3.9", "3.10"],
+ "timeout_seconds": 900
+ },
"django": {
"cli_arguments": [
"--experimental-string-processing",
)
self.assertEqual(2, results.stats["failed"])
+ def test_flatten_cli_args(self) -> None:
+ fake_long_args = ["--arg", ["really/", "|long", "|regex", "|splitup"], "--done"]
+ expected = ["--arg", "really/|long|regex|splitup", "--done"]
+ self.assertEqual(expected, lib._flatten_cli_args(fake_long_args))
+
@event_loop()
def test_gen_check_output(self) -> None:
loop = asyncio.get_event_loop()
@patch("sys.stdout", new_callable=StringIO)
@event_loop()
def test_process_queue(self, mock_stdout: Mock) -> None:
+ """Test the process queue on primer itself
+ - If you have non black conforming formatting in primer itself this can fail"""
loop = asyncio.get_event_loop()
config_path = Path(lib.__file__).parent / "primer.json"
with patch("black_primer.lib.git_checkout_or_rebase", return_false):