def test_simple_user_model_conversion(self): """ create a simple user model, pass to user_to_dict() assert resulting dictionary is correct """ sample_user = User(username='******', email='*****@*****.**', googleID='1').put() dict_user = user_to_dict(sample_user.get()) self.assertEqual(dict_user['username'], 'man') self.assertEqual(dict_user['email'], '*****@*****.**') self.assertEqual(dict_user['googleID'], '1')
def test_simple_user_model_conversion(self): """ create a simple user model, pass to user_to_dict() assert resulting dictionary is correct """ sample_user = User( username='******', email='*****@*****.**', googleID='1' ).put() dict_user = user_to_dict(sample_user.get()) self.assertEqual(dict_user['username'], 'man') self.assertEqual(dict_user['email'], '*****@*****.**') self.assertEqual(dict_user['googleID'], '1')
def put(self, user_id): """ a put request updates an existing user """ parser = reqparse.RequestParser() parser.add_argument('formalName', type=str) parser.add_argument('bio', type=str) #parser.add_argument('plan', type=str) args = parser.parse_args() user = ndb.Key(User, int(user_id)).get() if not user: return {'message':'user does not exist'}, 400 else: if args['formalName']: user.formalName = args['formalName'] if args['bio']: user.bio = args['bio'] user.put() return user_to_dict(user)
def put(self, user_id): """ a put request updates an existing user """ parser = reqparse.RequestParser() parser.add_argument('formalName', type=str) parser.add_argument('bio', type=str) #parser.add_argument('plan', type=str) args = parser.parse_args() user = ndb.Key(User, int(user_id)).get() if not user: return {'message': 'user does not exist'}, 400 else: if args['formalName']: user.formalName = args['formalName'] if args['bio']: user.bio = args['bio'] user.put() return user_to_dict(user)
def get(self, userID=None): """ handle the get request userID; return none if userID does not exist """ try: if userID is None: googleID = users.get_current_user().user_id() user = get_user_by_google_id(googleID) else: user = get_user(userID) if not user: raise Exception('that user does not exist') data = user_to_dict(user) except Exception as e: data = {'error' : str(e)} self.write_json(data)
def get(self, user_id = None): """ return the user if user_id provided, else return the users queryset """ if user_id: user = ndb.Key(User, int(user_id)).get() if not user: return {'message':'user does not exist'}, 400 else: google_user = users.get_current_user() data = user_to_dict(user) return data else: all_users = User.query() user_list = [] for user in all_users: user_dict = user.to_dict() user_dict['id'] = user.key.id() user_list.append(user_dict) return {'users' : user_list}
def get(self, user_id=None): """ return the user if user_id provided, else return the users queryset """ if user_id: user = ndb.Key(User, int(user_id)).get() if not user: return {'message': 'user does not exist'}, 400 else: google_user = users.get_current_user() data = user_to_dict(user) return data else: all_users = User.query() user_list = [] for user in all_users: user_dict = user.to_dict() user_dict['id'] = user.key.id() user_list.append(user_dict) return {'users': user_list}