def get_user(self, access_token): request_url = self._get_url() # This is a special case client, # because auth_token is not set yet (this is how we verify it) # So do not use the shared base client for this! response = requests.get(request_url, headers={"Authorization": "Bearer {}".format(access_token)}) # TODO: replace with self.get(...) 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)
def test_project_config(self): config_dict = { 'username': '******', 'uid': 'sdfsdf098sdf80s9dSDF800', 'email': '*****@*****.**' } config = UserConfig.from_dict(config_dict) assert config.to_dict() == config_dict
def test_get_user(self): user = UserConfig('user', '*****@*****.**').to_dict() httpretty.register_uri(httpretty.GET, AuthClient._build_url(self.client.base_url, AuthClient.ENDPOINT), body=json.dumps(user), content_type='application/json', status=200) user_result = self.client.get_user() assert user == user_result.to_dict()
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)
def retrieve(self, request, *args, **kwargs): user = request.user return Response(UserConfig.obj_to_dict(user))
def retrieve(self, request, *args, **kwargs): user = request.user return Response(UserConfig.obj_to_dict(user))
def test_user_config(self): config_dict = {'username': '******', 'email': '*****@*****.**'} config = UserConfig.from_dict(config_dict) assert config.to_dict() == config_dict