From 574d1d28750ef025d398c99c07946e34e470e843 Mon Sep 17 00:00:00 2001
From: "Edgar P. Burkhart" <git@edgarpierre.fr>
Date: Wed, 3 May 2023 12:18:20 +0200
Subject: [PATCH] Add total line to category plot Closes #9

---
 .../templates/category/category_plot.html     | 25 +++++++++++++++++++
 nummi/category/templatetags/category.py       | 14 +++++------
 nummi/main/static/main/css/plot.css           |  4 +++
 3 files changed, 36 insertions(+), 7 deletions(-)

diff --git a/nummi/category/templates/category/category_plot.html b/nummi/category/templates/category/category_plot.html
index 1313da6..33d67fd 100644
--- a/nummi/category/templates/category/category_plot.html
+++ b/nummi/category/templates/category/category_plot.html
@@ -58,5 +58,30 @@
                 {% endfor %}
             {% endspaceless %}
         </tbody>
+        {% if categories %}
+            <tfoot>
+                <tr>
+                    <th scope="row" colspan="2" class="l">{% translate "Total" %}</th>
+                    <td class="value">{{ total_m|pmrvalue }}</td>
+                    <td class="bar m">
+                        <div style="width: {% widthratio total_m max -100 %}%"></div>
+                        {% if total < 0 %}
+                            <div class="tot" style="width:{% widthratio total max -100 %}%">
+                                <span>{{ total|pmrvalue }}</span>
+                            </div>
+                        {% endif %}
+                    </td>
+                    <td class="bar p">
+                        <div style="width: {% widthratio total_p max 100 %}%"></div>
+                        {% if total > 0 %}
+                            <div class="tot" style="width:{% widthratio total max 100 %}%">
+                                <span>{{ total|pmrvalue }}</span>
+                            </div>
+                        {% endif %}
+                    </td>
+                    <td class="value">{{ total_p|pmrvalue }}</td>
+                </tr>
+            </tfoot>
+        {% endif %}
     </table>
 </div>
diff --git a/nummi/category/templatetags/category.py b/nummi/category/templatetags/category.py
index c43256b..5709af0 100644
--- a/nummi/category/templatetags/category.py
+++ b/nummi/category/templatetags/category.py
@@ -1,5 +1,6 @@
 from django import template
 from django.db import models
+from django.db.models.functions import Greatest
 
 register = template.Library()
 
@@ -18,10 +19,9 @@ def category_plot(transactions):
     )
     return {
         "categories": categories,
-        "max": max(
-            categories.aggregate(
-                max=models.Max("sum_p", default=0),
-                min=models.Min("sum_m", default=0),
-            ).values(),
-        ),
-    }
+    } | categories.aggregate(
+        max=Greatest(models.Sum("sum_m"), models.Sum("sum_p")),
+        total_m=models.Sum("sum_m"),
+        total_p=models.Sum("sum_p"),
+        total=models.Sum("value"),
+    )
diff --git a/nummi/main/static/main/css/plot.css b/nummi/main/static/main/css/plot.css
index d38d575..15fc7ab 100644
--- a/nummi/main/static/main/css/plot.css
+++ b/nummi/main/static/main/css/plot.css
@@ -70,3 +70,7 @@ table.full-width col.bar {
     overflow: hidden;
   }
 }
+
+.plot tfoot {
+  background: var(--bg-01);
+}