Merge pull request #169 from scop/feat/usersession-close-suppressions

Suppress login required and not supported on UserSession close
Salamek · Jan 15, 2023 · aefc8df · aefc8df
2 parents ab0ffb0 + f15812c
commit aefc8df
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/huawei_lte_api/api/User.py b/huawei_lte_api/api/User.py
@@ -16,6 +16,7 @@
     LoginErrorUsernamePasswordWrongException, \
     LoginErrorUsernameWrongException, \
     LoginErrorPasswordWrongException, \
+    ResponseErrorLoginRequiredException, \
     ResponseErrorNotSupportedException
 
 DEFAULT_USERNAME = 'admin'
@@ -27,7 +28,11 @@ def __init__(self, session: Session, username: str = DEFAULT_USERNAME, password:
         self.user.login(username, password, True)
 
     def close(self) -> None:
-        self.user.logout()
+        try:
+            self.user.logout()
+        except (ResponseErrorLoginRequiredException, ResponseErrorNotSupportedException):
+            # Idempotency/nothing further to do, suppress
+            pass
 
     def __enter__(self) -> 'UserSession':
         return self