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 @@
- Workbench Settings
+ Workbench
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 = {