1 files changed, 84 insertions, 0 deletions
diff --git a/users/views/admin/users.py b/users/views/admin/users.py
new file mode 100644
index 0000000..fab4616
--- /dev/null
+++ b/users/views/admin/users.py
@@ -0,0 +1,84 @@
+from django import forms
+from django.db import models
+from django.shortcuts import get_object_or_404, redirect
+from django.utils.decorators import method_decorator
+from django.views.generic import FormView, ListView
+
+from users.decorators import admin_required
+from users.models import User
+
+
+@method_decorator(admin_required, name="dispatch")
+class UsersRoot(ListView):
+
+    template_name = "admin/users.html"
+    paginate_by = 50
+
+    def get(self, request, *args, **kwargs):
+        self.query = request.GET.get("query")
+        self.extra_context = {
+            "section": "users",
+            "query": self.query or "",
+        }
+        return super().get(request, *args, **kwargs)
+
+    def get_queryset(self):
+        users = User.objects.annotate(
+            num_identities=models.Count("identities")
+        ).order_by("created")
+        if self.query:
+            users = users.filter(email__icontains=self.query)
+        return users
+
+
+@method_decorator(admin_required, name="dispatch")
+class UserEdit(FormView):
+
+    template_name = "admin/user_edit.html"
+    extra_context = {
+        "section": "users",
+    }
+
+    class form_class(forms.Form):
+        status = forms.ChoiceField(
+            choices=[
+                ("normal", "Normal User"),
+                ("moderator", "Moderator"),
+                ("admin", "Admin"),
+                ("banned", "Banned"),
+            ]
+        )
+
+    def dispatch(self, request, id, *args, **kwargs):
+        self.user = get_object_or_404(User, id=id)
+        return super().dispatch(request, *args, **kwargs)
+
+    def get_initial(self):
+        status = "normal"
+        if self.user.moderator:
+            status = "moderator"
+        if self.user.admin:
+            status = "admin"
+        if self.user.banned:
+            status = "banned"
+        return {
+            "email": self.user.email,
+            "status": status,
+        }
+
+    def form_valid(self, form):
+        # Don't let them change themselves
+        if self.user == self.request.user:
+            return redirect(".")
+        status = form.cleaned_data["status"]
+        self.user.banned = status == "banned"
+        self.user.moderator = status == "moderator"
+        self.user.admin = status == "admin"
+        self.user.save()
+        return redirect(self.user.urls.admin)
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context["editing_user"] = self.user
+        context["same_user"] = self.user == self.request.user
+        return context