sqlmap/lib/core/progress.py

90 lines
2.9 KiB
Python
Raw Normal View History

2008-10-15 19:38:22 +04:00
#!/usr/bin/env python
"""
2012-07-12 21:38:03 +04:00
Copyright (c) 2006-2012 sqlmap developers (http://sqlmap.org/)
2010-10-15 03:18:29 +04:00
See the file 'doc/COPYING' for copying permission
2008-10-15 19:38:22 +04:00
"""
2010-06-02 16:45:40 +04:00
from lib.core.common import getUnicode
2008-10-15 19:38:22 +04:00
from lib.core.common import dataToStdout
2010-03-12 15:46:26 +03:00
from lib.core.data import conf
2008-10-15 19:38:22 +04:00
class ProgressBar(object):
2008-10-15 19:38:22 +04:00
"""
This class defines methods to update and draw a progress bar
"""
2010-03-12 15:46:26 +03:00
def __init__(self, minValue=0, maxValue=10, totalWidth=None):
self._progBar = "[]"
self._oldProgBar = ""
2008-11-20 14:13:04 +03:00
self.__min = int(minValue)
self.__max = int(maxValue)
self.__span = self.__max - self.__min
2010-03-12 15:46:26 +03:00
self.__width = totalWidth if totalWidth else conf.progressWidth
2008-10-15 19:38:22 +04:00
self.__amount = 0
self.update()
def _convertSeconds(self, value):
2008-10-15 19:38:22 +04:00
seconds = value
minutes = seconds / 60
seconds = seconds - (minutes * 60)
return "%.2d:%.2d" % (minutes, seconds)
def update(self, newAmount=0):
"""
This method updates the progress bar
"""
if newAmount < self.__min:
newAmount = self.__min
elif newAmount > self.__max:
newAmount = self.__max
self.__amount = newAmount
# Figure out the new percent done, round to an integer
diffFromMin = float(self.__amount - self.__min)
percentDone = (diffFromMin / float(self.__span)) * 100.0
percentDone = round(percentDone)
percentDone = int(percentDone)
# Figure out how many hash bars the percentage should be
allFull = self.__width - 2
numHashes = (percentDone / 100.0) * allFull
numHashes = int(round(numHashes))
# Build a progress bar with an arrow of equal signs
if numHashes == 0:
self._progBar = "[>%s]" % (" " * (allFull - 1))
2008-10-15 19:38:22 +04:00
elif numHashes == allFull:
self._progBar = "[%s]" % ("=" * allFull)
2008-10-15 19:38:22 +04:00
else:
self._progBar = "[%s>%s]" % ("=" * (numHashes - 1),
2008-10-15 19:38:22 +04:00
" " * (allFull - numHashes))
# Add the percentage at the beginning of the progress bar
2010-06-02 16:45:40 +04:00
percentString = getUnicode(percentDone) + "%"
self._progBar = "%s %s" % (percentString, self._progBar)
2008-10-15 19:38:22 +04:00
def draw(self, eta=0):
"""
This method draws the progress bar if it has changed
"""
if self._progBar != self._oldProgBar:
self._oldProgBar = self._progBar
2008-10-15 19:38:22 +04:00
if eta and self.__amount < self.__max:
dataToStdout("\r%s %d/%d ETA %s" % (self._progBar, self.__amount, self.__max, self._convertSeconds(int(eta))))
2008-10-15 19:38:22 +04:00
else:
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))
2008-10-15 19:38:22 +04:00
def __str__(self):
"""
This method returns the progress bar string
"""
return getUnicode(self._progBar)