def _process_token(self, request_url, response, set_token=True, persist_token=False, token_path=None): try: token_dict = response.json() response.raise_for_status() except Exception: if response.status_code in [401, 403]: raise AuthenticationError(request_url, response, "Invalid credentials.", response.status_code) raise AuthenticationError( request_url, response, "Login failed.\nSee http://docs.polyaxon.com/faqs/authentication/ for help", response.status_code) token = token_dict.get('token') if set_token: self.config.token = token if persist_token: self._persist_token(token, token_path) return token
def get_user(self, token=None): token = token or self.config.token request_url = self._get_http_url() response = self.transport.get(request_url, headers={ "Authorization": "{} {}".format( self.config.authentication_type, token) }) try: user_dict = response.json() response.raise_for_status() except Exception: if response.status_code in [401, 403]: raise AuthenticationError( request_url, response, "Invalid Token.\nSee http://docs.polyaxon.com/faqs/authentication/ for help", response.status_code) raise AuthenticationError( request_url, response, "Login failed.\nSee http://docs.polyaxon.com/faqs/authentication/ for help", response.status_code) return self.prepare_results(response_json=user_dict, config=UserConfig)
def login(self, credentials): request_url = self._build_url(self._get_http_url(), 'token') response = requests.post(request_url, json=credentials.to_dict()) try: token_dict = response.json() response.raise_for_status() except Exception: if response.status_code == 401: raise AuthenticationError(request_url, response, "Invalid credentials.", 401) raise AuthenticationError( request_url, response, "Login failed.\nSee http://docs.polyaxon.com/faqs/authentication/ for help", response.status_code) return token_dict.get(self.authentication_type)
def get_user(self, token=None): token = token or self.token request_url = self._get_http_url() response = self.get(request_url, headers={ "Authorization": "{} {}".format(self.authentication_type, token) }) try: user_dict = response.json() response.raise_for_status() except Exception: if response.status_code == 401: raise AuthenticationError( request_url, response, "Invalid Token.\nSee http://docs.polyaxon.com/faqs/authentication/ for help", 401) raise AuthenticationError( request_url, response, "Login failed.\nSee http://docs.polyaxon.com/faqs/authentication/ for help", response.status_code) return UserConfig.from_dict(user_dict)