moviewyrm/bookwyrm/views/follow.py

''' views for actions you can take in the application '''
from django.db import  transaction
from django.contrib.auth.decorators import login_required
from django.http import HttpResponseBadRequest
from django.shortcuts import redirect
from django.views.decorators.http import require_POST

from bookwyrm import models
from bookwyrm.broadcast import broadcast
from .helpers import get_user_from_username

@login_required
@require_POST
def follow(request):
    ''' follow another user, here or abroad '''
    username = request.POST['user']
    try:
        to_follow = get_user_from_username(username)
    except models.User.DoesNotExist:
        return HttpResponseBadRequest()

    relationship, _ = models.UserFollowRequest.objects.get_or_create(
        user_subject=request.user,
        user_object=to_follow,
    )
    activity = relationship.to_activity()
    broadcast(
        request.user, activity, privacy='direct', direct_recipients=[to_follow])
    return redirect(to_follow.local_path)


@login_required
@require_POST
def unfollow(request):
    ''' unfollow a user '''
    username = request.POST['user']
    try:
        to_unfollow = get_user_from_username(username)
    except models.User.DoesNotExist:
        return HttpResponseBadRequest()

    relationship = models.UserFollows.objects.get(
        user_subject=request.user,
        user_object=to_unfollow
    )
    activity = relationship.to_undo_activity(request.user)
    broadcast(
        request.user, activity,
        privacy='direct', direct_recipients=[to_unfollow])

    to_unfollow.followers.remove(request.user)
    return redirect(to_unfollow.local_path)


@login_required
@require_POST
def accept_follow_request(request):
    ''' a user accepts a follow request '''
    username = request.POST['user']
    try:
        requester = get_user_from_username(username)
    except models.User.DoesNotExist:
        return HttpResponseBadRequest()

    try:
        follow_request = models.UserFollowRequest.objects.get(
            user_subject=requester,
            user_object=request.user
        )
    except models.UserFollowRequest.DoesNotExist:
        # Request already dealt with.
        return redirect(request.user.local_path)
    handle_accept(follow_request)

    return redirect(request.user.local_path)


def handle_accept(follow_request):
    ''' send an acceptance message to a follow request '''
    user = follow_request.user_subject
    to_follow = follow_request.user_object
    with transaction.atomic():
        relationship = models.UserFollows.from_request(follow_request)
        follow_request.delete()
        relationship.save()

    activity = relationship.to_accept_activity()
    broadcast(to_follow, activity, privacy='direct', direct_recipients=[user])


@login_required
@require_POST
def delete_follow_request(request):
    ''' a user rejects a follow request '''
    username = request.POST['user']
    try:
        requester = get_user_from_username(username)
    except models.User.DoesNotExist:
        return HttpResponseBadRequest()

    try:
        follow_request = models.UserFollowRequest.objects.get(
            user_subject=requester,
            user_object=request.user
        )
    except models.UserFollowRequest.DoesNotExist:
        return HttpResponseBadRequest()

    activity = follow_request.to_reject_activity()
    follow_request.delete()
    broadcast(
        request.user, activity, privacy='direct', direct_recipients=[requester])
    return redirect('/user/%s' % request.user.localname)