Making it work in Python3.

This commit is contained in:
Steven Black
2016-03-13 11:59:25 -04:00
parent 1dd1d41f5e
commit c565266416

View File

@@ -292,9 +292,12 @@ def removeDupsAndExcl(mergeFile):
os.makedirs(outputPath)
# Another mode is required to read and write the file in Python 3
finalFile = open(os.path.join(outputPath, 'hosts'), 'r+')
mergeFile.seek(0) # reset file pointer
if Python3:
finalFile = open(os.path.join(outputPath, 'hosts'), 'w+b')
else:
finalFile = open(os.path.join(outputPath, 'hosts'), 'w+')
mergeFile.seek(0) # reset file pointer
hostnames = set()
hostnames.add("localhost")
hostnames.add("localhost.localdomain")