Skip to content

Commit e4fbbcb

Browse files
committed
Do not parse unsuccessful /auth response
1 parent 86c0b03 commit e4fbbcb

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

yamcs-client/src/yamcs/client/core/__init__.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from google.protobuf import timestamp_pb2
2020
from yamcs.client.core.auth import APIKeyCredentials, Credentials
2121
from yamcs.client.core.context import Context
22+
from yamcs.client.core.exceptions import YamcsError
2223
from yamcs.client.core.futures import WebSocketSubscriptionFuture
2324
from yamcs.client.core.helpers import (
2425
delimit_protobuf,
@@ -273,9 +274,12 @@ def get_auth_info(self) -> AuthInfo:
273274
self.ctx.auth_root, # Full URL, so don't use get_proto
274275
headers={"Accept": "application/protobuf"},
275276
)
276-
message = auth_pb2.AuthInfo()
277-
message.ParseFromString(response.content)
278-
return AuthInfo(message)
277+
if 200 <= response.status_code < 300:
278+
message = auth_pb2.AuthInfo()
279+
message.ParseFromString(response.content)
280+
return AuthInfo(message)
281+
else:
282+
raise YamcsError(f"{response.status_code} Server Error")
279283

280284
def get_user_info(self) -> UserInfo:
281285
"""

0 commit comments

Comments
 (0)