From 7cb5c7656230efa90a714b489d540672701d1425 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@ribaguifi.com>
Date: Thu, 23 Jun 2022 18:08:56 +0200
Subject: [PATCH] add download iso

---
 .../templates/ereuse_devicehub/base_site.html    |  2 +-
 ereuse_devicehub/workbench/views.py              | 16 ++++++++++++++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/ereuse_devicehub/templates/ereuse_devicehub/base_site.html b/ereuse_devicehub/templates/ereuse_devicehub/base_site.html
index 658c2345..2eeb716a 100644
--- a/ereuse_devicehub/templates/ereuse_devicehub/base_site.html
+++ b/ereuse_devicehub/templates/ereuse_devicehub/base_site.html
@@ -95,7 +95,7 @@
           <li>
             <a class="dropdown-item d-flex align-items-center" href="{{ url_for('workbench.settings') }}">
               <i class="bi bi-tools"></i>
-              <span>Workbench Settings</span>
+              <span>Workbench</span>
             </a>
           </li>
           <li>
diff --git a/ereuse_devicehub/workbench/views.py b/ereuse_devicehub/workbench/views.py
index f70885ce..85bb22f6 100644
--- a/ereuse_devicehub/workbench/views.py
+++ b/ereuse_devicehub/workbench/views.py
@@ -1,3 +1,4 @@
+import os
 import time
 
 import flask
@@ -22,6 +23,7 @@ class SettingsView(GenericMixin):
 
     def dispatch_request(self):
         self.get_context()
+        self.get_iso()
         self.context.update(
             {
                 'page_title': self.page_title,
@@ -34,6 +36,20 @@ class SettingsView(GenericMixin):
 
         return flask.render_template(self.template_name, **self.context)
 
+    def get_iso(self):
+        files = [
+            f
+            for f in os.listdir('ereuse_devicehub/static/iso/')
+            if f[-3:].lower() == 'iso'
+        ]
+
+        self.context['iso'] = ''
+        self.context['iso_sha'] = ''
+
+        if files:
+            self.context['iso'] = files[0]
+            self.context['iso_sha'] = 'aaa'
+
     def download(self):
         url = "https://{}/api/inventory/".format(app.config['HOST'])
         self.wbContext = {