Fix log file encoding in ftpscan.py

[m_lib.git] / m_lib / net / ftp / ftpscan.py

diff --git a/m_lib/net/ftp/ftpscan.py b/m_lib/net/ftp/ftpscan.py
index ebff998f72a2c1c50ef0bd03bc537674a18618b4..eb0681c130bd8bdee61c1871f8154512cb4a1a2e 100755 (executable)
--- a/m_lib/net/ftp/ftpscan.py
+++ b/m_lib/net/ftp/ftpscan.py
@@ -5,6 +5,7 @@
 from __future__ import print_function
 import codecs
 import ftplib
+import sys
 from .ftpparse import ftpparse
 
 
@@ -222,7 +223,10 @@ def test(ftp_server, func, passive=None, reconnect=False):
    print(stop_time - start_time)
 
    logfname = "%s.list" % ftp_server
-   log = codecs.open(logfname, 'w', encoding='utf-8')
+   if sys.version_info[0] >= 3:
+       log = codecs.open(logfname, 'w', encoding='utf-8')
+   else:
+       log = open(logfname, 'w')
 
    for ftp_dir, files in tree:
       if ftp_dir == ftpscan_error_mark: