(add) database view template

Author: crosspolar

tapir/shifts/migrations/0057_auto_20240709_1056.py tapir/shifts/migrations/0057_auto_20240709_1941.py

@@ -1,5 +1,6 @@
-# Generated by Django 3.2.25 on 2024-07-09 08:56
+# Generated by Django 3.2.25 on 2024-07-09 17:41
 
+import datetime
 from django.conf import settings
 from django.db import migrations, models
 import django.db.models.deletion
@@ -14,17 +15,17 @@ class Migration(migrations.Migration):
 
     operations = [
         migrations.CreateModel(
-            name='ShiftObserve',
+            name='ShiftWatch',
             fields=[
                 ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
-                ('notification_timedelta', models.DurationField()),
+                ('notification_timedelta', models.DurationField(default=datetime.timedelta(days=2))),
                 ('created_at', models.DateTimeField(auto_now_add=True)),
-                ('shift_to_be_notified_about', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='shifts.shift')),
-                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='shift_notification_entry', to=settings.AUTH_USER_MODEL)),
+                ('shift_watched', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='shifts.shift')),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='user_watching_shift', to=settings.AUTH_USER_MODEL)),
             ],
         ),
         migrations.AddConstraint(
-            model_name='shiftobserve',
-            constraint=models.UniqueConstraint(fields=('user', 'shift_to_be_notified_about'), name='user_shift_constraint'),
+            model_name='shiftwatch',
+            constraint=models.UniqueConstraint(fields=('user', 'shift_watched'), name='shift_watch_constraint'),
         ),
     ]

tapir/shifts/models.py

@@ -1117,18 +1117,18 @@ class SolidarityShift(models.Model):
     date_used = models.DateField(null=True)
 
 
-class ShiftObserve(models.Model):
+class ShiftWatch(models.Model):
     class Meta:
         constraints = [
             models.UniqueConstraint(
-                fields=["user", "shift_to_be_notified_about"],
-                name="user_shift_constraint",
+                fields=["user", "shift_watched"],
+                name="shift_watch_constraint",
             )
         ]
 
     user = models.ForeignKey(
-        TapirUser, related_name="shift_notification_entry", on_delete=models.CASCADE
+        TapirUser, related_name="user_watching_shift", on_delete=models.CASCADE
     )
-    shift_to_be_notified_about = models.ForeignKey(Shift, on_delete=models.CASCADE)
-    notification_timedelta = models.DurationField()
+    shift_watched = models.ForeignKey(Shift, on_delete=models.CASCADE)
+    notification_timedelta = models.DurationField(default=datetime.timedelta(days=2))
     created_at = models.DateTimeField(auto_now_add=True)

tapir/shifts/templates/shifts/shift_detail.html

@@ -29,14 +29,14 @@ <h5 class="d-flex justify-content-between align-items-center">
                 {% if perms.shifts.manage %}
                     <span>
 
-                        {% if True %}
+                        {% if request.user.user_watching_shift|user_watching_shift:shift.id %}
                             <a class="{% tapir_button_link %}"
-                                href="{% url 'shifts:shift_detail_printable' shift.pk %}">
+                                href="{% url 'shifts:unwatch_shift' shift.pk %}">
                                  <span class="material-icons">visibility_off</span>{% translate 'Unwatch' %}
                             </a>
                         {% else %}
-                            <a class="{% tapir_button_link %}"
-                                href="{% url 'shifts:shift_detail_printable' shift.pk %}">
+                                                                                                    <a class="{% tapir_button_link %}"
+                                href="{% url 'shifts:watch_shift' shift.pk %}">
                                  <span class="material-icons">visibility</span>{% translate 'Watch' %}
                             </a>
                         {% endif %}

tapir/shifts/urls.py

@@ -252,4 +252,6 @@
         views.ShiftStatusEvolutionJsonView.as_view(),
         name="shift_status_evolution_json",
     ),
+    path("watch_shift/<int:shift_id>/", views.watch_shift, name="watch_shift"),
+    path("unwatch_shift/<int:shift_id>/", views.unwatch_shift, name="unwatch_shift"),
 ]

tapir/shifts/views/views.py

@@ -47,6 +47,7 @@
     SHIFT_ATTENDANCE_STATES,
     ShiftTemplate,
     SolidarityShift,
+    ShiftWatch,
 )
 from tapir.shifts.models import (
     ShiftSlot,
@@ -495,3 +496,21 @@ def get_context_data(self, **kwargs):
             },
         }
         return context_data
+
+
+def watch_shift(request, shift_id):
+    shift_to_watch = get_object_or_404(Shift, id=shift_id)
+    if not request.user.user_watching_shift.filter(shift_watched_id=shift_id).exists():
+        ShiftWatch.objects.create(user=request.user, shift_watched=shift_to_watch)
+    return redirect("shifts:shift_detail", pk=shift_id)
+
+
+def unwatch_shift(request, shift_id):
+    shift_to_unwatch = get_object_or_404(Shift, id=shift_id)
+    request.user.user_watching_shift.filter(shift_watched_id=shift_id).delete()
+    return redirect("shifts:shift_detail", pk=shift_id)
+
+
+@register.filter
+def user_watching_shift(user, shift_id) -> bool:
+    return user.filter(shift_watched_id=shift_id).exists()