"""group views""" from django.apps import apps from django.contrib.auth.decorators import login_required from django.db import IntegrityError from django.core.paginator import Paginator from django.http import HttpResponseNotFound, HttpResponseBadRequest from django.shortcuts import get_object_or_404, redirect from django.template.response import TemplateResponse from django.utils.decorators import method_decorator from django.views import View from django.views.decorators.http import require_POST from django.contrib.postgres.search import TrigramSimilarity from django.db.models.functions import Greatest from bookwyrm import forms, models from bookwyrm.suggested_users import suggested_users from .helpers import privacy_filter from .helpers import get_user_from_username from bookwyrm.settings import DOMAIN class BookwyrmGroup(View): """group page""" def get(self, request, group_id): """display a group""" group = get_object_or_404(models.BookwyrmGroup, id=group_id) lists = models.List.objects.filter(group=group).order_by("-updated_date") lists = privacy_filter(request.user, lists) # don't show groups to users who shouldn't see them if not group.visible_to_user(request.user): return HttpResponseNotFound() data = { "group": group, "lists": lists, "group_form": forms.GroupForm(instance=group), "path": "/group", } return TemplateResponse(request, "groups/group.html", data) @method_decorator(login_required, name="dispatch") def post(self, request, group_id): """edit a group""" user_group = get_object_or_404(models.BookwyrmGroup, id=group_id) form = forms.GroupForm(request.POST, instance=user_group) if not form.is_valid(): return redirect("group", user_group.id) user_group = form.save() return redirect("group", user_group.id) @method_decorator(login_required, name="dispatch") class UserGroups(View): """a user's groups page""" def get(self, request, username): """display a group""" user = get_user_from_username(request.user, username) memberships = models.BookwyrmGroupMember.objects.filter(user=user).all() paginated = Paginator(memberships, 12) data = { "memberships": paginated.get_page(request.GET.get("page")), "is_self": request.user.id == user.id, "user": user, "group_form": forms.GroupForm(), "path": user.local_path + "/group", } return TemplateResponse(request, "user/groups.html", data) @method_decorator(login_required, name="dispatch") # pylint: disable=unused-argument def post(self, request, username): """create a user group""" form = forms.GroupForm(request.POST) if not form.is_valid(): return redirect(request.user.local_path + "groups") group = form.save() # add the creator as a group member models.BookwyrmGroupMember.objects.create(group=group, user=request.user) return redirect("group", group.id) @method_decorator(login_required, name="dispatch") class FindUsers(View): """find friends to add to your group""" """this is mostly borrowed from the Get Started friend finder""" def get(self, request, group_id): """basic profile info""" query = request.GET.get("query") group = models.BookwyrmGroup.objects.get(id=group_id) user_results = ( models.User.viewer_aware_objects(request.user) .exclude(memberships__in=group.memberships.all()) # don't suggest users who are already members .annotate( similarity=Greatest( TrigramSimilarity("username", query), TrigramSimilarity("localname", query), ) ) .filter( similarity__gt=0.5, local=True ) .order_by("-similarity")[:5] ) data = {"no_results": not user_results} if user_results.count() < 5: user_results = list(user_results) + suggested_users.get_suggestions( request.user ) group = get_object_or_404(models.BookwyrmGroup, id=group_id) data = { "suggested_users": user_results, "group": group, "query": query, "requestor_is_manager": request.user == group.user } return TemplateResponse(request, "groups/find_users.html", data) @require_POST @login_required def invite_member(request): """invite a member to the group""" group = get_object_or_404(models.BookwyrmGroup, id=request.POST.get("group")) if not group: return HttpResponseBadRequest() user = get_user_from_username(request.user, request.POST["user"]) if not user: return HttpResponseBadRequest() if not group.user == request.user: return HttpResponseBadRequest() try: models.GroupMemberInvitation.objects.create( user=user, group=group ) except IntegrityError: pass return redirect(user.local_path) @require_POST @login_required def remove_member(request): """invite a member to the group""" group = get_object_or_404(models.BookwyrmGroup, id=request.POST.get("group")) if not group: return HttpResponseBadRequest() user = get_user_from_username(request.user, request.POST["user"]) if not user: return HttpResponseBadRequest() is_member = models.BookwyrmGroupMember.objects.filter(group=group,user=user).exists() is_invited = models.GroupMemberInvitation.objects.filter(group=group,user=user).exists() if is_invited: try: invitation = models.GroupMemberInvitation.objects.get( user=user, group=group ) invitation.reject() except IntegrityError: pass if is_member: try: membership = models.BookwyrmGroupMember.objects.get(group=group,user=user) membership.delete() except IntegrityError: pass # TODO: should send notification to all members including the now ex-member that they have been removed. return redirect(user.local_path) @require_POST @login_required def accept_membership(request): """accept an invitation to join a group""" group = models.BookwyrmGroup.objects.get(id=request.POST["group"]) if not group: return HttpResponseBadRequest() invite = models.GroupMemberInvitation.objects.get(group=group,user=request.user) if not invite: return HttpResponseBadRequest() try: invite.accept() except IntegrityError: pass return redirect(group.local_path) @require_POST @login_required def reject_membership(request): """reject an invitation to join a group""" group = models.BookwyrmGroup.objects.get(id=request.POST["group"]) if not group: return HttpResponseBadRequest() invite = models.GroupMemberInvitation.objects.get(group=group,user=request.user) if not invite: return HttpResponseBadRequest() try: invite.reject() except IntegrityError: pass return redirect(request.user.local_path)