strawberry-graphql 0.221.1__py3-none-any.whl → 0.223.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- strawberry/channels/handlers/http_handler.py +1 -0
- strawberry/federation/enum.py +37 -4
- strawberry/federation/field.py +53 -24
- strawberry/federation/object_type.py +81 -27
- strawberry/federation/scalar.py +26 -1
- strawberry/federation/schema_directives.py +70 -12
- strawberry/printer/printer.py +3 -1
- strawberry/schema_codegen/__init__.py +310 -32
- strawberry/types/nodes.py +1 -0
- {strawberry_graphql-0.221.1.dist-info → strawberry_graphql-0.223.0.dist-info}/METADATA +1 -1
- {strawberry_graphql-0.221.1.dist-info → strawberry_graphql-0.223.0.dist-info}/RECORD +14 -14
- {strawberry_graphql-0.221.1.dist-info → strawberry_graphql-0.223.0.dist-info}/LICENSE +0 -0
- {strawberry_graphql-0.221.1.dist-info → strawberry_graphql-0.223.0.dist-info}/WHEEL +0 -0
- {strawberry_graphql-0.221.1.dist-info → strawberry_graphql-0.223.0.dist-info}/entry_points.txt +0 -0
strawberry/federation/enum.py
CHANGED
@@ -1,6 +1,15 @@
|
|
1
1
|
from __future__ import annotations
|
2
2
|
|
3
|
-
from typing import
|
3
|
+
from typing import (
|
4
|
+
TYPE_CHECKING,
|
5
|
+
Any,
|
6
|
+
Callable,
|
7
|
+
Iterable,
|
8
|
+
List,
|
9
|
+
Optional,
|
10
|
+
Union,
|
11
|
+
overload,
|
12
|
+
)
|
4
13
|
|
5
14
|
from strawberry.enum import _process_enum
|
6
15
|
from strawberry.enum import enum_value as base_enum_value
|
@@ -36,7 +45,10 @@ def enum(
|
|
36
45
|
name: Optional[str] = None,
|
37
46
|
description: Optional[str] = None,
|
38
47
|
directives: Iterable[object] = (),
|
48
|
+
authenticated: bool = False,
|
39
49
|
inaccessible: bool = False,
|
50
|
+
policy: Optional[List[List[str]]] = None,
|
51
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
40
52
|
tags: Optional[Iterable[str]] = (),
|
41
53
|
) -> EnumType:
|
42
54
|
...
|
@@ -49,7 +61,10 @@ def enum(
|
|
49
61
|
name: Optional[str] = None,
|
50
62
|
description: Optional[str] = None,
|
51
63
|
directives: Iterable[object] = (),
|
64
|
+
authenticated: bool = False,
|
52
65
|
inaccessible: bool = False,
|
66
|
+
policy: Optional[List[List[str]]] = None,
|
67
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
53
68
|
tags: Optional[Iterable[str]] = (),
|
54
69
|
) -> Callable[[EnumType], EnumType]:
|
55
70
|
...
|
@@ -61,8 +76,11 @@ def enum(
|
|
61
76
|
name=None,
|
62
77
|
description=None,
|
63
78
|
directives=(),
|
64
|
-
|
65
|
-
|
79
|
+
authenticated: bool = False,
|
80
|
+
inaccessible: bool = False,
|
81
|
+
policy: Optional[List[List[str]]] = None,
|
82
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
83
|
+
tags: Optional[Iterable[str]] = (),
|
66
84
|
) -> Union[EnumType, Callable[[EnumType], EnumType]]:
|
67
85
|
"""Registers the enum in the GraphQL type system.
|
68
86
|
|
@@ -70,13 +88,28 @@ def enum(
|
|
70
88
|
the value passed of name instead of the Enum class name.
|
71
89
|
"""
|
72
90
|
|
73
|
-
from strawberry.federation.schema_directives import
|
91
|
+
from strawberry.federation.schema_directives import (
|
92
|
+
Authenticated,
|
93
|
+
Inaccessible,
|
94
|
+
Policy,
|
95
|
+
RequiresScopes,
|
96
|
+
Tag,
|
97
|
+
)
|
74
98
|
|
75
99
|
directives = list(directives)
|
76
100
|
|
101
|
+
if authenticated:
|
102
|
+
directives.append(Authenticated())
|
103
|
+
|
77
104
|
if inaccessible:
|
78
105
|
directives.append(Inaccessible())
|
79
106
|
|
107
|
+
if policy:
|
108
|
+
directives.append(Policy(policies=policy))
|
109
|
+
|
110
|
+
if requires_scopes:
|
111
|
+
directives.append(RequiresScopes(scopes=requires_scopes))
|
112
|
+
|
80
113
|
if tags:
|
81
114
|
directives.extend(Tag(name=tag) for tag in tags)
|
82
115
|
|
strawberry/federation/field.py
CHANGED
@@ -25,6 +25,7 @@ if TYPE_CHECKING:
|
|
25
25
|
from strawberry.field import _RESOLVER_TYPE, StrawberryField
|
26
26
|
from strawberry.permission import BasePermission
|
27
27
|
|
28
|
+
from .schema_directives import Override
|
28
29
|
|
29
30
|
T = TypeVar("T")
|
30
31
|
|
@@ -36,13 +37,16 @@ def field(
|
|
36
37
|
name: Optional[str] = None,
|
37
38
|
is_subscription: bool = False,
|
38
39
|
description: Optional[str] = None,
|
40
|
+
authenticated: bool = False,
|
41
|
+
external: bool = False,
|
42
|
+
inaccessible: bool = False,
|
43
|
+
policy: Optional[List[List[str]]] = None,
|
39
44
|
provides: Optional[List[str]] = None,
|
45
|
+
override: Optional[Union[Override, str]] = None,
|
40
46
|
requires: Optional[List[str]] = None,
|
41
|
-
|
42
|
-
shareable: bool = False,
|
47
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
43
48
|
tags: Optional[Iterable[str]] = (),
|
44
|
-
|
45
|
-
inaccessible: bool = False,
|
49
|
+
shareable: bool = False,
|
46
50
|
init: Literal[False] = False,
|
47
51
|
permission_classes: Optional[List[Type[BasePermission]]] = None,
|
48
52
|
deprecation_reason: Optional[str] = None,
|
@@ -61,13 +65,16 @@ def field(
|
|
61
65
|
name: Optional[str] = None,
|
62
66
|
is_subscription: bool = False,
|
63
67
|
description: Optional[str] = None,
|
68
|
+
authenticated: bool = False,
|
69
|
+
external: bool = False,
|
70
|
+
inaccessible: bool = False,
|
71
|
+
policy: Optional[List[List[str]]] = None,
|
64
72
|
provides: Optional[List[str]] = None,
|
73
|
+
override: Optional[Union[Override, str]] = None,
|
65
74
|
requires: Optional[List[str]] = None,
|
66
|
-
|
67
|
-
shareable: bool = False,
|
75
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
68
76
|
tags: Optional[Iterable[str]] = (),
|
69
|
-
|
70
|
-
inaccessible: bool = False,
|
77
|
+
shareable: bool = False,
|
71
78
|
init: Literal[True] = True,
|
72
79
|
permission_classes: Optional[List[Type[BasePermission]]] = None,
|
73
80
|
deprecation_reason: Optional[str] = None,
|
@@ -87,13 +94,16 @@ def field(
|
|
87
94
|
name: Optional[str] = None,
|
88
95
|
is_subscription: bool = False,
|
89
96
|
description: Optional[str] = None,
|
97
|
+
authenticated: bool = False,
|
98
|
+
external: bool = False,
|
99
|
+
inaccessible: bool = False,
|
100
|
+
policy: Optional[List[List[str]]] = None,
|
90
101
|
provides: Optional[List[str]] = None,
|
102
|
+
override: Optional[Union[Override, str]] = None,
|
91
103
|
requires: Optional[List[str]] = None,
|
92
|
-
|
93
|
-
shareable: bool = False,
|
104
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
94
105
|
tags: Optional[Iterable[str]] = (),
|
95
|
-
|
96
|
-
inaccessible: bool = False,
|
106
|
+
shareable: bool = False,
|
97
107
|
permission_classes: Optional[List[Type[BasePermission]]] = None,
|
98
108
|
deprecation_reason: Optional[str] = None,
|
99
109
|
default: Any = UNSET,
|
@@ -111,13 +121,16 @@ def field(
|
|
111
121
|
name: Optional[str] = None,
|
112
122
|
is_subscription: bool = False,
|
113
123
|
description: Optional[str] = None,
|
124
|
+
authenticated: bool = False,
|
125
|
+
external: bool = False,
|
126
|
+
inaccessible: bool = False,
|
127
|
+
policy: Optional[List[List[str]]] = None,
|
114
128
|
provides: Optional[List[str]] = None,
|
129
|
+
override: Optional[Union[Override, str]] = None,
|
115
130
|
requires: Optional[List[str]] = None,
|
116
|
-
|
117
|
-
shareable: bool = False,
|
131
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
118
132
|
tags: Optional[Iterable[str]] = (),
|
119
|
-
|
120
|
-
inaccessible: bool = False,
|
133
|
+
shareable: bool = False,
|
121
134
|
permission_classes: Optional[List[Type[BasePermission]]] = None,
|
122
135
|
deprecation_reason: Optional[str] = None,
|
123
136
|
default: Any = dataclasses.MISSING,
|
@@ -131,25 +144,47 @@ def field(
|
|
131
144
|
init: Literal[True, False, None] = None,
|
132
145
|
) -> Any:
|
133
146
|
from .schema_directives import (
|
147
|
+
Authenticated,
|
134
148
|
External,
|
135
149
|
Inaccessible,
|
136
150
|
Override,
|
151
|
+
Policy,
|
137
152
|
Provides,
|
138
153
|
Requires,
|
154
|
+
RequiresScopes,
|
139
155
|
Shareable,
|
140
156
|
Tag,
|
141
157
|
)
|
142
158
|
|
143
159
|
directives = list(directives)
|
144
160
|
|
161
|
+
if authenticated:
|
162
|
+
directives.append(Authenticated())
|
163
|
+
|
164
|
+
if external:
|
165
|
+
directives.append(External())
|
166
|
+
|
167
|
+
if inaccessible:
|
168
|
+
directives.append(Inaccessible())
|
169
|
+
|
170
|
+
if override:
|
171
|
+
directives.append(
|
172
|
+
Override(override_from=override, label=UNSET)
|
173
|
+
if isinstance(override, str)
|
174
|
+
else override
|
175
|
+
)
|
176
|
+
|
177
|
+
if policy:
|
178
|
+
directives.append(Policy(policies=policy))
|
179
|
+
|
145
180
|
if provides:
|
146
181
|
directives.append(Provides(fields=" ".join(provides)))
|
147
182
|
|
148
183
|
if requires:
|
149
184
|
directives.append(Requires(fields=" ".join(requires)))
|
150
185
|
|
151
|
-
if
|
152
|
-
directives.append(
|
186
|
+
if requires_scopes:
|
187
|
+
directives.append(RequiresScopes(scopes=requires_scopes))
|
153
188
|
|
154
189
|
if shareable:
|
155
190
|
directives.append(Shareable())
|
@@ -157,12 +192,6 @@ def field(
|
|
157
192
|
if tags:
|
158
193
|
directives.extend(Tag(name=tag) for tag in tags)
|
159
194
|
|
160
|
-
if override:
|
161
|
-
directives.append(Override(override_from=override))
|
162
|
-
|
163
|
-
if inaccessible:
|
164
|
-
directives.append(Inaccessible())
|
165
|
-
|
166
195
|
return base_field( # type: ignore
|
167
196
|
resolver=resolver, # type: ignore
|
168
197
|
name=name,
|
@@ -2,6 +2,7 @@ from typing import (
|
|
2
2
|
TYPE_CHECKING,
|
3
3
|
Callable,
|
4
4
|
Iterable,
|
5
|
+
List,
|
5
6
|
Optional,
|
6
7
|
Sequence,
|
7
8
|
Type,
|
@@ -31,19 +32,25 @@ def _impl_type(
|
|
31
32
|
name: Optional[str] = None,
|
32
33
|
description: Optional[str] = None,
|
33
34
|
directives: Iterable[object] = (),
|
35
|
+
authenticated: bool = False,
|
34
36
|
keys: Iterable[Union["Key", str]] = (),
|
35
37
|
extend: bool = False,
|
36
38
|
shareable: bool = False,
|
37
39
|
inaccessible: bool = UNSET,
|
40
|
+
policy: Optional[List[List[str]]] = None,
|
41
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
38
42
|
tags: Iterable[str] = (),
|
39
43
|
is_input: bool = False,
|
40
44
|
is_interface: bool = False,
|
41
45
|
is_interface_object: bool = False,
|
42
46
|
) -> T:
|
43
47
|
from strawberry.federation.schema_directives import (
|
48
|
+
Authenticated,
|
44
49
|
Inaccessible,
|
45
50
|
InterfaceObject,
|
46
51
|
Key,
|
52
|
+
Policy,
|
53
|
+
RequiresScopes,
|
47
54
|
Shareable,
|
48
55
|
Tag,
|
49
56
|
)
|
@@ -55,12 +62,21 @@ def _impl_type(
|
|
55
62
|
for key in keys
|
56
63
|
)
|
57
64
|
|
58
|
-
if
|
59
|
-
directives.append(
|
65
|
+
if authenticated:
|
66
|
+
directives.append(Authenticated())
|
60
67
|
|
61
68
|
if inaccessible is not UNSET:
|
62
69
|
directives.append(Inaccessible())
|
63
70
|
|
71
|
+
if policy:
|
72
|
+
directives.append(Policy(policies=policy))
|
73
|
+
|
74
|
+
if requires_scopes:
|
75
|
+
directives.append(RequiresScopes(scopes=requires_scopes))
|
76
|
+
|
77
|
+
if shareable:
|
78
|
+
directives.append(Shareable())
|
79
|
+
|
64
80
|
if tags:
|
65
81
|
directives.extend(Tag(name=tag) for tag in tags)
|
66
82
|
|
@@ -89,10 +105,15 @@ def type(
|
|
89
105
|
*,
|
90
106
|
name: Optional[str] = None,
|
91
107
|
description: Optional[str] = None,
|
92
|
-
|
108
|
+
directives: Iterable[object] = (),
|
109
|
+
authenticated: bool = False,
|
110
|
+
extend: bool = False,
|
93
111
|
inaccessible: bool = UNSET,
|
112
|
+
keys: Iterable[Union["Key", str]] = (),
|
113
|
+
policy: Optional[List[List[str]]] = None,
|
114
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
115
|
+
shareable: bool = False,
|
94
116
|
tags: Iterable[str] = (),
|
95
|
-
extend: bool = False,
|
96
117
|
) -> T:
|
97
118
|
...
|
98
119
|
|
@@ -107,12 +128,15 @@ def type(
|
|
107
128
|
*,
|
108
129
|
name: Optional[str] = None,
|
109
130
|
description: Optional[str] = None,
|
110
|
-
|
111
|
-
|
112
|
-
tags: Iterable[str] = (),
|
131
|
+
directives: Iterable[object] = (),
|
132
|
+
authenticated: bool = False,
|
113
133
|
extend: bool = False,
|
134
|
+
inaccessible: bool = UNSET,
|
135
|
+
keys: Iterable[Union["Key", str]] = (),
|
136
|
+
policy: Optional[List[List[str]]] = None,
|
137
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
114
138
|
shareable: bool = False,
|
115
|
-
|
139
|
+
tags: Iterable[str] = (),
|
116
140
|
) -> Callable[[T], T]:
|
117
141
|
...
|
118
142
|
|
@@ -122,22 +146,28 @@ def type(
|
|
122
146
|
*,
|
123
147
|
name: Optional[str] = None,
|
124
148
|
description: Optional[str] = None,
|
125
|
-
|
126
|
-
|
127
|
-
tags: Iterable[str] = (),
|
149
|
+
directives: Iterable[object] = (),
|
150
|
+
authenticated: bool = False,
|
128
151
|
extend: bool = False,
|
152
|
+
inaccessible: bool = UNSET,
|
153
|
+
keys: Iterable[Union["Key", str]] = (),
|
154
|
+
policy: Optional[List[List[str]]] = None,
|
155
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
129
156
|
shareable: bool = False,
|
130
|
-
|
157
|
+
tags: Iterable[str] = (),
|
131
158
|
):
|
132
159
|
return _impl_type(
|
133
160
|
cls,
|
134
161
|
name=name,
|
135
162
|
description=description,
|
136
163
|
directives=directives,
|
164
|
+
authenticated=authenticated,
|
137
165
|
keys=keys,
|
138
166
|
extend=extend,
|
139
|
-
shareable=shareable,
|
140
167
|
inaccessible=inaccessible,
|
168
|
+
policy=policy,
|
169
|
+
requires_scopes=requires_scopes,
|
170
|
+
shareable=shareable,
|
141
171
|
tags=tags,
|
142
172
|
)
|
143
173
|
|
@@ -208,10 +238,13 @@ def interface(
|
|
208
238
|
*,
|
209
239
|
name: Optional[str] = None,
|
210
240
|
description: Optional[str] = None,
|
211
|
-
|
241
|
+
directives: Iterable[object] = (),
|
242
|
+
authenticated: bool = False,
|
212
243
|
inaccessible: bool = UNSET,
|
244
|
+
keys: Iterable[Union["Key", str]] = (),
|
245
|
+
policy: Optional[List[List[str]]] = None,
|
246
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
213
247
|
tags: Iterable[str] = (),
|
214
|
-
directives: Iterable[object] = (),
|
215
248
|
) -> T:
|
216
249
|
...
|
217
250
|
|
@@ -226,10 +259,13 @@ def interface(
|
|
226
259
|
*,
|
227
260
|
name: Optional[str] = None,
|
228
261
|
description: Optional[str] = None,
|
229
|
-
|
262
|
+
directives: Iterable[object] = (),
|
263
|
+
authenticated: bool = False,
|
230
264
|
inaccessible: bool = UNSET,
|
265
|
+
keys: Iterable[Union["Key", str]] = (),
|
266
|
+
policy: Optional[List[List[str]]] = None,
|
267
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
231
268
|
tags: Iterable[str] = (),
|
232
|
-
directives: Iterable[object] = (),
|
233
269
|
) -> Callable[[T], T]:
|
234
270
|
...
|
235
271
|
|
@@ -239,20 +275,26 @@ def interface(
|
|
239
275
|
*,
|
240
276
|
name: Optional[str] = None,
|
241
277
|
description: Optional[str] = None,
|
242
|
-
|
278
|
+
directives: Iterable[object] = (),
|
279
|
+
authenticated: bool = False,
|
243
280
|
inaccessible: bool = UNSET,
|
281
|
+
keys: Iterable[Union["Key", str]] = (),
|
282
|
+
policy: Optional[List[List[str]]] = None,
|
283
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
244
284
|
tags: Iterable[str] = (),
|
245
|
-
directives: Iterable[object] = (),
|
246
285
|
):
|
247
286
|
return _impl_type(
|
248
287
|
cls,
|
249
288
|
name=name,
|
250
289
|
description=description,
|
251
290
|
directives=directives,
|
291
|
+
authenticated=authenticated,
|
252
292
|
keys=keys,
|
253
293
|
inaccessible=inaccessible,
|
254
|
-
|
294
|
+
policy=policy,
|
295
|
+
requires_scopes=requires_scopes,
|
255
296
|
tags=tags,
|
297
|
+
is_interface=True,
|
256
298
|
)
|
257
299
|
|
258
300
|
|
@@ -265,12 +307,15 @@ def interface(
|
|
265
307
|
def interface_object(
|
266
308
|
cls: T,
|
267
309
|
*,
|
268
|
-
keys: Iterable[Union["Key", str]],
|
269
310
|
name: Optional[str] = None,
|
270
311
|
description: Optional[str] = None,
|
312
|
+
directives: Iterable[object] = (),
|
313
|
+
authenticated: bool = False,
|
271
314
|
inaccessible: bool = UNSET,
|
315
|
+
keys: Iterable[Union["Key", str]] = (),
|
316
|
+
policy: Optional[List[List[str]]] = None,
|
317
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
272
318
|
tags: Iterable[str] = (),
|
273
|
-
directives: Iterable[object] = (),
|
274
319
|
) -> T:
|
275
320
|
...
|
276
321
|
|
@@ -283,12 +328,15 @@ def interface_object(
|
|
283
328
|
)
|
284
329
|
def interface_object(
|
285
330
|
*,
|
286
|
-
keys: Iterable[Union["Key", str]],
|
287
331
|
name: Optional[str] = None,
|
288
332
|
description: Optional[str] = None,
|
333
|
+
directives: Iterable[object] = (),
|
334
|
+
authenticated: bool = False,
|
289
335
|
inaccessible: bool = UNSET,
|
336
|
+
keys: Iterable[Union["Key", str]] = (),
|
337
|
+
policy: Optional[List[List[str]]] = None,
|
338
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
290
339
|
tags: Iterable[str] = (),
|
291
|
-
directives: Iterable[object] = (),
|
292
340
|
) -> Callable[[T], T]:
|
293
341
|
...
|
294
342
|
|
@@ -296,21 +344,27 @@ def interface_object(
|
|
296
344
|
def interface_object(
|
297
345
|
cls: Optional[T] = None,
|
298
346
|
*,
|
299
|
-
keys: Iterable[Union["Key", str]],
|
300
347
|
name: Optional[str] = None,
|
301
348
|
description: Optional[str] = None,
|
349
|
+
directives: Iterable[object] = (),
|
350
|
+
authenticated: bool = False,
|
302
351
|
inaccessible: bool = UNSET,
|
352
|
+
keys: Iterable[Union["Key", str]] = (),
|
353
|
+
policy: Optional[List[List[str]]] = None,
|
354
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
303
355
|
tags: Iterable[str] = (),
|
304
|
-
directives: Iterable[object] = (),
|
305
356
|
):
|
306
357
|
return _impl_type(
|
307
358
|
cls,
|
308
359
|
name=name,
|
309
360
|
description=description,
|
310
361
|
directives=directives,
|
362
|
+
authenticated=authenticated,
|
311
363
|
keys=keys,
|
312
364
|
inaccessible=inaccessible,
|
365
|
+
policy=policy,
|
366
|
+
requires_scopes=requires_scopes,
|
367
|
+
tags=tags,
|
313
368
|
is_interface=False,
|
314
369
|
is_interface_object=True,
|
315
|
-
tags=tags,
|
316
370
|
)
|
strawberry/federation/scalar.py
CHANGED
@@ -3,6 +3,7 @@ from typing import (
|
|
3
3
|
Any,
|
4
4
|
Callable,
|
5
5
|
Iterable,
|
6
|
+
List,
|
6
7
|
NewType,
|
7
8
|
Optional,
|
8
9
|
Type,
|
@@ -34,7 +35,10 @@ def scalar(
|
|
34
35
|
parse_value: Optional[Callable] = None,
|
35
36
|
parse_literal: Optional[Callable] = None,
|
36
37
|
directives: Iterable[object] = (),
|
38
|
+
authenticated: bool = False,
|
37
39
|
inaccessible: bool = False,
|
40
|
+
policy: Optional[List[List[str]]] = None,
|
41
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
38
42
|
tags: Optional[Iterable[str]] = (),
|
39
43
|
) -> Callable[[_T], _T]:
|
40
44
|
...
|
@@ -51,7 +55,10 @@ def scalar(
|
|
51
55
|
parse_value: Optional[Callable] = None,
|
52
56
|
parse_literal: Optional[Callable] = None,
|
53
57
|
directives: Iterable[object] = (),
|
58
|
+
authenticated: bool = False,
|
54
59
|
inaccessible: bool = False,
|
60
|
+
policy: Optional[List[List[str]]] = None,
|
61
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
55
62
|
tags: Optional[Iterable[str]] = (),
|
56
63
|
) -> _T:
|
57
64
|
...
|
@@ -67,7 +74,10 @@ def scalar(
|
|
67
74
|
parse_value: Optional[Callable] = None,
|
68
75
|
parse_literal: Optional[Callable] = None,
|
69
76
|
directives: Iterable[object] = (),
|
77
|
+
authenticated: bool = False,
|
70
78
|
inaccessible: bool = False,
|
79
|
+
policy: Optional[List[List[str]]] = None,
|
80
|
+
requires_scopes: Optional[List[List[str]]] = None,
|
71
81
|
tags: Optional[Iterable[str]] = (),
|
72
82
|
) -> Any:
|
73
83
|
"""Annotates a class or type as a GraphQL custom scalar.
|
@@ -95,16 +105,31 @@ def scalar(
|
|
95
105
|
>>> self.items = items
|
96
106
|
|
97
107
|
"""
|
98
|
-
from strawberry.federation.schema_directives import
|
108
|
+
from strawberry.federation.schema_directives import (
|
109
|
+
Authenticated,
|
110
|
+
Inaccessible,
|
111
|
+
Policy,
|
112
|
+
RequiresScopes,
|
113
|
+
Tag,
|
114
|
+
)
|
99
115
|
|
100
116
|
if parse_value is None:
|
101
117
|
parse_value = cls
|
102
118
|
|
103
119
|
directives = list(directives)
|
104
120
|
|
121
|
+
if authenticated:
|
122
|
+
directives.append(Authenticated())
|
123
|
+
|
105
124
|
if inaccessible:
|
106
125
|
directives.append(Inaccessible())
|
107
126
|
|
127
|
+
if policy:
|
128
|
+
directives.append(Policy(policies=policy))
|
129
|
+
|
130
|
+
if requires_scopes:
|
131
|
+
directives.append(RequiresScopes(scopes=requires_scopes))
|
132
|
+
|
108
133
|
if tags:
|
109
134
|
directives.extend(Tag(name=tag) for tag in tags)
|
110
135
|
|
@@ -5,13 +5,17 @@ from strawberry import directive_field
|
|
5
5
|
from strawberry.schema_directive import Location, schema_directive
|
6
6
|
from strawberry.unset import UNSET
|
7
7
|
|
8
|
-
from .types import
|
8
|
+
from .types import (
|
9
|
+
FieldSet,
|
10
|
+
LinkImport,
|
11
|
+
LinkPurpose,
|
12
|
+
)
|
9
13
|
|
10
14
|
|
11
15
|
@dataclass
|
12
16
|
class ImportedFrom:
|
13
17
|
name: str
|
14
|
-
url: str = "https://specs.apollo.dev/federation/v2.
|
18
|
+
url: str = "https://specs.apollo.dev/federation/v2.7"
|
15
19
|
|
16
20
|
|
17
21
|
class FederationDirective:
|
@@ -23,7 +27,7 @@ class FederationDirective:
|
|
23
27
|
)
|
24
28
|
class External(FederationDirective):
|
25
29
|
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
26
|
-
name="external", url="https://specs.apollo.dev/federation/v2.
|
30
|
+
name="external", url="https://specs.apollo.dev/federation/v2.7"
|
27
31
|
)
|
28
32
|
|
29
33
|
|
@@ -33,7 +37,7 @@ class External(FederationDirective):
|
|
33
37
|
class Requires(FederationDirective):
|
34
38
|
fields: FieldSet
|
35
39
|
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
36
|
-
name="requires", url="https://specs.apollo.dev/federation/v2.
|
40
|
+
name="requires", url="https://specs.apollo.dev/federation/v2.7"
|
37
41
|
)
|
38
42
|
|
39
43
|
|
@@ -43,7 +47,7 @@ class Requires(FederationDirective):
|
|
43
47
|
class Provides(FederationDirective):
|
44
48
|
fields: FieldSet
|
45
49
|
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
46
|
-
name="provides", url="https://specs.apollo.dev/federation/v2.
|
50
|
+
name="provides", url="https://specs.apollo.dev/federation/v2.7"
|
47
51
|
)
|
48
52
|
|
49
53
|
|
@@ -57,7 +61,7 @@ class Key(FederationDirective):
|
|
57
61
|
fields: FieldSet
|
58
62
|
resolvable: Optional[bool] = True
|
59
63
|
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
60
|
-
name="key", url="https://specs.apollo.dev/federation/v2.
|
64
|
+
name="key", url="https://specs.apollo.dev/federation/v2.7"
|
61
65
|
)
|
62
66
|
|
63
67
|
|
@@ -69,7 +73,7 @@ class Key(FederationDirective):
|
|
69
73
|
)
|
70
74
|
class Shareable(FederationDirective):
|
71
75
|
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
72
|
-
name="shareable", url="https://specs.apollo.dev/federation/v2.
|
76
|
+
name="shareable", url="https://specs.apollo.dev/federation/v2.7"
|
73
77
|
)
|
74
78
|
|
75
79
|
|
@@ -115,7 +119,7 @@ class Link:
|
|
115
119
|
class Tag(FederationDirective):
|
116
120
|
name: str
|
117
121
|
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
118
|
-
name="tag", url="https://specs.apollo.dev/federation/v2.
|
122
|
+
name="tag", url="https://specs.apollo.dev/federation/v2.7"
|
119
123
|
)
|
120
124
|
|
121
125
|
|
@@ -124,8 +128,9 @@ class Tag(FederationDirective):
|
|
124
128
|
)
|
125
129
|
class Override(FederationDirective):
|
126
130
|
override_from: str = directive_field(name="from")
|
131
|
+
label: Optional[str] = UNSET
|
127
132
|
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
128
|
-
name="override", url="https://specs.apollo.dev/federation/v2.
|
133
|
+
name="override", url="https://specs.apollo.dev/federation/v2.7"
|
129
134
|
)
|
130
135
|
|
131
136
|
|
@@ -147,7 +152,7 @@ class Override(FederationDirective):
|
|
147
152
|
)
|
148
153
|
class Inaccessible(FederationDirective):
|
149
154
|
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
150
|
-
name="inaccessible", url="https://specs.apollo.dev/federation/v2.
|
155
|
+
name="inaccessible", url="https://specs.apollo.dev/federation/v2.7"
|
151
156
|
)
|
152
157
|
|
153
158
|
|
@@ -157,7 +162,7 @@ class Inaccessible(FederationDirective):
|
|
157
162
|
class ComposeDirective(FederationDirective):
|
158
163
|
name: str
|
159
164
|
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
160
|
-
name="composeDirective", url="https://specs.apollo.dev/federation/v2.
|
165
|
+
name="composeDirective", url="https://specs.apollo.dev/federation/v2.7"
|
161
166
|
)
|
162
167
|
|
163
168
|
|
@@ -166,5 +171,58 @@ class ComposeDirective(FederationDirective):
|
|
166
171
|
)
|
167
172
|
class InterfaceObject(FederationDirective):
|
168
173
|
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
169
|
-
name="interfaceObject", url="https://specs.apollo.dev/federation/v2.
|
174
|
+
name="interfaceObject", url="https://specs.apollo.dev/federation/v2.7"
|
175
|
+
)
|
176
|
+
|
177
|
+
|
178
|
+
@schema_directive(
|
179
|
+
locations=[
|
180
|
+
Location.FIELD_DEFINITION,
|
181
|
+
Location.OBJECT,
|
182
|
+
Location.INTERFACE,
|
183
|
+
Location.SCALAR,
|
184
|
+
Location.ENUM,
|
185
|
+
],
|
186
|
+
name="authenticated",
|
187
|
+
print_definition=False,
|
188
|
+
)
|
189
|
+
class Authenticated(FederationDirective):
|
190
|
+
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
191
|
+
name="authenticated", url="https://specs.apollo.dev/federation/v2.7"
|
192
|
+
)
|
193
|
+
|
194
|
+
|
195
|
+
@schema_directive(
|
196
|
+
locations=[
|
197
|
+
Location.FIELD_DEFINITION,
|
198
|
+
Location.OBJECT,
|
199
|
+
Location.INTERFACE,
|
200
|
+
Location.SCALAR,
|
201
|
+
Location.ENUM,
|
202
|
+
],
|
203
|
+
name="requiresScopes",
|
204
|
+
print_definition=False,
|
205
|
+
)
|
206
|
+
class RequiresScopes(FederationDirective):
|
207
|
+
scopes: "List[List[str]]"
|
208
|
+
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
209
|
+
name="requiresScopes", url="https://specs.apollo.dev/federation/v2.7"
|
210
|
+
)
|
211
|
+
|
212
|
+
|
213
|
+
@schema_directive(
|
214
|
+
locations=[
|
215
|
+
Location.FIELD_DEFINITION,
|
216
|
+
Location.OBJECT,
|
217
|
+
Location.INTERFACE,
|
218
|
+
Location.SCALAR,
|
219
|
+
Location.ENUM,
|
220
|
+
],
|
221
|
+
name="policy",
|
222
|
+
print_definition=False,
|
223
|
+
)
|
224
|
+
class Policy(FederationDirective):
|
225
|
+
policies: "List[List[str]]"
|
226
|
+
imported_from: ClassVar[ImportedFrom] = ImportedFrom(
|
227
|
+
name="policy", url="https://specs.apollo.dev/federation/v2.7"
|
170
228
|
)
|
strawberry/printer/printer.py
CHANGED
@@ -74,7 +74,9 @@ def _serialize_dataclasses(value: Dict[_T, object]) -> Dict[_T, object]:
|
|
74
74
|
|
75
75
|
|
76
76
|
@overload
|
77
|
-
def _serialize_dataclasses(
|
77
|
+
def _serialize_dataclasses(
|
78
|
+
value: Union[List[object], Tuple[object]],
|
79
|
+
) -> List[object]:
|
78
80
|
...
|
79
81
|
|
80
82
|
|
@@ -2,6 +2,9 @@ from __future__ import annotations
|
|
2
2
|
|
3
3
|
import dataclasses
|
4
4
|
import keyword
|
5
|
+
from collections import defaultdict
|
6
|
+
from typing import TYPE_CHECKING, List, Tuple, Union
|
7
|
+
from typing_extensions import Protocol, TypeAlias
|
5
8
|
|
6
9
|
import libcst as cst
|
7
10
|
from graphql import (
|
@@ -19,14 +22,28 @@ from graphql import (
|
|
19
22
|
OperationType,
|
20
23
|
ScalarTypeDefinitionNode,
|
21
24
|
SchemaDefinitionNode,
|
25
|
+
SchemaExtensionNode,
|
22
26
|
StringValueNode,
|
23
27
|
TypeNode,
|
24
28
|
UnionTypeDefinitionNode,
|
25
29
|
parse,
|
26
30
|
)
|
31
|
+
from graphql.language.ast import (
|
32
|
+
BooleanValueNode,
|
33
|
+
ConstValueNode,
|
34
|
+
ListValueNode,
|
35
|
+
)
|
27
36
|
|
28
37
|
from strawberry.utils.str_converters import to_snake_case
|
29
38
|
|
39
|
+
if TYPE_CHECKING:
|
40
|
+
from graphql.language.ast import ConstDirectiveNode
|
41
|
+
|
42
|
+
|
43
|
+
class HasDirectives(Protocol):
|
44
|
+
directives: Tuple[ConstDirectiveNode, ...]
|
45
|
+
|
46
|
+
|
30
47
|
_SCALAR_MAP = {
|
31
48
|
"Int": cst.Name("int"),
|
32
49
|
"Float": cst.Name("float"),
|
@@ -48,6 +65,48 @@ _SCALAR_MAP = {
|
|
48
65
|
}
|
49
66
|
|
50
67
|
|
68
|
+
@dataclasses.dataclass(frozen=True)
|
69
|
+
class Import:
|
70
|
+
module: str | None
|
71
|
+
imports: tuple[str]
|
72
|
+
|
73
|
+
def module_path_to_cst(self, module_path: str) -> cst.Name | cst.Attribute:
|
74
|
+
parts = module_path.split(".")
|
75
|
+
|
76
|
+
module_name: cst.Name | cst.Attribute = cst.Name(parts[0])
|
77
|
+
|
78
|
+
for part in parts[1:]:
|
79
|
+
module_name = cst.Attribute(value=module_name, attr=cst.Name(part))
|
80
|
+
|
81
|
+
return module_name
|
82
|
+
|
83
|
+
def to_cst(self) -> cst.Import | cst.ImportFrom:
|
84
|
+
if self.module is None:
|
85
|
+
return cst.Import(
|
86
|
+
names=[cst.ImportAlias(name=cst.Name(name)) for name in self.imports]
|
87
|
+
)
|
88
|
+
|
89
|
+
return cst.ImportFrom(
|
90
|
+
module=self.module_path_to_cst(self.module),
|
91
|
+
names=[cst.ImportAlias(name=cst.Name(name)) for name in self.imports],
|
92
|
+
)
|
93
|
+
|
94
|
+
|
95
|
+
def _is_federation_link_directive(directive: ConstDirectiveNode) -> bool:
|
96
|
+
if directive.name.value != "link":
|
97
|
+
return False
|
98
|
+
|
99
|
+
return next(
|
100
|
+
(
|
101
|
+
argument.value.value
|
102
|
+
for argument in directive.arguments
|
103
|
+
if argument.name.value == "url"
|
104
|
+
if isinstance(argument.value, StringValueNode)
|
105
|
+
),
|
106
|
+
"",
|
107
|
+
).startswith("https://specs.apollo.dev/federation")
|
108
|
+
|
109
|
+
|
51
110
|
def _get_field_type(
|
52
111
|
field_type: TypeNode, was_non_nullable: bool = False
|
53
112
|
) -> cst.BaseExpression:
|
@@ -85,7 +144,19 @@ def _get_field_type(
|
|
85
144
|
)
|
86
145
|
|
87
146
|
|
88
|
-
def
|
147
|
+
def _sanitize_argument(value: ArgumentValue) -> cst.SimpleString | cst.Name | cst.List:
|
148
|
+
if isinstance(value, bool):
|
149
|
+
return cst.Name(value=str(value))
|
150
|
+
|
151
|
+
if isinstance(value, list):
|
152
|
+
return cst.List(
|
153
|
+
elements=[
|
154
|
+
cst.Element(value=_sanitize_argument(item))
|
155
|
+
for item in value
|
156
|
+
if item is not None
|
157
|
+
],
|
158
|
+
)
|
159
|
+
|
89
160
|
if "\n" in value:
|
90
161
|
argument_value = cst.SimpleString(f'"""\n{value}\n"""')
|
91
162
|
elif '"' in value:
|
@@ -93,6 +164,12 @@ def _get_argument(name: str, value: str) -> cst.Arg:
|
|
93
164
|
else:
|
94
165
|
argument_value = cst.SimpleString(f'"{value}"')
|
95
166
|
|
167
|
+
return argument_value
|
168
|
+
|
169
|
+
|
170
|
+
def _get_argument(name: str, value: ArgumentValue) -> cst.Arg:
|
171
|
+
argument_value = _sanitize_argument(value)
|
172
|
+
|
96
173
|
return cst.Arg(
|
97
174
|
value=argument_value,
|
98
175
|
keyword=cst.Name(name),
|
@@ -100,7 +177,27 @@ def _get_argument(name: str, value: str) -> cst.Arg:
|
|
100
177
|
)
|
101
178
|
|
102
179
|
|
103
|
-
|
180
|
+
# TODO: this might be removed now
|
181
|
+
def _get_argument_list(name: str, values: list[ArgumentValue]) -> cst.Arg:
|
182
|
+
value = cst.List(
|
183
|
+
elements=[cst.Element(value=_sanitize_argument(value)) for value in values],
|
184
|
+
)
|
185
|
+
|
186
|
+
return cst.Arg(
|
187
|
+
value=value,
|
188
|
+
keyword=cst.Name(name),
|
189
|
+
equal=cst.AssignEqual(cst.SimpleWhitespace(""), cst.SimpleWhitespace("")),
|
190
|
+
)
|
191
|
+
|
192
|
+
|
193
|
+
def _get_field_value(
|
194
|
+
field: FieldDefinitionNode | InputValueDefinitionNode,
|
195
|
+
alias: str | None,
|
196
|
+
is_apollo_federation: bool,
|
197
|
+
imports: set[Import],
|
198
|
+
) -> cst.Call | None:
|
199
|
+
description = field.description.value if field.description else None
|
200
|
+
|
104
201
|
args = list(
|
105
202
|
filter(
|
106
203
|
None,
|
@@ -111,6 +208,24 @@ def _get_field_value(description: str | None, alias: str | None) -> cst.Call | N
|
|
111
208
|
)
|
112
209
|
)
|
113
210
|
|
211
|
+
directives = _get_directives(field)
|
212
|
+
|
213
|
+
apollo_federation_args = _get_federation_arguments(directives, imports)
|
214
|
+
|
215
|
+
if is_apollo_federation and apollo_federation_args:
|
216
|
+
args.extend(apollo_federation_args)
|
217
|
+
|
218
|
+
return cst.Call(
|
219
|
+
func=cst.Attribute(
|
220
|
+
value=cst.Attribute(
|
221
|
+
value=cst.Name("strawberry"),
|
222
|
+
attr=cst.Name("federation"),
|
223
|
+
),
|
224
|
+
attr=cst.Name("field"),
|
225
|
+
),
|
226
|
+
args=args,
|
227
|
+
)
|
228
|
+
|
114
229
|
if args:
|
115
230
|
return cst.Call(
|
116
231
|
func=cst.Attribute(
|
@@ -125,6 +240,8 @@ def _get_field_value(description: str | None, alias: str | None) -> cst.Call | N
|
|
125
240
|
|
126
241
|
def _get_field(
|
127
242
|
field: FieldDefinitionNode | InputValueDefinitionNode,
|
243
|
+
is_apollo_federation: bool,
|
244
|
+
imports: set[Import],
|
128
245
|
) -> cst.SimpleStatementLine:
|
129
246
|
name = to_snake_case(field.name.value)
|
130
247
|
alias: str | None = None
|
@@ -141,19 +258,131 @@ def _get_field(
|
|
141
258
|
_get_field_type(field.type),
|
142
259
|
),
|
143
260
|
value=_get_field_value(
|
144
|
-
|
145
|
-
alias=alias
|
261
|
+
field,
|
262
|
+
alias=alias,
|
263
|
+
is_apollo_federation=is_apollo_federation,
|
264
|
+
imports=imports,
|
146
265
|
),
|
147
266
|
)
|
148
267
|
]
|
149
268
|
)
|
150
269
|
|
151
270
|
|
271
|
+
ArgumentValue: TypeAlias = Union[str, bool, List["ArgumentValue"]]
|
272
|
+
|
273
|
+
|
274
|
+
def _get_argument_value(argument_value: ConstValueNode) -> ArgumentValue:
|
275
|
+
if isinstance(argument_value, StringValueNode):
|
276
|
+
return argument_value.value
|
277
|
+
elif isinstance(argument_value, EnumValueDefinitionNode):
|
278
|
+
return argument_value.name.value
|
279
|
+
elif isinstance(argument_value, ListValueNode):
|
280
|
+
return [_get_argument_value(arg) for arg in argument_value.values]
|
281
|
+
elif isinstance(argument_value, BooleanValueNode):
|
282
|
+
return argument_value.value
|
283
|
+
else:
|
284
|
+
raise NotImplementedError(f"Unknown argument value {argument_value}")
|
285
|
+
|
286
|
+
|
287
|
+
def _get_directives(
|
288
|
+
definition: HasDirectives,
|
289
|
+
) -> dict[str, list[dict[str, ArgumentValue]]]:
|
290
|
+
directives: dict[str, list[dict[str, ArgumentValue]]] = defaultdict(list)
|
291
|
+
|
292
|
+
for directive in definition.directives:
|
293
|
+
directive_name = directive.name.value
|
294
|
+
|
295
|
+
directives[directive_name].append(
|
296
|
+
{
|
297
|
+
argument.name.value: _get_argument_value(argument.value)
|
298
|
+
for argument in directive.arguments
|
299
|
+
}
|
300
|
+
)
|
301
|
+
|
302
|
+
return directives
|
303
|
+
|
304
|
+
|
305
|
+
def _get_federation_arguments(
|
306
|
+
directives: dict[str, list[dict[str, ArgumentValue]]],
|
307
|
+
imports: set[Import],
|
308
|
+
) -> list[cst.Arg]:
|
309
|
+
def append_arg_from_directive(
|
310
|
+
directive: str,
|
311
|
+
argument_name: str,
|
312
|
+
keyword_name: str | None = None,
|
313
|
+
flatten: bool = True,
|
314
|
+
):
|
315
|
+
keyword_name = keyword_name or directive
|
316
|
+
|
317
|
+
if directive in directives:
|
318
|
+
values = [item[argument_name] for item in directives[directive]]
|
319
|
+
|
320
|
+
if flatten:
|
321
|
+
arguments.append(_get_argument(keyword_name, values))
|
322
|
+
else:
|
323
|
+
arguments.extend(_get_argument(keyword_name, value) for value in values)
|
324
|
+
|
325
|
+
arguments: list[cst.Arg] = []
|
326
|
+
|
327
|
+
append_arg_from_directive("key", "fields", "keys")
|
328
|
+
append_arg_from_directive("requires", "fields")
|
329
|
+
append_arg_from_directive("provides", "fields")
|
330
|
+
append_arg_from_directive(
|
331
|
+
"requiresScopes", "scopes", "requires_scopes", flatten=False
|
332
|
+
)
|
333
|
+
append_arg_from_directive("policy", "policies", "policy", flatten=False)
|
334
|
+
append_arg_from_directive("tag", "name", "tags")
|
335
|
+
|
336
|
+
boolean_keys = (
|
337
|
+
"shareable",
|
338
|
+
"inaccessible",
|
339
|
+
"external",
|
340
|
+
"authenticated",
|
341
|
+
)
|
342
|
+
|
343
|
+
arguments.extend(
|
344
|
+
_get_argument(key, True) for key in boolean_keys if directives.get(key, False)
|
345
|
+
)
|
346
|
+
|
347
|
+
if overrides := directives.get("override"):
|
348
|
+
override = overrides[0]
|
349
|
+
|
350
|
+
if "label" not in override:
|
351
|
+
arguments.append(_get_argument("override", override["from"]))
|
352
|
+
else:
|
353
|
+
imports.add(
|
354
|
+
Import(
|
355
|
+
module="strawberry.federation.schema_directives",
|
356
|
+
imports=("Override",),
|
357
|
+
)
|
358
|
+
)
|
359
|
+
|
360
|
+
arguments.append(
|
361
|
+
cst.Arg(
|
362
|
+
keyword=cst.Name("override"),
|
363
|
+
value=cst.Call(
|
364
|
+
func=cst.Name("Override"),
|
365
|
+
args=[
|
366
|
+
_get_argument("override_from", override["from"]),
|
367
|
+
_get_argument("label", override["label"]),
|
368
|
+
],
|
369
|
+
),
|
370
|
+
equal=cst.AssignEqual(
|
371
|
+
cst.SimpleWhitespace(""), cst.SimpleWhitespace("")
|
372
|
+
),
|
373
|
+
)
|
374
|
+
)
|
375
|
+
|
376
|
+
return arguments
|
377
|
+
|
378
|
+
|
152
379
|
def _get_strawberry_decorator(
|
153
380
|
definition: ObjectTypeDefinitionNode
|
154
381
|
| ObjectTypeExtensionNode
|
155
382
|
| InterfaceTypeDefinitionNode
|
156
383
|
| InputObjectTypeDefinitionNode,
|
384
|
+
is_apollo_federation: bool,
|
385
|
+
imports: set[Import],
|
157
386
|
) -> cst.Decorator:
|
158
387
|
type_ = {
|
159
388
|
ObjectTypeDefinitionNode: "type",
|
@@ -168,15 +397,36 @@ def _get_strawberry_decorator(
|
|
168
397
|
else None
|
169
398
|
)
|
170
399
|
|
400
|
+
directives = _get_directives(definition)
|
401
|
+
|
171
402
|
decorator: cst.BaseExpression = cst.Attribute(
|
172
403
|
value=cst.Name("strawberry"),
|
173
404
|
attr=cst.Name(type_),
|
174
405
|
)
|
175
406
|
|
407
|
+
arguments: list[cst.Arg] = []
|
408
|
+
|
176
409
|
if description is not None:
|
410
|
+
arguments.append(_get_argument("description", description.value))
|
411
|
+
|
412
|
+
federation_arguments = _get_federation_arguments(directives, imports)
|
413
|
+
|
414
|
+
# and has any directive that is a federation directive
|
415
|
+
if is_apollo_federation and federation_arguments:
|
416
|
+
decorator = cst.Attribute(
|
417
|
+
value=cst.Attribute(
|
418
|
+
value=cst.Name("strawberry"),
|
419
|
+
attr=cst.Name("federation"),
|
420
|
+
),
|
421
|
+
attr=cst.Name(type_),
|
422
|
+
)
|
423
|
+
|
424
|
+
arguments.extend(federation_arguments)
|
425
|
+
|
426
|
+
if arguments:
|
177
427
|
decorator = cst.Call(
|
178
428
|
func=decorator,
|
179
|
-
args=
|
429
|
+
args=arguments,
|
180
430
|
)
|
181
431
|
|
182
432
|
return cst.Decorator(
|
@@ -189,8 +439,10 @@ def _get_class_definition(
|
|
189
439
|
| ObjectTypeExtensionNode
|
190
440
|
| InterfaceTypeDefinitionNode
|
191
441
|
| InputObjectTypeDefinitionNode,
|
442
|
+
is_apollo_federation: bool,
|
443
|
+
imports: set[Import],
|
192
444
|
) -> cst.ClassDef:
|
193
|
-
decorator = _get_strawberry_decorator(definition)
|
445
|
+
decorator = _get_strawberry_decorator(definition, is_apollo_federation, imports)
|
194
446
|
|
195
447
|
bases = (
|
196
448
|
[cst.Arg(cst.Name(interface.name.value)) for interface in definition.interfaces]
|
@@ -204,7 +456,12 @@ def _get_class_definition(
|
|
204
456
|
return cst.ClassDef(
|
205
457
|
name=cst.Name(definition.name.value),
|
206
458
|
bases=bases,
|
207
|
-
body=cst.IndentedBlock(
|
459
|
+
body=cst.IndentedBlock(
|
460
|
+
body=[
|
461
|
+
_get_field(field, is_apollo_federation, imports)
|
462
|
+
for field in definition.fields
|
463
|
+
]
|
464
|
+
),
|
208
465
|
decorators=[decorator],
|
209
466
|
)
|
210
467
|
|
@@ -243,6 +500,7 @@ def _get_schema_definition(
|
|
243
500
|
root_query_name: str | None,
|
244
501
|
root_mutation_name: str | None,
|
245
502
|
root_subscription_name: str | None,
|
503
|
+
is_apollo_federation: bool,
|
246
504
|
) -> cst.SimpleStatementLine | None:
|
247
505
|
if not any([root_query_name, root_mutation_name, root_subscription_name]):
|
248
506
|
return None
|
@@ -265,39 +523,45 @@ def _get_schema_definition(
|
|
265
523
|
if root_subscription_name:
|
266
524
|
args.append(_get_arg("subscription", root_subscription_name))
|
267
525
|
|
526
|
+
schema_call = cst.Call(
|
527
|
+
func=cst.Attribute(
|
528
|
+
value=cst.Name("strawberry"),
|
529
|
+
attr=cst.Name("Schema"),
|
530
|
+
),
|
531
|
+
args=args,
|
532
|
+
)
|
533
|
+
|
534
|
+
if is_apollo_federation:
|
535
|
+
args.append(
|
536
|
+
cst.Arg(
|
537
|
+
keyword=cst.Name("enable_federation_2"),
|
538
|
+
value=cst.Name("True"),
|
539
|
+
equal=cst.AssignEqual(
|
540
|
+
cst.SimpleWhitespace(""), cst.SimpleWhitespace("")
|
541
|
+
),
|
542
|
+
)
|
543
|
+
)
|
544
|
+
schema_call = cst.Call(
|
545
|
+
func=cst.Attribute(
|
546
|
+
value=cst.Attribute(
|
547
|
+
value=cst.Name(value="strawberry"),
|
548
|
+
attr=cst.Name(value="federation"),
|
549
|
+
),
|
550
|
+
attr=cst.Name(value="Schema"),
|
551
|
+
),
|
552
|
+
args=args,
|
553
|
+
)
|
554
|
+
|
268
555
|
return cst.SimpleStatementLine(
|
269
556
|
body=[
|
270
557
|
cst.Assign(
|
271
558
|
targets=[cst.AssignTarget(cst.Name("schema"))],
|
272
|
-
value=
|
273
|
-
func=cst.Attribute(
|
274
|
-
value=cst.Name("strawberry"),
|
275
|
-
attr=cst.Name("Schema"),
|
276
|
-
),
|
277
|
-
args=args,
|
278
|
-
),
|
559
|
+
value=schema_call,
|
279
560
|
)
|
280
561
|
]
|
281
562
|
)
|
282
563
|
|
283
564
|
|
284
|
-
@dataclasses.dataclass(frozen=True)
|
285
|
-
class Import:
|
286
|
-
module: str | None
|
287
|
-
imports: tuple[str]
|
288
|
-
|
289
|
-
def to_cst(self) -> cst.Import | cst.ImportFrom:
|
290
|
-
if self.module is None:
|
291
|
-
return cst.Import(
|
292
|
-
names=[cst.ImportAlias(name=cst.Name(name)) for name in self.imports]
|
293
|
-
)
|
294
|
-
|
295
|
-
return cst.ImportFrom(
|
296
|
-
module=cst.Name(self.module),
|
297
|
-
names=[cst.ImportAlias(name=cst.Name(name)) for name in self.imports],
|
298
|
-
)
|
299
|
-
|
300
|
-
|
301
565
|
def _get_union_definition(definition: UnionTypeDefinitionNode) -> cst.Assign:
|
302
566
|
name = definition.name.value
|
303
567
|
|
@@ -430,6 +694,12 @@ def codegen(schema: str) -> str:
|
|
430
694
|
|
431
695
|
object_types: dict[str, cst.ClassDef] = {}
|
432
696
|
|
697
|
+
# when we encounter a extend schema @link ..., we check if is an apollo federation schema
|
698
|
+
# and we use this variable to keep track of it, but at the moment the assumption is that
|
699
|
+
# the schema extension is always done at the top, this might not be the case all the
|
700
|
+
# time
|
701
|
+
is_apollo_federation = False
|
702
|
+
|
433
703
|
for definition in document.definitions:
|
434
704
|
if isinstance(
|
435
705
|
definition,
|
@@ -440,7 +710,9 @@ def codegen(schema: str) -> str:
|
|
440
710
|
ObjectTypeExtensionNode,
|
441
711
|
),
|
442
712
|
):
|
443
|
-
class_definition = _get_class_definition(
|
713
|
+
class_definition = _get_class_definition(
|
714
|
+
definition, is_apollo_federation, imports
|
715
|
+
)
|
444
716
|
|
445
717
|
object_types[definition.name.value] = class_definition
|
446
718
|
|
@@ -478,6 +750,11 @@ def codegen(schema: str) -> str:
|
|
478
750
|
definitions.append(cst.EmptyLine())
|
479
751
|
definitions.append(scalar_definition)
|
480
752
|
definitions.append(cst.EmptyLine())
|
753
|
+
elif isinstance(definition, SchemaExtensionNode):
|
754
|
+
is_apollo_federation = any(
|
755
|
+
_is_federation_link_directive(directive)
|
756
|
+
for directive in definition.directives
|
757
|
+
)
|
481
758
|
else:
|
482
759
|
raise NotImplementedError(f"Unknown definition {definition}")
|
483
760
|
|
@@ -496,6 +773,7 @@ def codegen(schema: str) -> str:
|
|
496
773
|
root_query_name=root_query_name,
|
497
774
|
root_mutation_name=root_mutation_name,
|
498
775
|
root_subscription_name=root_subscription_name,
|
776
|
+
is_apollo_federation=is_apollo_federation,
|
499
777
|
)
|
500
778
|
|
501
779
|
if schema_definition:
|
strawberry/types/nodes.py
CHANGED
@@ -9,6 +9,7 @@ If a node has only one useful value, it's value is inlined.
|
|
9
9
|
If a list of nodes have unique names, it's transformed into a mapping.
|
10
10
|
Note Python dicts maintain ordering (for all supported versions).
|
11
11
|
"""
|
12
|
+
|
12
13
|
from __future__ import annotations
|
13
14
|
|
14
15
|
import dataclasses
|
@@ -22,7 +22,7 @@ strawberry/channels/__init__.py,sha256=9oRdAT7uIYETF-23gZ5NteOXSjwkUtwRmwu3YCFv1
|
|
22
22
|
strawberry/channels/handlers/base.py,sha256=tgXLjKCHrP-8yOJV-qo9iOMJUgDe2wW6rGs_nDZFi80,7869
|
23
23
|
strawberry/channels/handlers/graphql_transport_ws_handler.py,sha256=UKfxRoo1pEv1CBHhkhSTkbztJlfU-OoksFwyLktj8xc,1945
|
24
24
|
strawberry/channels/handlers/graphql_ws_handler.py,sha256=PHRkwnXt3tY4E0XBVHh4hpTyFBt9jQfTreBUmEo9vhI,2497
|
25
|
-
strawberry/channels/handlers/http_handler.py,sha256=
|
25
|
+
strawberry/channels/handlers/http_handler.py,sha256=c1Jpf7L4b7LvSaOolP2e9TWcEM5IqHXhScK0kZrhSg8,9561
|
26
26
|
strawberry/channels/handlers/ws_handler.py,sha256=sHL44eay4tNoKzkrRn3WewSYH-3ZSJzxJpmBJ-aTkeM,4650
|
27
27
|
strawberry/channels/router.py,sha256=dyOBbSF8nFiygP0zz6MM14mhkvFQAEbbLBXzcpubSHM,1927
|
28
28
|
strawberry/channels/testing.py,sha256=IWj1CuIS3vOo2f2fw0W-0GCz-YSs7QSAAscC6suqtiI,5668
|
@@ -127,14 +127,14 @@ strawberry/fastapi/handlers/graphql_ws_handler.py,sha256=U84lpd7Lxl2bEnFr9Qroswm
|
|
127
127
|
strawberry/fastapi/router.py,sha256=o55NLbL7n-LoKM30eUCJgzyPlRYPsybWb07L8-fcZZo,11573
|
128
128
|
strawberry/federation/__init__.py,sha256=FeUxLiBVuk9TKBmJJi51SeUaI8c80rS8hbl9No-hjII,535
|
129
129
|
strawberry/federation/argument.py,sha256=5qyJYlQGEZd6iXWseQ7dnnejCYj5HyglfK10jOCIJBg,802
|
130
|
-
strawberry/federation/enum.py,sha256=
|
131
|
-
strawberry/federation/field.py,sha256=
|
130
|
+
strawberry/federation/enum.py,sha256=wpM1z2NOkoBCqTVyD6vJJXoNlcnrNIERt2YpB4R9ybU,2977
|
131
|
+
strawberry/federation/field.py,sha256=HHOs8FL52_jxuYxPiR2EwJsXtk7LyapJeBUPqvJ6pCg,6184
|
132
132
|
strawberry/federation/mutation.py,sha256=0lV5HJwgw4HYR_59pwxWqnPs342HwakTNMc98w5Hb-c,43
|
133
|
-
strawberry/federation/object_type.py,sha256=
|
134
|
-
strawberry/federation/scalar.py,sha256=
|
133
|
+
strawberry/federation/object_type.py,sha256=nQLFbuS6KvAnGCGie_KXxjmtJC6gnQZFh5FAwFHTH-c,9037
|
134
|
+
strawberry/federation/scalar.py,sha256=jrSvaqfGEnj_Vl7b-LYMdq_NDMwfdu0t5Pfzlw5WAmU,3807
|
135
135
|
strawberry/federation/schema.py,sha256=Z9ZtDav5dwFQvts1XeeT4xoKeIhDwdGoocBUX-y233Y,13707
|
136
136
|
strawberry/federation/schema_directive.py,sha256=TpqoVeN3-iE-acndIRAVyU4LIXh6FTHz-Pv2kI8zGu0,1719
|
137
|
-
strawberry/federation/schema_directives.py,sha256=
|
137
|
+
strawberry/federation/schema_directives.py,sha256=gpJwpPg_UNiJdQtolY3qQ0lyVYD215c3JFyYH0yFRG8,6131
|
138
138
|
strawberry/federation/types.py,sha256=mM70g1aLgjplOc3SdtuJjy7NAKFLv35Z4BDC4s_j5us,301
|
139
139
|
strawberry/federation/union.py,sha256=QXeh-nhApqFtXa3To9MX_IwvtwErZBhWYnssUK7JB2E,1005
|
140
140
|
strawberry/field.py,sha256=2LvlnoQyRQHYcZV2Yy1E-2DYKXpOROUa8jyWgxG2QsI,18883
|
@@ -165,7 +165,7 @@ strawberry/parent.py,sha256=mCnJcLBQKwtokXcGxkm5HqGg7Wkst97rn61ViuaotrM,829
|
|
165
165
|
strawberry/permission.py,sha256=Rm_dft2AGZCSgBRQuxWEfncgWj88nmSdL-5mJ6TvHeA,5917
|
166
166
|
strawberry/printer/__init__.py,sha256=DmepjmgtkdF5RxK_7yC6qUyRWn56U-9qeZMbkztYB9w,62
|
167
167
|
strawberry/printer/ast_from_value.py,sha256=MFIX2V51d9ocRvD0Njemjk8YIzKh2BB1g2iUcX6a3d8,4946
|
168
|
-
strawberry/printer/printer.py,sha256=
|
168
|
+
strawberry/printer/printer.py,sha256=PXuYfAJxMU92aRETeSixhJjk-XUDu2ZFAFwkogzTu_Y,17461
|
169
169
|
strawberry/private.py,sha256=fS5IbVUajzr-6NcR41yt4ukXDPAMzwe1WPNdRJP5wWk,534
|
170
170
|
strawberry/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
171
171
|
strawberry/quart/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
@@ -194,7 +194,7 @@ strawberry/schema/types/__init__.py,sha256=oHO3COWhL3L1KLYCJNY1XFf5xt2GGtHiMC-Ua
|
|
194
194
|
strawberry/schema/types/base_scalars.py,sha256=Z_BmgwLicNexLipGyw6MmZ7OBnkGJU3ySgaY9SwBWrw,1837
|
195
195
|
strawberry/schema/types/concrete_type.py,sha256=HB30G1hMUuuvjAvfSe6ADS35iI_T_wKO-EprVOWTMSs,746
|
196
196
|
strawberry/schema/types/scalar.py,sha256=SVJ8HiKncCvOw2xwABI5xYaHcC7KkGHG-tx2WDtSoCA,2802
|
197
|
-
strawberry/schema_codegen/__init__.py,sha256=
|
197
|
+
strawberry/schema_codegen/__init__.py,sha256=2TO50tEZXQkH9yLUEBCZXEd0tUuzZ_XcY5dcKaf26ps,24040
|
198
198
|
strawberry/schema_directive.py,sha256=GxiOedFB-RJAflpQNUZv00C5Z6gavR-AYdsvoCA_0jc,1963
|
199
199
|
strawberry/starlite/__init__.py,sha256=v209swT8H9MljVL-npvANhEO1zz3__PSfxb_Ix-NoeE,134
|
200
200
|
strawberry/starlite/controller.py,sha256=x6Mm3r36cRfzo6hz9B4AYWbVh2QlYtndYcXFOr_3THM,11860
|
@@ -223,7 +223,7 @@ strawberry/types/fields/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG
|
|
223
223
|
strawberry/types/fields/resolver.py,sha256=Lwy2XVKnTbDyibk9pPsWsunlm22akXjy-gKCg22Tp1A,14130
|
224
224
|
strawberry/types/graphql.py,sha256=3SWZEsa0Zy1eVW6vy75BnB7t9_lJVi6TBV3_1j3RNBs,687
|
225
225
|
strawberry/types/info.py,sha256=b1ZWW_wUop6XrGNcGHKBQeUYjlX-y8u3s2Wm_XhKPYI,3412
|
226
|
-
strawberry/types/nodes.py,sha256=
|
226
|
+
strawberry/types/nodes.py,sha256=5tTYmxGpVDshbydicHTTBWEiUe8A7p7mdiaSV8Ry80Y,5027
|
227
227
|
strawberry/types/type_resolver.py,sha256=F0z_geS4VEun8EhD571LaTgI8ypjCeLfp910gF0Q3MY,6280
|
228
228
|
strawberry/types/types.py,sha256=t5MOV4xiutPL2Ka02u3Z2V3eqy2R2JPYQxsqCnY2Aqk,7139
|
229
229
|
strawberry/union.py,sha256=5DOzsmsrI4tU4aXO7BBEN-570ABQ9_VkRDvUJUrIofY,9926
|
@@ -241,8 +241,8 @@ strawberry/utils/logging.py,sha256=flS7hV0JiIOEdXcrIjda4WyIWix86cpHHFNJL8gl1y4,7
|
|
241
241
|
strawberry/utils/operation.py,sha256=Um-tBCPl3_bVFN2Ph7o1mnrxfxBes4HFCj6T0x4kZxE,1135
|
242
242
|
strawberry/utils/str_converters.py,sha256=avIgPVLg98vZH9mA2lhzVdyyjqzLsK2NdBw9mJQ02Xk,813
|
243
243
|
strawberry/utils/typing.py,sha256=Qxz1LwyVsNGV7LQW1dFsaUbsswj5LHBOdKLMom5eyEA,13491
|
244
|
-
strawberry_graphql-0.
|
245
|
-
strawberry_graphql-0.
|
246
|
-
strawberry_graphql-0.
|
247
|
-
strawberry_graphql-0.
|
248
|
-
strawberry_graphql-0.
|
244
|
+
strawberry_graphql-0.223.0.dist-info/LICENSE,sha256=m-XnIVUKqlG_AWnfi9NReh9JfKhYOB-gJfKE45WM1W8,1072
|
245
|
+
strawberry_graphql-0.223.0.dist-info/METADATA,sha256=_W5hqZRCE54piac0oXb4DUZtiJGTYgTFZRDQvlhWEaU,7740
|
246
|
+
strawberry_graphql-0.223.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
|
247
|
+
strawberry_graphql-0.223.0.dist-info/entry_points.txt,sha256=Nk7-aT3_uEwCgyqtHESV9H6Mc31cK-VAvhnQNTzTb4k,49
|
248
|
+
strawberry_graphql-0.223.0.dist-info/RECORD,,
|
File without changes
|
File without changes
|
{strawberry_graphql-0.221.1.dist-info → strawberry_graphql-0.223.0.dist-info}/entry_points.txt
RENAMED
File without changes
|