def _save_follower_relation_to_db(user, follower_list):
follower_id_list = []
refined_follower_list = []
rejected_fields = ['domain','firstname','lastname','cover_url','thumbnail_background_url','avatars']
for follower in follower_list:
follower_id_list.append(follower['id'])
if not user_manager.user_seen(follower['id']):
user_manager.add_user(follower['id'])
bfs_queue.put(follower['id'])
for field in rejected_fields:
follower.pop(field, None)
refined_follower_list.append(follower)
#save users into users collection
user_collection = mydb.users
if len(refined_follower_list)>0:
user_collection.insert_many(refined_follower_list)
#add the following relation of the given user into user relation collection
user_relation_collection = mydb.user_relation
relation = {}
relation['user'] = user
relation['followers'] = follower_id_list
user_relation_collection.insert(relation)
def addUser():
username = request.form.get('username')
password = request.form.get('password')
age = request.form.get('age')
mobile_number = request.form.get('mobile_number')
email = request.form.get('email')
response = user_manager.add_user(password, username, email, age,
mobile_number)
if response[0] == 0:
return jsonify({"ERROR_CODE": response[0], "USER_ID": response[1]})
else:
return jsonify({"ERROR_CODE": response[0]})
def test():
user_manager.clear_all()
print user_manager.user_seen(5)
user_manager.add_user(5)
print user_manager.user_seen(5)
