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
}
