Handle read only setting in users form

babybuddy/forms.py

@@ -1,12 +1,20 @@
 # -*- coding: utf-8 -*-
 from django import forms
-from django.contrib.auth.forms import PasswordChangeForm, UserCreationForm
 from django.contrib.auth import get_user_model
+from django.contrib.auth.forms import PasswordChangeForm, UserCreationForm
+from django.contrib.auth.models import Group
+from django.utils.translation import gettext_lazy as _
 
 from .models import Settings
 
 
-class UserAddForm(UserCreationForm):
+class BabyBuddyUserForm(forms.ModelForm):
+    is_readonly = forms.BooleanField(
+        required=False,
+        label=_("Read only"),
+        help_text=_("Restricts user to viewing data only."),
+    )
+
     class Meta:
         model = get_user_model()
         fields = [
@@ -15,29 +23,41 @@ class UserAddForm(UserCreationForm):
             "last_name",
             "email",
             "is_staff",
+            "is_readonly",
             "is_active",
         ]
 
+    def __init__(self, *args, **kwargs):
+        user = kwargs["instance"]
+        if user:
+            kwargs["initial"].update(
+                {"is_readonly": user.groups.filter(name="read_only").exists()}
+            )
+        super(BabyBuddyUserForm, self).__init__(*args, **kwargs)
+
     def save(self, commit=True):
-        user = super(UserAddForm, self).save(commit=False)
+        user = super(BabyBuddyUserForm, self).save(commit=False)
-        # All Baby Buddy users are superusers.
+        is_readonly = self.cleaned_data["is_readonly"]
+        if is_readonly:
+            user.is_superuser = False
+        else:
             user.is_superuser = True
         if commit:
             user.save()
+        readonly_group = Group.objects.get(name="read_only")
+        if is_readonly:
+            user.groups.add(readonly_group.id)
+        else:
+            user.groups.remove(readonly_group.id)
         return user
 
 
-class UserUpdateForm(forms.ModelForm):
+class UserAddForm(BabyBuddyUserForm, UserCreationForm):
-    class Meta:
+    pass
-        model = get_user_model()
+
-        fields = [
+
-            "username",
+class UserUpdateForm(BabyBuddyUserForm):
-            "first_name",
+    pass
-            "last_name",
-            "email",
-            "is_staff",
-            "is_active",
-        ]
 
 
 class UserForm(forms.ModelForm):

babybuddy/templates/babybuddy/user_list.html

@@ -1,5 +1,5 @@
 {% extends 'babybuddy/page.html' %}
-{% load bootstrap i18n widget_tweaks %}
+{% load babybuddy_tags bootstrap i18n widget_tweaks %}
 
 {% block title %}{% trans "Users" %}{% endblock %}
 
@@ -18,6 +18,7 @@
                     <th>{% trans "First Name" %}</th>
                     <th>{% trans "Last Name" %}</th>
                     <th>{% trans "Email" %}</th>
+                    <th>{% trans "Read only" %}</th>
                     <th>{% trans "Staff" %}</th>
                     <th>{% trans "Active" %}</th>
                     <th class="text-center">{% trans "Actions" %}</th>
@@ -30,6 +31,8 @@
                         <td>{{ object.first_name }}</td>
                         <td>{{ object.last_name }}</td>
                         <td>{{ object.email }}</td>
+                        {% user_is_read_only object as is_readonly %}
+                        <td>{{ is_readonly|bool_icon }}</td>
                         <td>{{ object.is_staff|bool_icon }}</td>
                         <td>{{ object.is_active|bool_icon }}</td>
                         <td class="text-center">

babybuddy/templatetags/babybuddy_tags.py

@@ -63,3 +63,8 @@ def make_absolute_url(context, url):
     request = context["request"]
     abs_url = request.build_absolute_uri(url)
     return abs_url
+
+
+@register.simple_tag()
+def user_is_read_only(user):
+    return user.groups.filter(name="read_only").exists()