-#!/usr/bin/env python3
-
import asyncio
import errno
import json
async def _gen_check_output(
cmd: Sequence[str],
- timeout: float = 300,
+ timeout: float = 600,
env: Optional[Dict[str, str]] = None,
cwd: Optional[Path] = None,
) -> Tuple[bytes, bytes]:
async def black_run(
- repo_path: Path, project_config: Dict[str, Any], results: Results
+ repo_path: Path,
+ project_config: Dict[str, Any],
+ results: Results,
+ no_diff: bool = False,
) -> None:
"""Run Black and record failures"""
cmd = [str(which(BLACK_BINARY))]
if "cli_arguments" in project_config and project_config["cli_arguments"]:
- cmd.extend(*project_config["cli_arguments"])
- cmd.extend(["--check", "--diff", "."])
+ cmd.extend(project_config["cli_arguments"])
+ cmd.append("--check")
+ if no_diff:
+ cmd.append(".")
+ else:
+ cmd.extend(["--diff", "."])
with TemporaryDirectory() as tmp_path:
- # Prevent reading top-level user configs by manipulating envionment variables
+ # Prevent reading top-level user configs by manipulating environment variables
env = {
**os.environ,
"XDG_CONFIG_HOME": tmp_path, # Unix-like
long_checkouts: bool = False,
rebase: bool = False,
keep: bool = False,
+ no_diff: bool = False,
) -> None:
"""Check out project and run Black on it + record result"""
loop = asyncio.get_event_loop()
repo_path = await git_checkout_or_rebase(work_path, project_config, rebase)
if not repo_path:
continue
- await black_run(repo_path, project_config, results)
+ await black_run(repo_path, project_config, results, no_diff)
if not keep:
LOG.debug(f"Removing {repo_path}")
keep: bool = False,
long_checkouts: bool = False,
rebase: bool = False,
+ no_diff: bool = False,
) -> int:
"""
Process the queue with X workers and evaluate results
await asyncio.gather(
*[
project_runner(
- i, config, queue, work_path, results, long_checkouts, rebase, keep
+ i,
+ config,
+ queue,
+ work_path,
+ results,
+ long_checkouts,
+ rebase,
+ keep,
+ no_diff,
)
for i in range(workers)
]