Added admin actions for loading test data

Author: Ken Lippold

hydroserver/urls.py

@@ -2,9 +2,11 @@
 from django.conf.urls.static import static
 from django.contrib import admin
 from django.urls import path, include
-from hydroserver.views import index
+from hydroserver.views import index, load_test_data, load_default_data
 
 urlpatterns = [
+    path("admin/actions/load-test-data/", load_test_data, name="load_test_data"),
+    path("admin/actions/load-default-data/", load_default_data, name="load_default_data"),
     path("admin/", admin.site.urls),
     path("accounts/", include("allauth.urls")),
     path("api/auth/", include("iam.urls")),

hydroserver/views.py

@@ -1,13 +1,42 @@
-from django.shortcuts import render
+from django.shortcuts import render, redirect
 from django.views.decorators.csrf import csrf_exempt
+from django.core.management import call_command
+from django.contrib import messages
+from django.contrib.admin.views.decorators import staff_member_required
 from django.conf import settings
 
 
 @csrf_exempt
 def index(request):
 
     context = {
-        'proxy_base_url': settings.PROXY_BASE_URL
+        "proxy_base_url": settings.PROXY_BASE_URL
     }
 
-    return render(request, 'index.html', context)
+    return render(request, "index.html", context)
+
+
+@staff_member_required
+def load_test_data(request):
+    if request.method == "POST":
+        if settings.DEBUG is False:
+            messages.warning(request, "Test data is only available on development deployments.")
+        else:
+            try:
+                call_command("load_iam_test_data")
+                call_command("load_sta_test_data")
+                messages.success(request, "Test data loaded successfully.")
+            except Exception as e:
+                messages.error(request, f"Error loading test data: {e}")
+    return redirect('admin:index')
+
+
+@staff_member_required
+def load_default_data(request):
+    if request.method == "POST":
+        try:
+            # call_command("load_default_data")
+            messages.success(request, "Default data loaded successfully.")
+        except Exception as e:
+            messages.error(request, f"Error loading default data: {e}")
+    return redirect('admin:index')

templates/admin/index.html

@@ -0,0 +1,68 @@
+{% extends "admin/base_site.html" %}
+{% load i18n static %}
+
+{% block extrastyle %}{{ block.super }}<link rel="stylesheet" href="{% static "admin/css/dashboard.css" %}">{% endblock %}
+
+{% block coltype %}colMS{% endblock %}
+
+{% block bodyclass %}{{ block.super }} dashboard{% endblock %}
+
+{% block nav-breadcrumbs %}{% endblock %}
+
+{% block nav-sidebar %}{% endblock %}
+
+{% block content %}
+<div id="content-main">
+  {% include "admin/app_list.html" with app_list=app_list show_changelinks=True %}
+</div>
+{% endblock %}
+
+{% block sidebar %}
+<div id="content-related">
+    <div class="module" id="admin-actions-module">
+        <h2>{% translate 'Admin actions' %}</h2>
+        <div class="module-content">
+            <form method="post" action="{% url 'load_test_data' %}">
+                {% csrf_token %}
+                <button type="submit" class="button" style="padding: 10px 20px; margin-left: 15px;">
+                    {% trans "Load Test Data" %}
+                </button>
+            </form>
+            <form method="post" action="{% url 'load_default_data' %}">
+                {% csrf_token %}
+                <button type="submit" class="button" style="padding: 10px 20px; margin-left: 15px;">
+                    {% trans "Load Default Data" %}
+                </button>
+            </form>
+        </div>
+    </div>
+    <div class="module" id="recent-actions-module">
+        <h2>{% translate 'Recent actions' %}</h2>
+        <h3>{% translate 'My actions' %}</h3>
+            {% load log %}
+            {% get_admin_log 10 as admin_log for_user user %}
+            {% if not admin_log %}
+            <p>{% translate 'None available' %}</p>
+            {% else %}
+            <ul class="actionlist">
+            {% for entry in admin_log %}
+            <li class="{% if entry.is_addition %}addlink{% endif %}{% if entry.is_change %}changelink{% endif %}{% if entry.is_deletion %}deletelink{% endif %}">
+                <span class="visually-hidden">{% if entry.is_addition %}{% translate 'Added:' %}{% elif entry.is_change %}{% translate 'Changed:' %}{% elif entry.is_deletion %}{% translate 'Deleted:' %}{% endif %}</span>
+                {% if entry.is_deletion or not entry.get_admin_url %}
+                    {{ entry.object_repr }}
+                {% else %}
+                    <a href="{{ entry.get_admin_url }}">{{ entry.object_repr }}</a>
+                {% endif %}
+                <br>
+                {% if entry.content_type %}
+                    <span class="mini quiet">{% filter capfirst %}{{ entry.content_type.name }}{% endfilter %}</span>
+                {% else %}
+                    <span class="mini quiet">{% translate 'Unknown content' %}</span>
+                {% endif %}
+            </li>
+            {% endfor %}
+            </ul>
+            {% endif %}
+    </div>
+</div>
+{% endblock %}