Add token login method

Author: D34DPlayer

tb_rest_client/rest_client_base.py

@@ -118,6 +118,15 @@ def public_login(self, public_id):
         self.__save_token(token_json)
         self.__load_configuration()
 
+    def token_login(self, token, refresh_token=None):
+        token_json = {
+            "token": token,
+            "refresh_token": refresh_token,
+        }
+
+        self.__save_token(token_json)
+        self.__load_configuration()
+                
     def refresh(self):
         if not self.token_info["refreshToken"]:
             return

tb_rest_client/rest_client_pe.py

@@ -33,6 +33,10 @@ def public_login(self, public_id):
         super(RestClientPE, self).public_login(public_id=public_id)
         self.__load_controllers()
 
+    def token_login(self, token, refresh_token=None):
+        super(RestClientPE, self).token_login(token=token, refresh_token=refresh_token)
+        self.__load_controllers()
+
     # Self Registration Controller
     def get_privacy_policy(self, ):
         return self.self_registration_controller.get_privacy_policy_using_get()