diff options
| author | Vitalii Gridnev <gridnevvvit@gmail.com> | 2022-07-05 11:29:53 +0300 |
|---|---|---|
| committer | Vitalii Gridnev <gridnevvvit@gmail.com> | 2022-07-05 11:29:53 +0300 |
| commit | 10f39882e03a53066a41abc9f60dbe36843523b2 (patch) | |
| tree | 7b27e6d6bb8a9c7e841c94f03b211f6eda08535f | |
| parent | bfc8c84ea20b4a366f558f0777d01676f376a669 (diff) | |
| download | ydb-10f39882e03a53066a41abc9f60dbe36843523b2.tar.gz | |
support service account file KIKIMR-15243
ref:b3f1a611677f38269ef140532541edf57c55038e
| -rw-r--r-- | ydb/public/sdk/python/ydb/iam/auth.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/ydb/public/sdk/python/ydb/iam/auth.py b/ydb/public/sdk/python/ydb/iam/auth.py index 8a21346d05c..5bc59ef3c0e 100644 --- a/ydb/public/sdk/python/ydb/iam/auth.py +++ b/ydb/public/sdk/python/ydb/iam/auth.py @@ -204,6 +204,9 @@ class TokenServiceCredentials(IamTokenCredentials): "Install jwt & yandex python cloud library to use service account credentials provider" ) + self._iam_token_service_pb2 = iam_token_service_pb2 + self._iam_token_service_pb2_grpc = iam_token_service_pb2_grpc + def _channel_factory(self): return grpc.secure_channel( self._iam_endpoint, @@ -218,7 +221,7 @@ class TokenServiceCredentials(IamTokenCredentials): def _get_iam_token(self): with self._channel_factory() as channel: tracing.trace(self.tracer, {"iam_token.from_service": True}) - stub = iam_token_service_pb2_grpc.IamTokenServiceStub(channel) + stub = self._iam_token_service_pb2_grpc.IamTokenServiceStub(channel) response = stub.Create( self._get_token_request(), timeout=self._get_token_request_timeout ) @@ -268,7 +271,7 @@ class JWTIamCredentials(TokenServiceCredentials, BaseJWTCredentials): BaseJWTCredentials.__init__(self, account_id, access_key_id, private_key) def _get_token_request(self): - return iam_token_service_pb2.CreateIamTokenRequest( + return self._iam_token_service_pb2.CreateIamTokenRequest( jwt=get_jwt( self._account_id, self._access_key_id, |