__copyright__ = \

__version__ = "$Revision$"

from common import webutils;

from testmanager.core.testset import TestSetData;

from testmanager.webui.wuicontentbase import WuiContentBase, WuiTmLink;

from testmanager.webui.wuimain import WuiMain;

class WuiLogViewer(WuiContentBase):

"""Log viewer."""

def __init__(self, oTestSet, oLogFile, cbChunk, iChunk, oDisp = None, fnDPrint = None):

WuiContentBase.__init__(self, oDisp = oDisp, fnDPrint = fnDPrint);

self._oTestSet = oTestSet;

self._oLogFile = oLogFile;

self._cbChunk = cbChunk;

self._iChunk = iChunk;

def _generateNavigation(self, cbFile):

"""Generate the HTML for the log navigation."""

dParams = {

WuiMain.ksParamAction: WuiMain.ksActionViewLog,

WuiMain.ksParamLogSetId: self._oTestSet.idTestSet,

WuiMain.ksParamLogFileId: self._oLogFile.idTestResultFile,

WuiMain.ksParamLogChunkSize: self._cbChunk,

WuiMain.ksParamLogChunkNo: self._iChunk,

};

#

# The page walker.

#

dParams2 = dict(dParams);

del dParams2[WuiMain.ksParamLogChunkNo];

sHrefFmt = '<a href="?%s&%s=%%s" title="%%s">%%s</a>' \

% (webutils.encodeUrlParams(dParams2).replace('%', '%%'), WuiMain.ksParamLogChunkNo,);

sHtmlWalker = self.genericPageWalker(self._iChunk, (cbFile + self._cbChunk - 1) / self._cbChunk,

sHrefFmt, 11, 0, 'chunk');

#

# The chunk size selector.

#

dParams2 = dict(dParams);

del dParams2[WuiMain.ksParamLogChunkSize];

sHtmlSize = '<form name="ChunkSizeForm" method="GET">\n' \

' Max <select name="%s" onchange="window.location=\'?%s&%s=\' + ' \

'this.options[this.selectedIndex].value;" title="Max items per page">\n' \

% ( WuiMain.ksParamLogChunkSize, webutils.encodeUrlParams(dParams2), WuiMain.ksParamLogChunkSize,);

for cbChunk in [ 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072, 262144, 524288, 1048576, 2097152,

4194304, 8388608, 16777216 ]:

sHtmlSize += ' <option value="%d" %s>%d bytes</option>\n' \

% (cbChunk, 'selected="selected"' if cbChunk == self._cbChunk else '', cbChunk);

sHtmlSize += ' </select> per page\n' \

'</form>\n'

#

# Download links.

#

oRawLink = WuiTmLink('View Raw', '',

{ WuiMain.ksParamAction: WuiMain.ksActionGetFile,

WuiMain.ksParamGetFileSetId: self._oTestSet.idTestSet,

WuiMain.ksParamGetFileId: self._oLogFile.idTestResultFile,

WuiMain.ksParamGetFileDownloadIt: False,

},

sTitle = '%u MiB' % ((cbFile + 1048576 - 1) / 1048576,) );

oDownloadLink = WuiTmLink('Download Log', '',

{ WuiMain.ksParamAction: WuiMain.ksActionGetFile,

WuiMain.ksParamGetFileSetId: self._oTestSet.idTestSet,

WuiMain.ksParamGetFileId: self._oLogFile.idTestResultFile,

WuiMain.ksParamGetFileDownloadIt: True,

},

sTitle = '%u MiB' % ((cbFile + 1048576 - 1) / 1048576,) );

oTestSetLink = WuiTmLink('Test Set', '',

{ WuiMain.ksParamAction: WuiMain.ksActionTestResultDetails,

TestSetData.ksParam_idTestSet: self._oTestSet.idTestSet,

});

#

# Combine the elements and return.

#

return '<div class="tmlogviewernavi">\n' \

' <table width=100%>\n' \

' <tr>\n' \

' <td width=20%>\n' \

' ' + oTestSetLink.toHtml() + '\n' \

' ' + oRawLink.toHtml() + '\n' \

' ' + oDownloadLink.toHtml() + '\n' \

' </td>\n' \

' <td width=60% align=center>' + sHtmlWalker + '</td>' \

' <td width=20% align=right>' + sHtmlSize + '</td>\n' \

' </tr>\n' \

' </table>\n' \

'</div>\n';

def _displayLog(self, oFile, offFile, cbFile):

"""Displays the current section of the log file."""

offEnd = offFile + self._cbChunk;

if offEnd > cbFile:

offEnd = cbFile;

#

# Here is an annoying thing, we cannot seek in zip file members. So,

# since we have to read from the start, we can just as well count line

# numbers while we're at it.

#

offCur = 0;

iLine = 0;

while True:

sLine = oFile.readline().decode('utf-8', 'replace');

offLine = offCur;

iLine += 1;

offCur += len(sLine);

if offCur >= offFile or len(sLine) == 0:

break;

#

# Got to where we wanted, format the chunk.

#

sHtml = '\n<div class="tmlog">\n<pre>\n';

while True:

sHtml += '<a id="L%d" href="#L%d">%05d</a><a id="O%d"> </a>%s\n' \

% (iLine, iLine, iLine, offLine, webutils.escapeElem(sLine.rstrip()));

if offCur >= offEnd:

break;

sLine = oFile.readline().decode('utf-8', 'replace');

offLine = offCur;

iLine += 1;

offCur += len(sLine);

if len(sLine) == 0:

break;

sHtml += '<pre/></div>\n';

return sHtml;

def show(self):

"""Shows the log."""

if self._oLogFile.sDescription not in [ '', None ]:

sTitle = '%s - %s' % (self._oLogFile.sFile, self._oLogFile.sDescription);

else:

sTitle = '%s' % (self._oLogFile.sFile,);

#

# Open the log file. No universal line endings here.

#

(oFile, oSizeOrError, _) = self._oTestSet.openFile(self._oLogFile.sFile, 'rb');

if oFile is None:

return (sTitle, '<p>%s</p>\n' % (webutils.escapeElem(oSizeOrError),),);

cbFile = oSizeOrError;

#

# Generate the page.

#

# Start with a focus hack.

sHtml = '<div id="tmlogoutdiv" tabindex="0">\n' \

'<script lang="text/javascript">\n' \

'document.getElementById(\'tmlogoutdiv\').focus();\n' \

'</script>\n';

sNaviHtml = self._generateNavigation(cbFile);

sHtml += sNaviHtml;

offFile = self._iChunk * self._cbChunk;

if offFile < cbFile:

sHtml += self._displayLog(oFile, offFile, cbFile);

sHtml += sNaviHtml;

else:

sHtml += '<p>End Of File</p>';

return (sTitle, sHtml);