diff --git a/extra/shutils/pylint.py b/extra/shutils/pylint.py index e9489f76c..db7baa59c 100644 --- a/extra/shutils/pylint.py +++ b/extra/shutils/pylint.py @@ -10,6 +10,8 @@ import sys total = 0.0 count = 0 +__RATING__ = False + def check(module): global total, count @@ -20,15 +22,15 @@ def check(module): for line in pout: if re.match("E....:.", line): print line - if "Your code has been rated at" in line: - print line - score = re.findall("\d.\d\d", line)[0] - total += float(score) - count += 1 + if __RATING__ and "Your code has been rated at" in line: + print line + score = re.findall("\d.\d\d", line)[0] + total += float(score) + count += 1 if __name__ == "__main__": try: - print sys.argv + print sys.argv BASE_DIRECTORY = sys.argv[1] except IndexError: print "no directory specified, defaulting to current working directory" @@ -40,6 +42,7 @@ if __name__ == "__main__": filepath = os.path.join(root, name) check(filepath) - print "==" * 50 - print "%d modules found"% count - print "AVERAGE SCORE = %.02f"% (total / count) + if __RATING__: + print "==" * 50 + print "%d modules found"% count + print "AVERAGE SCORE = %.02f"% (total / count) diff --git a/lib/core/shell.py b/lib/core/shell.py index fafa52fff..064b6912b 100644 --- a/lib/core/shell.py +++ b/lib/core/shell.py @@ -14,6 +14,7 @@ import rlcompleter from lib.core import readlineng as readline from lib.core.common import backend from lib.core.data import kb +from lib.core.data import logger from lib.core.data import paths from lib.core.data import queries @@ -25,7 +26,11 @@ def loadHistory(): historyPath = os.path.expanduser(paths.SQLMAP_HISTORY) if os.path.exists(historyPath): - readline.read_history_file(historyPath) + try: + readline.read_history_file(historyPath) + except IOError, msg: + warnMsg = "there was a problem with loading of history file '%s' (%s)" % (historyPath, msg) + logger.warn(warnMsg) def queriesForAutoCompletion(): autoComplQueries = {}