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()
