PenTest/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-12-06 12:41:30 +00:00
Code Issues Projects Releases Wiki Activity

code refactoring regarding issue #184

Browse Source
This commit is contained in:
Miroslav Stampar
2010-05-24 11:12:40 +00:00
parent e9be60e1ac
commit 0197f8db5c
4 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lib/core/common.py
View File

@@ -356,7 +356,7 @@ def dataToStdout(data):
sys.stdout.write(data)
sys.stdout.flush()
except UnicodeEncodeError:
print data.encode("utf8")
print data.encode(conf.dataEncoding)
def dataToSessionFile(data):
if not conf.sessionFile:
@@ -375,7 +375,7 @@ def dataToOutFile(data):
rFile = filePathToString(conf.rFile)
rFilePath = "%s%s%s" % (conf.filePath, os.sep, rFile)
rFileFP = codecs.open(rFilePath, "wb", "utf-8")
rFileFP = codecs.open(rFilePath, "wb", conf.dataEncoding)
rFileFP.write(data)
rFileFP.flush()
@@ -416,7 +416,7 @@ def fileToStr(fileName):
@rtype: C{str}
"""
filePointer = codecs.open(fileName, "r", "utf-8")
filePointer = codecs.open(fileName, "r", conf.dataEncoding)
fileText = filePointer.read()
return fileText.replace(" ", "").replace("\t", "").replace("\r", "").replace("\n", " ")