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)