Chat finish; add invitation; friend; see online users if he is your friend

profiles/models.py

@@ -26,3 +26,67 @@ class BlockModel(models.Model):
 
 	def __str__(self):
 		return "blocker_id: " + str(self.blocker) + ", blocked_id: " + str(self.blocked)
+
+class AskFriendModel(models.Model):
+	asker = IntegerField(primary_key=False)
+	asked = IntegerField(primary_key=False)
+
+	def getAsked(self, asker):
+		askeds = []
+
+		for ask in AskFriendModel.objects.filter(asker=asker):
+			askeds.append(ask.asked)
+
+		return askeds
+
+	def getAsker(self, asked):
+		askers = []
+
+		for ask in AskFriendModel.objects.filter(asked=asked):
+			askers.append(ask.asker)
+
+		return askers
+
+	def deleteAsk(self, asker, asked):
+
+		deleted = AskFriendModel.objects.filter(asker=asker, asked=asked)
+
+		if (deleted.count() == 0 or not deleted):
+			return False
+
+		deleted.delete()
+		return True
+
+class FriendModel(models.Model):
+	user_id1 = IntegerField(primary_key=False)
+	user_id2 = IntegerField(primary_key=False)
+
+	def getFriends(self, user_id):
+		friends = []
+
+		for friend in FriendModel.objects.filter(user_id1=user_id):
+			friends.append(friend.user_id2)
+
+		for friend in FriendModel.objects.filter(user_id2=user_id):
+			friends.append(friend.user_id1)
+
+		return friends
+
+	def isFriend(self, user_id1, user_id2):
+
+		return user_id2 in self.getFriends(user_id1)
+
+	def deleteFriend(self, user_id1, user_id2):
+
+		first = FriendModel.objects.filter(user_id1=user_id1, user_id2=user_id2)
+		if (first.count() == 1):
+			first.delete()
+			return True
+
+		second = FriendModel.objects.filter(user_id1=user_id2, user_id2=user_id1)
+		if (second.count() == 1):
+			second.delete()
+			return True
+
+		return False
+