From 87df8d460c4f0c153581600da754da3893c108a6 Mon Sep 17 00:00:00 2001
From: Adam Goldsmith <adam@adamgoldsmith.name>
Date: Tue, 15 Mar 2022 21:59:48 -0400
Subject: [PATCH] [rentals] Allow reserving lockers for internal makerspace use

---
 rentals/forms.py                              |  1 +
 .../0003_lockerinfo_reserved_and_more.py      | 30 +++++++++++++++++++
 rentals/models.py                             | 21 +++++++++++--
 rentals/templates/rentals/lockers.dj.html     | 21 +++++++++----
 4 files changed, 65 insertions(+), 8 deletions(-)
 create mode 100644 rentals/migrations/0003_lockerinfo_reserved_and_more.py

diff --git a/rentals/forms.py b/rentals/forms.py
index 911fb06..8497dbd 100644
--- a/rentals/forms.py
+++ b/rentals/forms.py
@@ -10,4 +10,5 @@ class LockerInfoForm(forms.ModelForm):
             "blind_code",
             "bitting_code",
             "renter",
+            "reserved",
         ]
diff --git a/rentals/migrations/0003_lockerinfo_reserved_and_more.py b/rentals/migrations/0003_lockerinfo_reserved_and_more.py
new file mode 100644
index 0000000..4467cee
--- /dev/null
+++ b/rentals/migrations/0003_lockerinfo_reserved_and_more.py
@@ -0,0 +1,30 @@
+# Generated by Django 4.0.2 on 2022-03-16 01:54
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("rentals", "0002_lockerinfo_notes"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="lockerinfo",
+            name="reserved",
+            field=models.BooleanField(
+                default=False,
+                help_text="Locker is reserved for MakerSpace use, and cannot be rented.",
+            ),
+        ),
+        migrations.AddConstraint(
+            model_name="lockerinfo",
+            constraint=models.CheckConstraint(
+                check=models.Q(
+                    ("reserved", False), ("renter__isnull", True), _connector="OR"
+                ),
+                name="locker_not_reserved_and_rented",
+            ),
+        ),
+    ]
diff --git a/rentals/models.py b/rentals/models.py
index 55b1a4d..09c0cf3 100644
--- a/rentals/models.py
+++ b/rentals/models.py
@@ -1,7 +1,8 @@
 from django.contrib import admin
 from django.core import validators
+from django.core.exceptions import ValidationError
 from django.db import models, transaction
-from django.db.models import F
+from django.db.models import F, Q
 from django.db.models.functions import Chr, Ord, Concat
 
 
@@ -85,15 +86,31 @@ class LockerInfo(models.Model):
     renter = models.ForeignKey(
         Member, on_delete=models.CASCADE, null=True, blank=True, db_constraint=False
     )
+    reserved = models.BooleanField(
+        default=False,
+        help_text="Locker is reserved for MakerSpace use, and cannot be rented.",
+    )
     notes = models.TextField(blank=True)
 
+    def clean(self):
+        if self.reserved and self.renter is not None:
+            raise ValidationError("Locker cannot both be reserved and rented!")
+
     class Meta:
         constraints = [
             models.UniqueConstraint(
                 fields=["locker_unit", "column", "row"], name="unique_locker_info"
-            )
+            ),
+            models.CheckConstraint(
+                check=Q(reserved=False) | Q(renter__isnull=True),
+                name="locker_not_reserved_and_rented",
+            ),
         ]
 
+    @property
+    def available(self) -> bool:
+        return self.renter is None and not self.reserved
+
     @property
     def letter(self) -> str:
         return self.locker_unit.letter_for_column(self.column)
diff --git a/rentals/templates/rentals/lockers.dj.html b/rentals/templates/rentals/lockers.dj.html
index b869700..efd50af 100644
--- a/rentals/templates/rentals/lockers.dj.html
+++ b/rentals/templates/rentals/lockers.dj.html
@@ -39,7 +39,7 @@
     justify-content: center;
     color: lightgray;
   }
-  .locker .door:not([data-rented="false"]) {
+  .locker .door[data-available="False"] {
     background-color: hsl(24, 15%, 25%);;
   }
   </style>
@@ -60,7 +60,7 @@
                       data-bs-auto-close="outside"
                       aria-haspopup="true"
                       aria-expanded="false"
-                      data-rented="{{ locker_form.instance.renter|yesno:"true,false" }}">
+                      data-available="{{ locker_form.instance.available }}">
                   <h3 class="locker-address mb-0">{{ locker_form.instance.address }}</h3>
                   <div class="locker-status text-wrap">
                     {% if locker_form.instance.renter is not None %}
@@ -69,6 +69,8 @@
                       {% else %}
                         Occupied
                       {% endif %}
+                    {% elif locker_form.instance.reserved %}
+                      Reserved
                     {% else %}
                       Empty
                     {% endif %}
@@ -83,12 +85,19 @@
                       {% csrf_token %}
                       <fieldset {{ perms.rentals.change_lockerinfo|yesno:",disabled" }}>
                         {% for field in locker_form.visible_fields %}
-                          <div class="form-floating mb-3">
-                            {% render_field field class+="form-control" %}
-                            {{ field.label_tag }}
+                          <div class="mb-3">
+                            {% if field.widget_type == 'checkbox' %}
+                              {% render_field field class+="btn-check" %}
+                              {{ field|add_label_class:"btn btn-outline-primary" }}
+                            {% else %}
+                              <div class="form-floating">
+                                {% render_field field class+="form-control" %}
+                                {{ field.label_tag }}
+                              </div>
+                            {% endif %}
                             {{ field.errors }}
                             {% if field.help_text %}
-                              <p class="form-text text-nowrap">
+                              <p class="form-text">
                                 {{ field.help_text|safe }}
                               </p>
                             {% endif %}