strawberry-graphql 0.274.2__py3-none-any.whl → 0.274.3__py3-none-any.whl

strawberry/cli/commands/upgrade/_run_codemod.py

@@ -7,7 +7,6 @@ from multiprocessing import Pool, cpu_count
 from typing import TYPE_CHECKING, Any, Union
 
 from libcst.codemod._cli import ExecutionConfig, ExecutionResult, _execute_transform
-from libcst.codemod._dummy_pool import DummyPool
 from rich.progress import Progress
 
 from ._fake_progress import FakeProgress
@@ -18,7 +17,6 @@ if TYPE_CHECKING:
     from libcst.codemod import Codemod
 
 ProgressType = Union[type[Progress], type[FakeProgress]]
-PoolType = Union[type[Pool], type[DummyPool]]  # type: ignore
 
 
 def _get_libcst_version() -> tuple[int, int, int]:
@@ -37,29 +35,21 @@ def _execute_transform_wrap(
 ) -> ExecutionResult:
     additional_kwargs: dict[str, Any] = {}
 
-    if _get_libcst_version() >= (1, 4, 0):
+    libcst_version = _get_libcst_version()
+
+    if (1, 4, 0) <= libcst_version < (1, 8, 0):
         additional_kwargs["scratch"] = {}
 
+    if libcst_version >= (1, 8, 0):
+        additional_kwargs["original_scratch"] = {}
+        additional_kwargs["codemod_args"] = {}
+        additional_kwargs["repo_manager"] = None
+
     # TODO: maybe capture warnings?
     with open(os.devnull, "w") as null, contextlib.redirect_stderr(null):  # noqa: PTH123
         return _execute_transform(**job, **additional_kwargs)
 
 
-def _get_progress_and_pool(
-    total_files: int, jobs: int
-) -> tuple[PoolType, ProgressType]:
-    poll_impl: PoolType = Pool  # type: ignore
-    progress_impl: ProgressType = Progress
-
-    if total_files == 1 or jobs == 1:
-        poll_impl = DummyPool
-
-    if total_files == 1:
-        progress_impl = FakeProgress
-
-    return poll_impl, progress_impl
-
-
 def run_codemod(
     codemod: Codemod,
     files: Sequence[str],
@@ -70,8 +60,6 @@ def run_codemod(
 
     config = ExecutionConfig()
 
-    pool_impl, progress_impl = _get_progress_and_pool(total, jobs)
-
     tasks = [
         {
             "transformer": codemod,
@@ -81,7 +69,7 @@ def run_codemod(
         for filename in files
     ]
 
-    with pool_impl(processes=jobs) as p, progress_impl() as progress:  # type: ignore
+    with Pool(processes=jobs) as p, Progress() as progress:
         task_id = progress.add_task("[cyan]Updating...", total=len(tasks))
 
         for result in p.imap_unordered(

{strawberry_graphql-0.274.2.dist-info → strawberry_graphql-0.274.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: strawberry-graphql
-Version: 0.274.2
+Version: 0.274.3
 Summary: A library for creating GraphQL APIs
 License: MIT
 Keywords: graphql,api,rest,starlette,async
@@ -37,9 +37,9 @@ Requires-Dist: channels (>=3.0.5) ; extra == "channels"
 Requires-Dist: fastapi (>=0.65.2) ; extra == "fastapi"
 Requires-Dist: flask (>=1.1) ; extra == "flask"
 Requires-Dist: graphql-core (>=3.2.0,<3.4.0)
-Requires-Dist: libcst (>=0.4.7,<1.8.0) ; extra == "cli"
-Requires-Dist: libcst (>=0.4.7,<1.8.0) ; extra == "debug"
-Requires-Dist: libcst (>=0.4.7,<1.8.0) ; extra == "debug-server"
+Requires-Dist: libcst ; extra == "cli"
+Requires-Dist: libcst ; extra == "debug"
+Requires-Dist: libcst ; extra == "debug-server"
 Requires-Dist: litestar (>=2) ; (python_version ~= "3.10") and (extra == "litestar")
 Requires-Dist: opentelemetry-api (<2) ; extra == "opentelemetry"
 Requires-Dist: opentelemetry-sdk (<2) ; extra == "opentelemetry"

{strawberry_graphql-0.274.2.dist-info → strawberry_graphql-0.274.3.dist-info}/RECORD

@@ -26,7 +26,7 @@ strawberry/cli/commands/schema_codegen.py,sha256=G6eV08a51sjVxCm3jn75oPn9hC8YarK
 strawberry/cli/commands/server.py,sha256=qj5wn22HvyJDzwnWzduIWRnS912XvD7GRhPGJkbLaz4,2217
 strawberry/cli/commands/upgrade/__init__.py,sha256=nY_Cj4yOj3CVdzEPWMAgof-dIr804sEJ-cCVOfI6UWo,2480
 strawberry/cli/commands/upgrade/_fake_progress.py,sha256=fefLgJwTXe4kG9RntdEJdzkPPRBK_pZqnmMH-pxD85Y,484
-strawberry/cli/commands/upgrade/_run_codemod.py,sha256=5xGGaqQZZagiKGyO3mrM1jKaVeTBojW8ktAlpEI4Dfw,2497
+strawberry/cli/commands/upgrade/_run_codemod.py,sha256=LZd5D1PP65bwVZjBvPPVrZ9t-bfvrafZ__HPBrW2WYA,2168
 strawberry/cli/constants.py,sha256=GhhDZOl9lN4glq50OI1oSbPSGqQXEarZ6r_Grq8pcpI,138
 strawberry/cli/debug_server.py,sha256=mKxJZf_-SbWWusoFMzT8-E5qTshIx3IuciG6xlC21kI,999
 strawberry/cli/utils/__init__.py,sha256=5h6QMXbY4zbWVGg8xpsKlgWSEsNgn1fcjbRrJjgzdEc,987
@@ -232,8 +232,8 @@ strawberry/utils/logging.py,sha256=U1cseHGquN09YFhFmRkiphfASKCyK0HUZREImPgVb0c,7
 strawberry/utils/operation.py,sha256=ZgVOw3K2jQuLjNOYUHauF7itJD0QDNoPw9PBi0IYf6k,1234
 strawberry/utils/str_converters.py,sha256=-eH1Cl16IO_wrBlsGM-km4IY0IKsjhjnSNGRGOwQjVM,897
 strawberry/utils/typing.py,sha256=SDvX-Du-9HAV3-XXjqi7Q5f5qPDDFd_gASIITiwBQT4,14073
-strawberry_graphql-0.274.2.dist-info/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
-strawberry_graphql-0.274.2.dist-info/METADATA,sha256=wiQIHSo_eBtZMKje2nrJOi6pDCfyovgEXL8bECIeefI,7444
-strawberry_graphql-0.274.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-strawberry_graphql-0.274.2.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
-strawberry_graphql-0.274.2.dist-info/RECORD,,
+strawberry_graphql-0.274.3.dist-info/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
+strawberry_graphql-0.274.3.dist-info/METADATA,sha256=MKO1coLJGJVdouLUvLOLladnxYZbB4SITr4gBdsoTM0,7393
+strawberry_graphql-0.274.3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+strawberry_graphql-0.274.3.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
+strawberry_graphql-0.274.3.dist-info/RECORD,,