Document jupyter hook (#2416)

[etc/vim.git] / src / blackd / __init__.py

diff --git a/src/blackd/__init__.py b/src/blackd/__init__.py
index fc684730e4b8a4dde8eb8cfabbe444999da235b0..5fdec1522261b9775a90a2ee465d3304590ee8e6 100644 (file)
--- a/src/blackd/__init__.py
+++ b/src/blackd/__init__.py
@@ -1,6 +1,5 @@
 import asyncio
 import logging
 import asyncio
 import logging

-import sys

 from concurrent.futures import Executor, ProcessPoolExecutor
 from datetime import datetime
 from functools import partial
 from concurrent.futures import Executor, ProcessPoolExecutor
 from datetime import datetime
 from functools import partial


@@ -11,15 +10,14 @@ try:
     from aiohttp import web
     import aiohttp_cors
 except ImportError as ie:
     from aiohttp import web
     import aiohttp_cors
 except ImportError as ie:

-    print(
+    raise ImportError(

         f"aiohttp dependency is not installed: {ie}. "
         + "Please re-install black with the '[d]' extra install "
         f"aiohttp dependency is not installed: {ie}. "
         + "Please re-install black with the '[d]' extra install "

-        + "to obtain aiohttp_cors: `pip install black[d]`",
-        file=sys.stderr,
-    )
-    sys.exit(-1)
+        + "to obtain aiohttp_cors: `pip install black[d]`"
+    ) from None

 
 import black
 
 import black

+from black.concurrency import maybe_install_uvloop

 import click
 
 from _black_version import version as __version__
 import click
 
 from _black_version import version as __version__


@@ -202,6 +200,7 @@ def parse_python_variant_header(value: str) -> Tuple[bool, Set[black.TargetVersi
 
 
 def patched_main() -> None:
 
 
 def patched_main() -> None:

+    maybe_install_uvloop()

     freeze_support()
     black.patch_click()
     main()
     freeze_support()
     black.patch_click()
     main()