From 75660644eb4715018de40fb6ff5233b641fbb53a Mon Sep 17 00:00:00 2001
From: FreneticScribbler <aj@aronajones.com>
Date: Mon, 17 Feb 2020 21:25:25 +0000
Subject: [PATCH] FEAT: More handy buttons

---
 .../commands/generateSampleAssetsData.py      |  4 +-
 assets/models.py                              |  2 +-
 assets/templates/asset_audit.html             | 17 +++++--
 .../templates/partials/cable_form_body.html   | 50 +++++++++++++++++++
 4 files changed, 65 insertions(+), 8 deletions(-)
 create mode 100644 assets/templates/partials/cable_form_body.html

diff --git a/assets/management/commands/generateSampleAssetsData.py b/assets/management/commands/generateSampleAssetsData.py
index a4e0dcdb..3ed45eed 100644
--- a/assets/management/commands/generateSampleAssetsData.py
+++ b/assets/management/commands/generateSampleAssetsData.py
@@ -29,8 +29,8 @@ class Command(BaseCommand):
     def create_profile(self):
         name = "Fred Johnson"
         models.Profile.objects.create(username=name.replace(" ", ""), first_name=name.split(" ")[0], last_name=name.split(" ")[-1],
-                                                       email=name.replace(" ", "") + "@example.com",
-                                                       initials="".join([j[0].upper() for j in name.split()]))
+                                      email=name.replace(" ", "") + "@example.com",
+                                      initials="".join([j[0].upper() for j in name.split()]))
 
     def create_categories(self):
         categories = ['Case', 'Video', 'General', 'Sound', 'Lighting', 'Rigging']
diff --git a/assets/models.py b/assets/models.py
index c7376a4f..bda0a30f 100644
--- a/assets/models.py
+++ b/assets/models.py
@@ -104,7 +104,7 @@ class Asset(models.Model, RevisionMixin):
     length = models.DecimalField(decimal_places=1, max_digits=10,
                                  blank=True, null=True, help_text='m')
     csa = models.DecimalField(decimal_places=2, max_digits=10,
-                              blank=True, null=True, help_text='mm^2')
+                              blank=True, null=True, help_text='mm²')
     circuits = models.IntegerField(blank=True, null=True)
     cores = models.IntegerField(blank=True, null=True)
 
diff --git a/assets/templates/asset_audit.html b/assets/templates/asset_audit.html
index f918ae84..e31ce451 100644
--- a/assets/templates/asset_audit.html
+++ b/assets/templates/asset_audit.html
@@ -14,6 +14,9 @@
   function setLength(length) {
       $('#id_length').val(length);
   }
+  function setCSA(CSA) {
+      $('#id_csa').val(CSA);
+  }
   function checkIfCableHidden() {
     if (document.getElementById("id_is_cable").checked) {
       document.getElementById("cable-table").hidden = false;
@@ -106,19 +109,23 @@
         </div>
       </div>
       <div class="col-sm-4">
-        <btn class="btn btn-danger" onclick="setLength('5');">5m</btn>
-        <btn class="btn btn-success" onclick="setLength('10');">10m</btn>
-        <btn class="btn btn-info" onclick="setLength('20');">20m</btn>
-    </div>
+        <btn class="btn btn-danger" onclick="setLength('5');">5{{ form.length.help_text }}</btn>
+        <btn class="btn btn-success" onclick="setLength('10');">10{{ form.length.help_text }}</btn>
+        <btn class="btn btn-info" onclick="setLength('20');">20{{ form.length.help_text }}</btn>
+      </div>
     </div>
     <div class="form-group">
       <label for="{{ form.csa.id_for_label }}" class="col-sm-2 control-label">Cross Sectional Area</label>
-      <div class="col-sm-10">
+      <div class="col-sm-6">
         <div class="input-group">
           {% render_field form.csa|add_class:'form-control' value=object.csa %}
           <span class="input-group-addon">{{ form.csa.help_text }}</span>
         </div>
       </div>
+      <div class="col-sm-4">
+        <btn class="btn btn-default" onclick="setCSA('1.5');">1.5{{ form.csa.help_text }}</btn>
+        <btn class="btn btn-default" onclick="setCSA('2.5');">2.5{{ form.csa.help_text }}</btn>
+      </div>
     </div>
     <div class="form-group">
       <label for="{{ form.circuits.id_for_label }}" class="col-sm-2 control-label">Circuits</label>
diff --git a/assets/templates/partials/cable_form_body.html b/assets/templates/partials/cable_form_body.html
new file mode 100644
index 00000000..ffe5ed77
--- /dev/null
+++ b/assets/templates/partials/cable_form_body.html
@@ -0,0 +1,50 @@
+{% load widget_tweaks %}
+<div id="cable-table">
+    <div class="form-group">
+      <label for="{{ form.plug.id_for_label }}" class="col-sm-2 control-label">Plug</label>
+      <div class="col-sm-10">
+        {% render_field form.plug|add_class:'form-control'%}
+      </div>
+    </div>
+    <div class="form-group">
+      <label for="{{ form.socket.id_for_label }}" class="col-sm-2 control-label">Socket</label>
+      <div class="col-sm-10">
+        {% render_field form.socket|add_class:'form-control'%}
+      </div>
+    </div>
+    <div class="form-group">
+      <label for="{{ form.length.id_for_label }}" class="col-sm-2 control-label">Length</label>
+      <div class="col-sm-6">
+        <div class="input-group">
+          {% render_field form.length|add_class:'form-control' %}
+          <span class="input-group-addon">{{ form.length.help_text }}</span>
+        </div>
+      </div>
+      <div class="col-sm-4">
+        <btn class="btn btn-danger btn-xs" onclick="setLength('5');">5m</btn>
+        <btn class="btn btn-success btn-xs" onclick="setLength('10');">10m</btn>
+        <btn class="btn btn-info btn-xs" onclick="setLength('20');">20m</btn>
+      </div>
+    </div>
+    <div class="form-group">
+      <label for="{{ form.csa.id_for_label }}" class="col-sm-2 control-label">Cross Sectional Area</label>
+      <div class="col-sm-10">
+        <div class="input-group">
+          {% render_field form.csa|add_class:'form-control' value=object.csa %}
+          <span class="input-group-addon">{{ form.csa.help_text }}</span>
+        </div>
+      </div>
+    </div>
+    <div class="form-group">
+      <label for="{{ form.circuits.id_for_label }}" class="col-sm-2 control-label">Circuits</label>
+      <div class="col-sm-10">
+        {% render_field form.circuits|add_class:'form-control' value=object.circuits %}
+      </div>
+    </div>
+    <div class="form-group">
+      <label for="{{ form.cores.id_for_label }}" class="col-sm-2 control-label">Cores</label>
+      <div class="col-sm-10">
+        {% render_field form.cores|add_class:'form-control' value=object.cores %}
+      </div>
+    </div>
+</div>