Merge pull request #96 from fishso/master

fix python3 show error :'str' object has no attribute 'decode'
This commit is contained in:
Alone_Monkey 2020-01-15 19:04:45 +08:00 committed by GitHub
commit 64b58d1c3e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions

View File

@ -23,7 +23,8 @@ from scp import SCPClient
from tqdm import tqdm
import traceback
if sys.version_info[0] < 3:
IS_PY2 = sys.version_info[0] < 3
if IS_PY2:
reload(sys)
sys.setdefaultencoding('utf8')
@ -96,7 +97,11 @@ def on_message(message, data):
last_sent = [0]
def progress(filename, size, sent):
t.desc = os.path.basename(filename).decode("utf-8")
baseName = os.path.basename(filename)
if IS_PY2:
t.desc = baseName.decode("utf-8")
else:
t.desc = baseName
t.total = size
t.update(sent - last_sent[0])
last_sent[0] = 0 if size == sent else sent