95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
|
import time
|
||
|
|
|
||
|
|
import flask
|
||
|
|
from flask import Blueprint
|
||
|
|
from flask import current_app as app
|
||
|
|
from flask import g, make_response, request
|
||
|
|
from flask_login import login_required
|
||
|
|
|
||
|
|
from ereuse_devicehub import auth
|
||
|
|
from ereuse_devicehub.db import db
|
||
|
|
from ereuse_devicehub.resources.enums import SessionType
|
||
|
|
from ereuse_devicehub.resources.user.models import Session
|
||
|
|
from ereuse_devicehub.views import GenericMixView
|
||
|
|
|
||
|
|
workbench = Blueprint('workbench', __name__, url_prefix='/workbench')
|
||
|
|
|
||
|
|
|
||
|
|
class SettingsView(GenericMixView):
|
||
|
|
decorators = [login_required]
|
||
|
|
template_name = 'workbench/settings.html'
|
||
|
|
page_title = "Workbench Settings"
|
||
|
|
|
||
|
|
def dispatch_request(self):
|
||
|
|
self.get_context()
|
||
|
|
self.context.update(
|
||
|
|
{
|
||
|
|
'page_title': self.page_title,
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
self.opt = request.values.get('opt')
|
||
|
|
if self.opt in ['register', 'soft', 'hard']:
|
||
|
|
return self.download()
|
||
|
|
|
||
|
|
return flask.render_template(self.template_name, **self.context)
|
||
|
|
|
||
|
|
def download(self):
|
||
|
|
self.wbContext = {
|
||
|
|
'token': self.get_token(),
|
||
|
|
'host': app.config['HOST'],
|
||
|
|
'inventory': app.config['SCHEMA'],
|
||
|
|
'benchmark': False,
|
||
|
|
'stress_test': 1,
|
||
|
|
'erase': '',
|
||
|
|
'steps': 0,
|
||
|
|
'leading_zeros': False,
|
||
|
|
}
|
||
|
|
options = {"register": self.register, "soft": self.soft, "hard": self.hard}
|
||
|
|
return options[self.opt]()
|
||
|
|
|
||
|
|
def register(self):
|
||
|
|
data = flask.render_template('workbench/wbSettings.ini', **self.wbContext)
|
||
|
|
return self.response_download(data)
|
||
|
|
|
||
|
|
def soft(self):
|
||
|
|
self.wbContext['erase'] = 'EraseBasic'
|
||
|
|
self.wbContext['steps'] = 1
|
||
|
|
data = flask.render_template('workbench/wbSettings.ini', **self.wbContext)
|
||
|
|
return self.response_download(data)
|
||
|
|
|
||
|
|
def hard(self):
|
||
|
|
self.wbContext['erase'] = 'EraseSectors'
|
||
|
|
self.wbContext['steps'] = 1
|
||
|
|
self.wbContext['leading_zeros'] = True
|
||
|
|
data = flask.render_template('workbench/wbSettings.ini', **self.wbContext)
|
||
|
|
return self.response_download(data)
|
||
|
|
|
||
|
|
def response_download(self, data):
|
||
|
|
bfile = str.encode(data)
|
||
|
|
output = make_response(bfile)
|
||
|
|
output.headers['Content-Disposition'] = 'attachment; filename=settings.ini'
|
||
|
|
output.headers['Content-type'] = 'text/plain'
|
||
|
|
return output
|
||
|
|
|
||
|
|
def get_token(self):
|
||
|
|
if not g.user.sessions:
|
||
|
|
ses = Session(user=g.user)
|
||
|
|
db.session.add(ses)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
tk = ''
|
||
|
|
now = time.time()
|
||
|
|
for s in g.user.sessions:
|
||
|
|
if s.type == SessionType.Internal and (s.expired == 0 or s.expired > now):
|
||
|
|
tk = s.token
|
||
|
|
break
|
||
|
|
|
||
|
|
assert tk != ''
|
||
|
|
|
||
|
|
token = auth.Auth.encode(tk)
|
||
|
|
return token
|
||
|
|
|
||
|
|
|
||
|
|
workbench.add_url_rule('/settings/', view_func=SettingsView.as_view('settings'))
|