"""group views""" 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 # TODO: from .helpers import get_user_from_username class Group(View): """group page""" def get(self, request, group_id): """display a group""" group = models.Group.objects.get(id=group_id) # groups = privacy_filter( # request.user, groups, privacy_levels=["public", "followers"] # ) data = { "group": group, "list_form": forms.GroupForm(), "path": "/group", } return TemplateResponse(request, "groups/group.html", data) @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) groups = models.Group.objects.filter(members=user).order_by("-updated_date") paginated = Paginator(groups, 12) data = { "user": user, "is_self": request.user.id == user.id, "groups": paginated.get_page(request.GET.get("page")), "group_form": forms.GroupForm(), "path": user.local_path + "/group", } return TemplateResponse(request, "user/groups.html", data) @method_decorator(login_required, name="dispatch") class FindUsers(View): """find friends to add to your group""" """this is mostly taken from the Get Started friend finder""" def get(self, request, group_id): """basic profile info""" query = request.GET.get("query") user_results = ( models.User.viewer_aware_objects(request.user) .annotate( similarity=Greatest( TrigramSimilarity("username", query), TrigramSimilarity("localname", query), ) ) .filter( similarity__gt=0.5, ) .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 ) data["suggested_users"] = user_results data["group"] = get_object_or_404(models.Group, id=group_id) data["query"] = query return TemplateResponse(request, "groups/find_users.html", data) @login_required @require_POST def create_group(request): """user groups""" form = forms.GroupForm(request.POST) if not form.is_valid(): print("invalid!") return redirect(request.headers.get("Referer", "/")) group = form.save() # add the creator as a group member models.GroupMember.objects.create(group=group, user=request.user) return redirect(group.local_path) @require_POST @login_required def add_member(request): """add a member to the group""" # TODO: if groups become AP values we need something like get_group_from_group_fullname # group = get_object_or_404(models.Group, id=request.POST.get("group")) group = models.Group.objects.get(id=request.POST["group"]) if not group: return HttpResponseBadRequest() user = get_user_from_username(request.user, request.POST["user"]) if not user: return HttpResponseBadRequest() if not group.manager == request.user: return HttpResponseBadRequest() try: models.GroupMember.objects.create( group=group, user=user ) except IntegrityError: print("no integrity") pass # TODO: how do we return and update AJAX data? return redirect(user.local_path) @require_POST @login_required def remove_member(request): """remove a member from the group""" # TODO: if groups become AP values we need something like get_group_from_group_fullname # group = get_object_or_404(models.Group, id=request.POST.get("group")) group = models.Group.objects.get(id=request.POST["group"]) if not group: return HttpResponseBadRequest() user = get_user_from_username(request.user, request.POST["user"]) if not user: return HttpResponseBadRequest() if not group.manager == request.user: return HttpResponseBadRequest() try: membership = models.GroupMember.objects.get(group=group,user=user) membership.delete() except IntegrityError: print("no integrity") pass # TODO: how do we return and update AJAX data? return redirect(user.local_path)