def user_addasfriend_view(request, id): current_user = request.user request_to_user = User.objects.get(id=id) Friend.add_friend(current_user, request_to_user) friend = Friend.objects.get(request_from=current_user, request_to=request_to_user, accepted=False) Notification.notify(friend, request_to_user) return redirect(reverse("user", args=[id]))
def test_models(self): friends, friend_requests, follow = Friend.are_connected( self.user_1, self.user_2) self.assertEqual(friends, 0) self.assertEqual(friend_requests, 0) self.assertEqual(follow, False) friend_list = Friend.friends(self.user_1) self.assertEqual(len(friend_list), 0) self.client.login(username='******', password='******') response = self.client.post(reverse('posts'), {'status': "1.1"}) self.assertEqual(response.status_code, 302) self.client.logout() self.client.login(username='******', password='******') response = self.client.post(reverse('comment', args=[1]), {'comment': "5.1.(1.2)"}) self.assertEqual(response.status_code, 302) self.client.logout() self.client.login(username='******', password='******') notification = Notification.objects.get(id=1) message = notification.display() self.assertEqual(message, "User def commented on your post") self.client.logout()
def test_models(self): friends, friend_requests, follow = Friend.are_connected( self.user_1, self.user_2) self.assertEqual(friends, 0) self.assertEqual(friend_requests, 0) self.assertEqual(follow, False) friend_list = Friend.friends(self.user_1) self.assertEqual(len(friend_list), 0) self.client.login(username='******', password='******') response = self.client.post(reverse('posts'), {'status': "1.1"}) self.assertEqual(response.status_code, 302) self.client.logout() self.client.login(username='******', password='******') response = self.client.post(reverse('comment', args=[1]), {'comment': "5.1.(1.2)"}) self.assertEqual(response.status_code, 302) self.client.logout() self.client.login(username='******', password='******') notification = Notification.objects.get(id=1) message = notification.display() self.assertEqual(message, "User def commented on your post" ) self.client.logout()
def user_addasfriend_view(request, id): current_user = request.user request_to_user = User.objects.get(id=id) Friend.add_friend(current_user, request_to_user) friend = Friend.objects.get(request_from=current_user, request_to=request_to_user, accepted=False) Notification.notify(friend, request_to_user) return redirect(reverse('user', args=[id]))
def user_deletefriend_view(request, id): request_from = User.objects.get(id=id) request_to = request.user try: instance = Friend.objects.get(request_from=request_from, request_to=request_to, accepted=False) friend_type = ContentType.objects.get_for_model(Friend) notification = Notification.objects.get( notification_for=request.user, seen=False, content_type=friend_type, object_id=instance.id ) if notification: notification.seen = True notification.save() except Friend.DoesNotExist: pass Friend.delete_friend(request_from, request_to) return redirect(reverse("home"))
def get_context_data(self, **kwargs): context = super(HomeView, self).get_context_data(**kwargs) context["post_form"] = PostForm() context["comment_form"] = CommentForm() friends = Friend.friends(self.request.user) post_list = [] for friend in friends: post_list.extend(Post.objects.filter(posted_by=friend)) paginator = Paginator(post_list, 5) page = self.request.GET.get("page") try: posts = paginator.page(page) except PageNotAnInteger: posts = paginator.page(1) except EmptyPage: posts = paginator.page(paginator.num_pages) context["posts"] = posts return context
def get_context_data(self, **kwargs): context = super(HomeView, self).get_context_data(**kwargs) context["post_form"] = PostForm() context["comment_form"] = CommentForm() friends = Friend.friends(self.request.user) post_list = [] for friend in friends: post_list.extend(Post.objects.filter(posted_by=friend)) paginator = Paginator(post_list, 5) page = self.request.GET.get('page') try: posts = paginator.page(page) except PageNotAnInteger: posts = paginator.page(1) except EmptyPage: posts = paginator.page(paginator.num_pages) context["posts"] = posts return context
def are_connected(request_from, request_to): friend, friend_request, follow = Friend.are_connected( request_from, request_to) return {'friend': friend, 'friend_request': friend_request, 'request_from': request_from, 'request_to': request_to, 'follow': follow}
def user_deletefriend_view(request, id): request_from = User.objects.get(id=id) request_to = request.user try: instance = Friend.objects.get(request_from=request_from, request_to=request_to, accepted=False) friend_type = ContentType.objects.get_for_model(Friend) notification = Notification.objects.get(notification_for=request.user, seen=False, content_type=friend_type, object_id=instance.id) if notification: notification.seen = True notification.save() except Friend.DoesNotExist: pass Friend.delete_friend(request_from, request_to) return redirect(reverse('home'))
def are_connected(request_from, request_to): friend, friend_request, follow = Friend.are_connected( request_from, request_to) return { 'friend': friend, 'friend_request': friend_request, 'request_from': request_from, 'request_to': request_to, 'follow': follow }