From a2423ec4f8188c3f6c0533a2b4ff87e853c79bda Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A9my=20Taymans?= <remytms@tsmail.eu>
Date: Tue, 10 Jan 2023 19:55:53 +0100
Subject: [PATCH] [IMP] estimate-workload: add total column in report

---
 odoo_openupgrade_wizard/templates/analysis.html.j2 | 8 ++++++--
 odoo_openupgrade_wizard/tools/tools_odoo_module.py | 7 +++++++
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/odoo_openupgrade_wizard/templates/analysis.html.j2 b/odoo_openupgrade_wizard/templates/analysis.html.j2
index 6359a2f..8fae5ef 100644
--- a/odoo_openupgrade_wizard/templates/analysis.html.j2
+++ b/odoo_openupgrade_wizard/templates/analysis.html.j2
@@ -65,6 +65,7 @@
       <thead>
         <tr>
           <th>&nbsp;</th>
+          <th>Estimated time (min)</th>
 {%- for odoo_version in ctx.obj['config']['odoo_versions'] -%}
           <th>{{ odoo_version }}</th>
 {% endfor %}
@@ -87,7 +88,7 @@
     and odoo_module.module_type != 'odoo') %}
     {% set ns.current_module_type = odoo_module.module_type %}
         <tr>
-          <th colspan="{{1 + ctx.obj['config']['odoo_versions']|length}}">
+          <th colspan="{{2 + ctx.obj['config']['odoo_versions']|length}}">
             {{ ns.current_module_type}}
           </th>
         <tr>
@@ -102,7 +103,7 @@
 
     {% if ns.current_repository %}
         <tr>
-          <th colspan="{{1 + ctx.obj['config']['odoo_versions']|length}}">
+          <th colspan="{{2 + ctx.obj['config']['odoo_versions']|length}}">
             {{ ns.current_repository}}
           </th>
         <tr>
@@ -130,6 +131,9 @@
   {% endif %}
 
           </td>
+
+          <td>{{odoo_module.workload}}</td>
+
   {% for version in odoo_module.analyse.all_versions %}
     {% set module_version = odoo_module.get_module_version(version) %}
     {% if module_version %}
diff --git a/odoo_openupgrade_wizard/tools/tools_odoo_module.py b/odoo_openupgrade_wizard/tools/tools_odoo_module.py
index 15bd050..dd70372 100644
--- a/odoo_openupgrade_wizard/tools/tools_odoo_module.py
+++ b/odoo_openupgrade_wizard/tools/tools_odoo_module.py
@@ -328,6 +328,13 @@ class OdooModule(object):
         else:
             self.module_type = "custom"
 
+    @property
+    def workload(self):
+        return sum(
+            round(module_version.workload)
+            for _, module_version in self.module_versions.items()
+        )
+
     def get_module_version(self, current_version):
         res = self.module_versions.get(current_version, False)
         return res