def test_friends(self): u1 = User(username='******', email='*****@*****.**') u2 = User(username='******', email='*****@*****.**') db.session.add(u1) db.session.add(u2) db.session.commit() u1.add_friend(u1) db.session.commit() self.assertEqual(u1.friends.all(), []) u1.add_friend(u2) db.session.commit() self.assertEqual(u1.friends.all(), [u2]) self.assertEqual(u2.friends.all(), [u1])
def allFriends(): form = SendMassage_addFriend() state = '' friends_list = [] friends_id = User.get_friend(current_user) for id in friends_id: friend = User.query.get(id) friends_list.append(friend) uni = current_user.university field = current_user.studyField suggested = [] if current_user.complete == True and uni != 'university' and field != 'studyField': users = User.query.all() for user in users: if user != current_user and user not in friends_list: if user.complete == True and user.university == uni and user.studyField == field: suggested.append(user) error = '' if (request.method == 'POST') and ('add' in request.form): friendusername = request.form['addfriend'] new_friend = User.query.filter_by(username=friendusername).first() if (new_friend): if new_friend.id not in friends_id: User.add_friend(current_user, new_friend.id) db.session.commit() flash('Friend aded!') # redirect(url_for('allFriends')) return redirect(url_for('allFriends')) else: flash('Friend already exists!') error = 'exists' else: error = 'there is no account with this username !!' return render_template('landing_page/Friends.html', title='Friends', Friends=friends_list, form=form, error=error, state=state, suggested=suggested)