Minor style update

This commit is contained in:
Miroslav Stampar 2012-12-21 15:06:03 +01:00
parent 8b3e17ed4d
commit 00e55828e4
2 changed files with 19 additions and 19 deletions

View File

@ -17,11 +17,11 @@ class ProgressBar(object):
def __init__(self, minValue=0, maxValue=10, totalWidth=None): def __init__(self, minValue=0, maxValue=10, totalWidth=None):
self._progBar = "[]" self._progBar = "[]"
self._oldProgBar = "" self._oldProgBar = ""
self.__min = int(minValue) self._min = int(minValue)
self.__max = int(maxValue) self._max = int(maxValue)
self.__span = self.__max - self.__min self._span = self._max - self._min
self.__width = totalWidth if totalWidth else conf.progressWidth self._width = totalWidth if totalWidth else conf.progressWidth
self.__amount = 0 self._amount = 0
self.update() self.update()
def _convertSeconds(self, value): def _convertSeconds(self, value):
@ -36,21 +36,21 @@ class ProgressBar(object):
This method updates the progress bar This method updates the progress bar
""" """
if newAmount < self.__min: if newAmount < self._min:
newAmount = self.__min newAmount = self._min
elif newAmount > self.__max: elif newAmount > self._max:
newAmount = self.__max newAmount = self._max
self.__amount = newAmount self._amount = newAmount
# Figure out the new percent done, round to an integer # Figure out the new percent done, round to an integer
diffFromMin = float(self.__amount - self.__min) diffFromMin = float(self._amount - self._min)
percentDone = (diffFromMin / float(self.__span)) * 100.0 percentDone = (diffFromMin / float(self._span)) * 100.0
percentDone = round(percentDone) percentDone = round(percentDone)
percentDone = int(percentDone) percentDone = int(percentDone)
# Figure out how many hash bars the percentage should be # Figure out how many hash bars the percentage should be
allFull = self.__width - 2 allFull = self._width - 2
numHashes = (percentDone / 100.0) * allFull numHashes = (percentDone / 100.0) * allFull
numHashes = int(round(numHashes)) numHashes = int(round(numHashes))
@ -75,11 +75,11 @@ class ProgressBar(object):
if self._progBar != self._oldProgBar: if self._progBar != self._oldProgBar:
self._oldProgBar = self._progBar self._oldProgBar = self._progBar
if eta and self.__amount < self.__max: if eta and self._amount < self._max:
dataToStdout("\r%s %d/%d ETA %s" % (self._progBar, self.__amount, self.__max, self._convertSeconds(int(eta)))) dataToStdout("\r%s %d/%d ETA %s" % (self._progBar, self._amount, self._max, self._convertSeconds(int(eta))))
else: else:
blank = " " * (80 - len("\r%s %d/%d" % (self._progBar, self.__amount, self.__max))) blank = " " * (80 - len("\r%s %d/%d" % (self._progBar, self._amount, self._max)))
dataToStdout("\r%s %d/%d%s" % (self._progBar, self.__amount, self.__max, blank)) dataToStdout("\r%s %d/%d%s" % (self._progBar, self._amount, self._max, blank))
def __str__(self): def __str__(self):
""" """

View File

@ -161,9 +161,9 @@ class Xp_cmdshell:
# Obfuscate the command to execute, also useful to bypass filters # Obfuscate the command to execute, also useful to bypass filters
# on single-quotes # on single-quotes
self._randStr = randomStr(lowercase=True) self._randStr = randomStr(lowercase=True)
self.__cmd = "0x%s" % hexencode(cmd) self._cmd = "0x%s" % hexencode(cmd)
self._forgedCmd = "DECLARE @%s VARCHAR(8000);" % self._randStr self._forgedCmd = "DECLARE @%s VARCHAR(8000);" % self._randStr
self._forgedCmd += "SET @%s=%s;" % (self._randStr, self.__cmd) self._forgedCmd += "SET @%s=%s;" % (self._randStr, self._cmd)
# Insert the command standard output into a support table, # Insert the command standard output into a support table,
# 'sqlmapoutput', except when DBMS credentials are provided because # 'sqlmapoutput', except when DBMS credentials are provided because