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 = {