## # Copyright (c) 2012-2017 Apple Inc. All rights reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. # ## """ loadsim Web Admin UI. """ __all__ = [ "LoadSimAdminResource", ] import cStringIO as StringIO import uuid from time import clock from twisted.web import resource class LoadSimAdminResource (resource.Resource): """ Web administration HTTP resource. """ isLeaf = True HEAD = """\

Load Simulator Web Admin

""" BODY = """\ """ BODY_RESULTS = """\

%s

Generated in %.1f milliseconds

""" BODY_RESULTS_STOPPED = "

LoadSim Stopped - Final Results

" + BODY_RESULTS def __init__(self, loadsim): self.loadsim = loadsim self.token = str(uuid.uuid4()) def render_GET(self, request): return self._renderReport() def render_POST(self, request): html = self.HEAD + self.BODY if 'token' not in request.args or request.args['token'][0] != self.token: return html % (self.token,) if 'stop' in request.args: self.loadsim.stop() return self._renderReport(True) elif 'results' in request.args: return self._renderReport() return html % (self.token,) def _renderReport(self, stopped=False): report = StringIO.StringIO() before = clock() self.loadsim.reporter.generateReport(report) after = clock() ms = (after - before) * 1000 if stopped: html = self.HEAD + self.BODY_RESULTS_STOPPED return html % (None, report.getvalue(), ms) else: html = self.HEAD + self.BODY_RESULTS return html % (self.token, report.getvalue(), ms)