minor updates

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)

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 = {}