def addFriendView(request):
creator = request.user
friend_name = request.GET.get('friend', '')
action = request.GET.get('accepted', '')
friend = dbUser.objects.get(username=friend_name)
if (action == 'false'):
Friendship.objects.filter(creator=friend, friend=creator).delete()
return redirect('/friends/')
if not Friendship.objects.filter(creator=creator, friend=friend):
if Friendship.objects.filter(creator=friend, friend=creator).exists():
friendship1 = Friendship.objects.get(creator=friend,
friend=creator)
friendship1.status = 'accepted'
friendship2 = Friendship(creator=creator,
friend=friend,
status='accepted')
friendship1.save()
notification1 = Notification(
user=friend,
user2=creator,
notification_type="friend-accept",
date=datetime.date.today(),
time=datetime.datetime.now().strftime('%H:%M:%S'),
notification="You are now friends with %s" % creator.username)
notification2 = Notification(
user=creator,
user2=friend,
notification_type="friend-accept",
date=datetime.date.today(),
time=datetime.datetime.now().strftime('%H:%M:%S'),
notification="You are now friends with %s" % friend.username)
notification1.save()
notification2.save()
else:
friendship2 = Friendship(creator=creator, friend=friend)
notification = Notification(
user=friend,
user2=creator,
notification_type="friend-request",
date=datetime.date.today(),
time=datetime.datetime.now().strftime('%H:%M:%S'),
notification="You have a friend request from %s" %
creator.username)
notification.save()
friendship2.save()
sent = True
context_dict = {'sent': sent}
return redirect('/friends/')
def add_friend(current_user, friend):
user2 = User.query.filter_by(username=friend).first()
if user2 is None:
return jsonify({'msg': 'User not found.'})
user1 = current_user
#print(user2)
query = Friendship.query.filter((
(Friendship.user_a == current_user.public_id)
& (Friendship.user_b == user2.public_id)) | (
(Friendship.user_b == current_user.public_id)
& (Friendship.user_a == user2.public_id))).first()
#print('Query is',query)
if query is None:
if user1.id > user2.id:
user1, user2 = user2, user1
new_friendship = Friendship(user_a=user1.public_id,
user_b=user2.public_id,
status=1,
action_user=current_user.public_id)
#print('New is', new_friendship)
db.session.add(new_friendship)
db.session.commit()
return jsonify({'msg': 'Request Sent.'})
else:
query.status = 0 if query.status == 1 else 1
db.session.commit()
return jsonify({'msg': 'Invalid Request.'})
def friend_add(friend_id):
if session.get('user'):
new_friendship = Friendship(friend_id, session['user_id'])
db.session.add(new_friendship)
db.session.commit()
return redirect(url_for('get_friends'))
else:
return redirect(url_for('login'))
def add_facebook_friend(user_id1, user_id2):
row1 = Friendship(user_id1=user_id1, user_id2=user_id2, is_facebook=True)
row2 = Friendship(user_id1=user_id2, user_id2=user_id1, is_facebook=True)
session.add(row1)
session.add(row2)
session.commit()
def add_friend(user_id1, user_id2):
row1 = Friendship(user_id1=user_id1, user_id2=user_id2)
row2 = Friendship(user_id1=user_id2, user_id2=user_id1)
session.add(row1)
session.add(row2)
session.commit()
def setUp(self):
self.user1 = User.objects.get(pk=1)
self.user2 = User.objects.get(pk=2)
friendship = Friendship(from_user=self.user1, to_user=self.user2)
friendship.save()
def setUp(self):
super(TestFriendshipAcceptReject, self).setUp()
users = User.query.all()
friendship = Friendship(sender=users[0], receiver=users[1])
self.db.session.add(friendship)
self.db.session.commit()
