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}
