OSS-Fuzz integration (#1930)

[etc/vim.git] / src / black_primer / cli.py

diff --git a/src/black_primer/cli.py b/src/black_primer/cli.py
index 010ea6c7af2e26ec5c23c609760bada411db6e23..5903adc72d4867553c520ed1a9e05b5101c9bda4 100644 (file)
--- a/src/black_primer/cli.py
+++ b/src/black_primer/cli.py
@@ -1,10 +1,11 @@
 #!/usr/bin/env python3
 
 #!/usr/bin/env python3
 

+# coding=utf8
+

 import asyncio
 import logging
 import sys
 from datetime import datetime
 import asyncio
 import logging
 import sys
 from datetime import datetime

-from os import cpu_count

 from pathlib import Path
 from shutil import rmtree, which
 from tempfile import gettempdir
 from pathlib import Path
 from shutil import rmtree, which
 from tempfile import gettempdir


@@ -50,7 +51,7 @@ async def async_main(
         work_path.mkdir()
 
     if not which("black"):
         work_path.mkdir()
 
     if not which("black"):

-        LOG.error(f"Can not find 'black' executable in PATH. No point in running")
+        LOG.error("Can not find 'black' executable in PATH. No point in running")

         return -1
 
     try:
         return -1
 
     try:


@@ -61,9 +62,9 @@ async def async_main(
     finally:
         if not keep and work_path.exists():
             LOG.debug(f"Removing {work_path}")
     finally:
         if not keep and work_path.exists():
             LOG.debug(f"Removing {work_path}")

-            rmtree(work_path)
+            rmtree(work_path, onerror=lib.handle_PermissionError)

 
 

-    return -1
+    return -2

 
 
 @click.command(context_settings={"help_option_names": ["-h", "--help"]})
 
 
 @click.command(context_settings={"help_option_names": ["-h", "--help"]})


@@ -109,21 +110,21 @@ async def async_main(
     default=str(DEFAULT_WORKDIR),
     type=click.Path(exists=False),
     show_default=True,
     default=str(DEFAULT_WORKDIR),
     type=click.Path(exists=False),
     show_default=True,

-    help="Directory Path for repo checkouts",
+    help="Directory path for repo checkouts",

 )
 @click.option(
     "-W",
     "--workers",
 )
 @click.option(
     "-W",
     "--workers",

-    default=int((cpu_count() or 4) / 2) or 1,
+    default=2,

     type=int,
     show_default=True,
     help="Number of parallel worker coroutines",
 )
 @click.pass_context
 def main(ctx: click.core.Context, **kwargs: Any) -> None:
     type=int,
     show_default=True,
     help="Number of parallel worker coroutines",
 )
 @click.pass_context
 def main(ctx: click.core.Context, **kwargs: Any) -> None:

-    """primer - prime projects for blackening ... 🏴"""
+    """primer - prime projects for blackening... 🏴"""

     LOG.debug(f"Starting {sys.argv[0]}")
     LOG.debug(f"Starting {sys.argv[0]}")

-    # TODO: Change to asyncio.run when black >= 3.7 only
+    # TODO: Change to asyncio.run when Black >= 3.7 only

     loop = asyncio.get_event_loop()
     try:
         ctx.exit(loop.run_until_complete(async_main(**kwargs)))
     loop = asyncio.get_event_loop()
     try:
         ctx.exit(loop.run_until_complete(async_main(**kwargs)))


@@ -131,5 +132,5 @@ def main(ctx: click.core.Context, **kwargs: Any) -> None:
         loop.close()
 
 
         loop.close()
 
 

-if __name__ == "__main__":
+if __name__ == "__main__":  # pragma: nocover

     main()
     main()