strawberry-graphql 0.274.1__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/schema/schema.py

@@ -412,6 +412,7 @@ class Schema(BaseSchema):
         context_value: Optional[Any] = None,
         root_value: Optional[Any] = None,
         operation_name: Optional[str] = None,
+        operation_extensions: Optional[dict[str, Any]] = None,
     ) -> ExecutionContext:
         return ExecutionContext(
             query=query,
@@ -421,6 +422,7 @@ class Schema(BaseSchema):
             root_value=root_value,
             variables=variable_values,
             provided_operation_name=operation_name,
+            operation_extensions=operation_extensions,
         )
 
     @lru_cache
@@ -554,6 +556,7 @@ class Schema(BaseSchema):
             context_value=context_value,
             root_value=root_value,
             operation_name=operation_name,
+            operation_extensions=operation_extensions,
         )
         extensions = self.get_extensions()
         # TODO (#3571): remove this when we implement execution context as parameter.
@@ -646,6 +649,7 @@ class Schema(BaseSchema):
             context_value=context_value,
             root_value=root_value,
             operation_name=operation_name,
+            operation_extensions=operation_extensions,
         )
         extensions = self._sync_extensions
         # TODO (#3571): remove this when we implement execution context as parameter.

strawberry/types/execution.py

@@ -52,6 +52,8 @@ class ExecutionContext:
     result: Optional[GraphQLExecutionResult] = None
     extensions_results: dict[str, Any] = dataclasses.field(default_factory=dict)
 
+    operation_extensions: Optional[dict[str, Any]] = None
+
     def __post_init__(self, provided_operation_name: str | None) -> None:
         self._provided_operation_name = provided_operation_name
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: strawberry-graphql
-Version: 0.274.1
+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.1.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
@@ -170,7 +170,7 @@ strawberry/schema/compat.py,sha256=xNpOEDfi-MODpplMGaKuKeQIVcr-tcAaKaU3TlBc1Zs,1
 strawberry/schema/config.py,sha256=6d2MPrAgq97-7aze555dRcB3yw-aeUexYMP3KVN22c0,1024
 strawberry/schema/exceptions.py,sha256=8gsMxxFDynMvRkUDuVL9Wwxk_zsmo6QoJ2l4NPxd64M,1137
 strawberry/schema/name_converter.py,sha256=xFOXEgqldFkxXRkIQvsJN1dPkWbEUaIrTYNOMYSEVwQ,6945
-strawberry/schema/schema.py,sha256=dnmSxoNdlrT02mca32Crtl0F55eweIqE-VOC77kuGjE,37723
+strawberry/schema/schema.py,sha256=-dVGJfOAyq-w2lpEbzCaiRNPMH68nirv-Q6gZ4aW0UU,37951
 strawberry/schema/schema_converter.py,sha256=vxFghA8c4vPLx0XM07gjIbWqMWUnaognzmsuCWTSkTk,39112
 strawberry/schema/types/__init__.py,sha256=oHO3COWhL3L1KLYCJNY1XFf5xt2GGtHiMC-UaYbFfnA,68
 strawberry/schema/types/base_scalars.py,sha256=JRUq0WjEkR9dFewstZnqnZKp0uOEipo4UXNF5dzRf4M,1971
@@ -203,7 +203,7 @@ strawberry/types/auto.py,sha256=WZ2cQAI8nREUigBzpzFqIKGjJ_C2VqpAPNe8vPjTciM,3007
 strawberry/types/base.py,sha256=Bfa-5Wen8qR7m6tlSMRRGlGE-chRGMHjQMopfNdbbrk,15197
 strawberry/types/cast.py,sha256=fx86MkLW77GIximBAwUk5vZxSGwDqUA6XicXvz8EXwQ,916
 strawberry/types/enum.py,sha256=IcCz0FLswJtDC_bU8aG1cjreawcqHywAzzVRBZUSAqs,6229
-strawberry/types/execution.py,sha256=m27VlmvaRLQsMnWWymxMMPOlaf0lUge4ZXea_OsoH2k,3761
+strawberry/types/execution.py,sha256=kW5rgkWP98oysJWNc_fcxyO1PZ36NJpPTPZpZP2ZXYk,3820
 strawberry/types/field.py,sha256=vxb7JvkHfRmDCYsjhDmVnO2lMbtSOteQm3jQUeSFu6g,21605
 strawberry/types/fields/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 strawberry/types/fields/resolver.py,sha256=b6lxfw6AMOUFWm7vs7a9KzNkpR8b_S110DoIosrrWDQ,14679
@@ -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.1.dist-info/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
-strawberry_graphql-0.274.1.dist-info/METADATA,sha256=dlqg7FwnCWKp0jQHBq0qi2IiZJALsgpEIed6YMuhneg,7444
-strawberry_graphql-0.274.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-strawberry_graphql-0.274.1.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
-strawberry_graphql-0.274.1.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,,