Set Docker to use 3.11 for now (#3927)

[etc/vim.git] / src / black / concurrency.py

diff --git a/src/black/concurrency.py b/src/black/concurrency.py
index 1598f51e43f5a08b4ed428ebb15413798a0e3991..ce01657839981794b5ee1a2995cf079ac74298c1 100644 (file)
--- a/src/black/concurrency.py
+++ b/src/black/concurrency.py
@@ -17,7 +17,7 @@ from typing import Any, Iterable, Optional, Set
 from mypy_extensions import mypyc_attr
 
 from black import WriteBack, format_file_in_place
 from mypy_extensions import mypyc_attr
 
 from black import WriteBack, format_file_in_place

-from black.cache import Cache, filter_cached, read_cache, write_cache
+from black.cache import Cache

 from black.mode import Mode
 from black.output import err
 from black.report import Changed, Report
 from black.mode import Mode
 from black.output import err
 from black.report import Changed, Report


@@ -80,7 +80,8 @@ def reformat_many(
 
     executor: Executor
     if workers is None:
 
     executor: Executor
     if workers is None:

-        workers = os.cpu_count() or 1
+        workers = int(os.environ.get("BLACK_NUM_WORKERS", 0))
+        workers = workers or os.cpu_count() or 1

     if sys.platform == "win32":
         # Work around https://bugs.python.org/issue26903
         workers = min(workers, 60)
     if sys.platform == "win32":
         # Work around https://bugs.python.org/issue26903
         workers = min(workers, 60)


@@ -132,10 +133,9 @@ async def schedule_formatting(
     `write_back`, `fast`, and `mode` options are passed to
     :func:`format_file_in_place`.
     """
     `write_back`, `fast`, and `mode` options are passed to
     :func:`format_file_in_place`.
     """

-    cache: Cache = {}
+    cache = Cache.read(mode)

     if write_back not in (WriteBack.DIFF, WriteBack.COLOR_DIFF):
     if write_back not in (WriteBack.DIFF, WriteBack.COLOR_DIFF):

-        cache = read_cache(mode)
-        sources, cached = filter_cached(cache, sources)
+        sources, cached = cache.filtered_cached(sources)

         for src in sorted(cached):
             report.done(src, Changed.CACHED)
     if not sources:
         for src in sorted(cached):
             report.done(src, Changed.CACHED)
     if not sources:


@@ -184,4 +184,4 @@ async def schedule_formatting(
     if cancelled:
         await asyncio.gather(*cancelled, return_exceptions=True)
     if sources_to_cache:
     if cancelled:
         await asyncio.gather(*cancelled, return_exceptions=True)
     if sources_to_cache:

-        write_cache(cache, sources_to_cache, mode)
+        cache.write(sources_to_cache)