def add_friend(phone): user_id = session.get('user_id') user_model = User.query.filter(User.id == user_id).first() friend_model = User.query.filter(User.phone == phone).first() if request.method == 'GET': return render_template('search.html', user=user_model, friend=friend_model) else: test = Friend.query.filter(Friend.friend_id == friend_model.id, Friend.user_id == user_id).first() if test: print('1') return render_template('search.html', user=user_model, friend=friend_model) else: print('2') friend = Friend(friend_id=friend_model.id) friend.user_id = session.get('user_id') db.session.add(friend) db.session.commit() print('3') return render_template('search.html', user=user_model, friend=friend_model)
def save_friends(user_id, friend_jarray): for name in friend_jarray: friend = Friend() friend.user_id = user_id friend.friends = name friend.save()