Skip to content

Commit 8c7d01d

Browse files
committed
Simplify auth metadata plugin
1 parent bd4f236 commit 8c7d01d

File tree

1 file changed

+13
-11
lines changed
  • packages/jumpstarter/jumpstarter/config

1 file changed

+13
-11
lines changed

packages/jumpstarter/jumpstarter/config/grpc.py

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,17 @@
44

55

66
def call_credentials(kind: str, metadata: ObjectMeta, token: str):
7-
class AuthMetadataPlugin(grpc.AuthMetadataPlugin):
8-
def __call__(context: grpc.AuthMetadataContext, callback: grpc.AuthMetadataPluginCallback):
9-
callback(
10-
[
11-
("jumpstarter.dev/kind", kind),
12-
("jumpstarter.dev/namespace", metadata.namespace),
13-
("jumpstarter.dev/name", metadata.name),
14-
],
15-
None,
16-
)
7+
def metadata_call_credentials(context: grpc.AuthMetadataContext, callback: grpc.AuthMetadataPluginCallback):
8+
callback(
9+
[
10+
("jumpstarter.dev/kind", kind),
11+
("jumpstarter.dev/namespace", metadata.namespace),
12+
("jumpstarter.dev/name", metadata.name),
13+
],
14+
None,
15+
)
1716

18-
return grpc.composite_call_credentials(AuthMetadataPlugin, grpc.access_token_call_credentials(token))
17+
return grpc.composite_call_credentials(
18+
metadata_call_credentials,
19+
grpc.access_token_call_credentials(token),
20+
)

0 commit comments

Comments
 (0)