nautobot 2.3.16__py3-none-any.whl → 2.4.0b1__py3-none-any.whl

nautobot/users/tests/test_views.py

@@ -1,3 +1,4 @@
+import json
 from unittest import mock
 
 from django.conf import settings
@@ -5,9 +6,13 @@ from django.contrib.auth import get_user_model
 from django.contrib.sessions.middleware import SessionMiddleware
 from django.test import override_settings, RequestFactory
 from django.urls import reverse
+from django.utils import timezone
 from social_django.utils import load_backend, load_strategy
 
 from nautobot.core.testing import TestCase, utils
+from nautobot.core.testing.context import load_event_broker_override_settings
+from nautobot.core.testing.utils import post_data
+from nautobot.users.utils import serialize_user_without_config_and_views
 
 User = get_user_model()
 
@@ -38,6 +43,37 @@ class PasswordUITest(TestCase):
         )
         self.assertBodyContains(post_response, "The two password fields")
 
+    @load_event_broker_override_settings(
+        EVENT_BROKERS={
+            "SyslogEventBroker": {
+                "CLASS": "nautobot.core.events.SyslogEventBroker",
+                "TOPICS": {
+                    "INCLUDE": ["*"],
+                },
+            }
+        }
+    )
+    def test_change_password(self):
+        self.user.set_password("foo")
+        self.user.save()
+        self.client.force_login(self.user)
+        with self.assertLogs("nautobot.events") as cm:
+            self.client.post(
+                reverse("user:change_password"),
+                data={
+                    "old_password": "foo",
+                    "new_password1": "bar",
+                    "new_password2": "bar",
+                },
+            )
+        payload = serialize_user_without_config_and_views(self.user)
+        self.assertEqual(
+            cm.output,
+            [f"INFO:nautobot.events.nautobot.users.user.change_password:{json.dumps(payload, indent=4)}"],
+        )
+        self.user.refresh_from_db()
+        self.assertTrue(self.user.check_password("bar"))
+
     @override_settings(
         AUTHENTICATION_BACKENDS=[
             "social_core.backends.google.GoogleOAuth2",
@@ -133,3 +169,24 @@ class AdvancedProfileSettingsViewTest(TestCase):
 
         # Check if the session has the correct value
         self.assertFalse(self.client.session.get("silk_record_requests"))
+
+
+class PreferenceTestCase(TestCase):
+    def test_timezone_change(self):
+        self.user.is_superuser = True
+        self.user.save()
+        self.client.force_login(self.user)
+
+        timezone_name = timezone.get_current_timezone_name()
+        new_timezone_name = "US/Eastern"
+        form_data = {"timezone": new_timezone_name, "_update_preference_form": [""]}
+        url = reverse("user:preferences")
+        request = {
+            "path": url,
+            "data": post_data(form_data),
+        }
+        response = self.client.post(**request)
+        response = self.client.get(url)
+        self.assertEqual(timezone.get_current_timezone_name(), new_timezone_name)
+        self.assertNotEqual(timezone_name, new_timezone_name)
+        self.assertHttpStatus(response, 200)

nautobot/users/utils.py

@@ -0,0 +1,8 @@
+from nautobot.core.models.utils import serialize_object_v2
+
+
+def serialize_user_without_config_and_views(user):
+    serialized_data = serialize_object_v2(user)
+    for key in ["config_data", "default_saved_views"]:
+        serialized_data.pop(key, None)
+    return {"data": serialized_data}

nautobot/users/views.py

@@ -13,13 +13,16 @@ from django.urls import reverse
 from django.utils.decorators import method_decorator
 from django.utils.encoding import iri_to_uri
 from django.utils.http import url_has_allowed_host_and_scheme
+from django.utils.timezone import get_default_timezone_name
 from django.views.decorators.debug import sensitive_post_parameters
 from django.views.generic import View
 
+from nautobot.core.events import publish_event
 from nautobot.core.forms import ConfirmationForm
 from nautobot.core.views.generic import GenericView
+from nautobot.users.utils import serialize_user_without_config_and_views
 
-from .forms import AdvancedProfileSettingsForm, LoginForm, PasswordChangeForm, TokenForm
+from .forms import AdvancedProfileSettingsForm, LoginForm, PasswordChangeForm, PreferenceProfileSettingsForm, TokenForm
 from .models import Token
 
 #
@@ -33,7 +36,6 @@ class LoginView(View):
     """
 
     template_name = "login.html"
-    use_new_ui = True
 
     @method_decorator(sensitive_post_parameters("password"))
     def dispatch(self, *args, **kwargs):
@@ -62,8 +64,11 @@ class LoginView(View):
             logger.debug("Login form validation was successful")
 
             # Authenticate user
+            user = form.get_user()
             auth_login(request, form.get_user())
             messages.info(request, f"Logged in as {request.user}.")
+            payload = serialize_user_without_config_and_views(user)
+            publish_event(topic="nautobot.users.user.login", payload=payload)
 
             return self.redirect_to_next(request, logger)
 
@@ -92,6 +97,10 @@ class LoginView(View):
         return HttpResponseRedirect(iri_to_uri(redirect_to))
 
 
+# TODO: The LogoutView should inherit from `LoginRequiredMixin` or `GenericView`
+#   to prevent unauthenticated users from accessing the logout page.
+#   However, using `LoginRequiredMixin` or `GenericView` as-is currently redirects
+#   users to the login page with `?next=/logout/`, which is not desired.
 class LogoutView(View):
     """
     Deauthenticate a web user.
@@ -99,6 +108,9 @@ class LogoutView(View):
 
     def get(self, request):
         # Log out the user
+        if request.user.is_authenticated:
+            payload = serialize_user_without_config_and_views(request.user)
+            publish_event(topic="nautobot.users.user.logout", payload=payload)
         auth_logout(request)
         messages.info(request, "You have logged out.")
 
@@ -136,28 +148,51 @@ class UserConfigView(GenericView):
     template_name = "users/preferences.html"
 
     def get(self, request):
+        tzname = request.user.get_config("timezone", get_default_timezone_name())
+        form = PreferenceProfileSettingsForm(initial={"timezone": tzname})
         return render(
             request,
             self.template_name,
             {
                 "preferences": request.user.all_config(),
+                "form": form,
                 "active_tab": "preferences",
                 "is_django_auth_user": is_django_auth_user(request),
             },
         )
 
     def post(self, request):
-        user = request.user
-        data = user.all_config()
+        is_preference_update_post = "_update_preference_form" in request.POST
+        if is_preference_update_post:
+            form = PreferenceProfileSettingsForm(request.POST)
+            if form.is_valid():
+                if timezone := form.cleaned_data["timezone"]:
+                    request.user.set_config("timezone", str(timezone), commit=True)
+                return redirect("user:preferences")
+
+            return render(
+                request,
+                self.template_name,
+                {
+                    "preferences": request.user.all_config(),
+                    "form": form,
+                    "active_tab": "preferences",
+                    "is_django_auth_user": is_django_auth_user(request),
+                },
+            )
+
+        else:
+            user = request.user
+            data = user.all_config()
 
-        # Delete selected preferences
-        for key in request.POST.getlist("pk"):
-            if key in data:
-                user.clear_config(key)
-        user.save()
-        messages.success(request, "Your preferences have been updated.")
+            # Delete selected preferences
+            for key in request.POST.getlist("pk"):
+                if key in data:
+                    user.clear_config(key)
+            user.save()
+            messages.success(request, "Your preferences have been updated.")
 
-        return redirect("user:preferences")
+            return redirect("user:preferences")
 
 
 class ChangePasswordView(GenericView):
@@ -200,6 +235,8 @@ class ChangePasswordView(GenericView):
             form.save()
             update_session_auth_hash(request, form.user)
             messages.success(request, "Your password has been changed successfully.")
+            payload = serialize_user_without_config_and_views(request.user)
+            publish_event(topic="nautobot.users.user.change_password", payload=payload)
             return redirect("user:profile")
 
         return render(

nautobot/virtualization/api/serializers.py

@@ -87,10 +87,10 @@ class VMInterfaceSerializer(
         model = VMInterface
         fields = "__all__"
 
-    def validate(self, attrs):
+    def validate(self, data):
         # Validate many-to-many VLAN assignments
-        virtual_machine = self.instance.virtual_machine if self.instance else attrs.get("virtual_machine")
-        for vlan in attrs.get("tagged_vlans", []):
+        virtual_machine = self.instance.virtual_machine if self.instance else data.get("virtual_machine")
+        for vlan in data.get("tagged_vlans", []):
             if vlan.location not in [virtual_machine.location, None]:
                 raise serializers.ValidationError(
                     {
@@ -99,4 +99,4 @@ class VMInterfaceSerializer(
                     }
                 )
 
-        return super().validate(attrs)
+        return super().validate(data)

nautobot/virtualization/filters.py

@@ -224,26 +224,12 @@ class VirtualMachineFilterSet(
         to_field_name="version",
         label="Software version (version or ID)",
     )
-    ip_addresses = MultiValueCharFilter(
-        method="filter_ip_addresses",
-        label="IP addresses (address or ID)",
-        distinct=True,
-    )
-    has_ip_addresses = RelatedMembershipBooleanFilter(field_name="interfaces__ip_addresses", label="Has IP addresses")
-
-    def filter_ip_addresses(self, queryset, name, value):
-        pk_values = set(item for item in value if is_uuid(item))
-        addresses = set(item for item in value if item not in pk_values)
-
-        ip_queryset = IPAddress.objects.filter_address_or_pk_in(addresses, pk_values)
-        return queryset.filter(interfaces__ip_addresses__in=ip_queryset).distinct()
 
     class Meta:
         model = VirtualMachine
         fields = [
             "id",
             "name",
-            "cluster",
             "vcpus",
             "memory",
             "disk",
@@ -354,11 +340,7 @@ class VMInterfaceFilterSet(
         queryset=VLAN.objects.all(),
         label="Untagged VLAN (VID or ID)",
     )
-    ip_addresses = MultiValueCharFilter(
-        method="filter_ip_addresses",
-        label="IP addresses (address or ID)",
-        distinct=True,
-    )
+    ip_addresses = MultiValueCharFilter(method="filter_ip_addresses", label="IP addresses (address or ID)")
     has_ip_addresses = RelatedMembershipBooleanFilter(field_name="ip_addresses", label="Has IP addresses")
 
     def filter_ip_addresses(self, queryset, name, value):
@@ -366,7 +348,7 @@ class VMInterfaceFilterSet(
         addresses = set(item for item in value if item not in pk_values)
 
         ip_queryset = IPAddress.objects.filter_address_or_pk_in(addresses, pk_values)
-        return queryset.filter(ip_addresses__in=ip_queryset).distinct()
+        return queryset.filter(ip_addresses__in=ip_queryset)
 
     class Meta:
         model = VMInterface

nautobot/virtualization/navigation.py

@@ -1,7 +1,4 @@
 from nautobot.core.apps import (
-    NavContext,
-    NavGrouping,
-    NavItem,
     NavMenuAddButton,
     NavMenuGroup,
     NavMenuItem,
@@ -101,48 +98,3 @@ menu_items = (
         ),
     ),
 )
-
-
-navigation = (
-    NavContext(
-        name="Inventory",
-        groups=(
-            NavGrouping(
-                name="Virtualization",
-                weight=600,
-                items=(
-                    NavItem(
-                        name="Virtual Machines",
-                        weight=100,
-                        link="virtualization:virtualmachine_list",
-                        permissions=["virtualization.view_virtualmachine"],
-                    ),
-                    NavItem(
-                        name="VM Interfaces",
-                        weight=200,
-                        link="virtualization:vminterface_list",
-                        permissions=["virtualization.view_vminterface"],
-                    ),
-                    NavItem(
-                        name="Clusters",
-                        weight=300,
-                        link="virtualization:cluster_list",
-                        permissions=["virtualization.view_cluster"],
-                    ),
-                    NavItem(
-                        name="Cluster Types",
-                        weight=400,
-                        link="virtualization:clustertype_list",
-                        permissions=["virtualization.view_clustertype"],
-                    ),
-                    NavItem(
-                        name="Cluster Groups",
-                        weight=500,
-                        link="virtualization:clustergroup_list",
-                        permissions=["virtualization.view_clustergroup"],
-                    ),
-                ),
-            ),
-        ),
-    ),
-)

nautobot/virtualization/templates/virtualization/clustertype.html

@@ -1,40 +1 @@
 {% extends 'generic/object_detail.html' %}
-{% load helpers %}
-
-{% block content_left_page %}
-        <div class="panel panel-default">
-            <div class="panel-heading">
-                <strong>Cluster Type</strong>
-            </div>
-            <table class="table table-hover panel-body attr-table">
-                <tr>
-                    <td>Description</td>
-                    <td>{{ object.description|placeholder }}</td>
-                </tr>
-                <tr>
-                    <td>Clusters</td>
-                    <td>
-                        <a href="{% url 'virtualization:cluster_list' %}?cluster_type={{ object.name }}">{{ cluster_table.rows|length }}</a>
-                    </td>
-                </tr>
-            </table>
-        </div>
-{% endblock content_left_page %}
-
-{% block content_right_page %}
-        <div class="panel panel-default">
-            <div class="panel-heading">
-                <strong>Clusters</strong>
-            </div>
-            {% include 'inc/table.html' with table=cluster_table %}
-            {% if perms.virtualization.add_cluster %}
-                <div class="panel-footer text-right noprint">
-                    <a href="{% url 'virtualization:cluster_add' %}?cluster_type={{ object.pk }}" class="btn btn-xs btn-primary">
-                        <span class="mdi mdi-plus-thick" aria-hidden="true"></span> Add cluster
-                    </a>
-                </div>
-            {% endif %}
-        </div>
-        {% include 'inc/paginator.html' with paginator=cluster_table.paginator page=cluster_table.page %}
-        <div class="row"></div>
-{% endblock content_right_page %}