65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
#coding: utf-8
|
|
# +-------------------------------------------------------------------
|
|
# | YakPanel
|
|
# +-------------------------------------------------------------------
|
|
# | Copyright (c) 2015-2099 YakPanel(www.yakpanel.com) All rights reserved.
|
|
# +-------------------------------------------------------------------
|
|
# | Author: hwliang <hwl@yakpanel.com>
|
|
# +-------------------------------------------------------------------
|
|
import os,sys,public,json,time
|
|
class downloadFile:
|
|
logPath = 'data/speed.json'
|
|
timeoutCount = 0;
|
|
oldTime = 0;
|
|
writeTime = 0;
|
|
down_count = 0;
|
|
#下载文件
|
|
def DownloadFile(self,url,filename):
|
|
try:
|
|
path = os.path.dirname(filename)
|
|
if not os.path.exists(path): os.makedirs(path)
|
|
import urllib,socket,ssl
|
|
try:
|
|
ssl._create_default_https_context = ssl._create_unverified_context
|
|
except:pass
|
|
socket.setdefaulttimeout(30)
|
|
self.pre = 0;
|
|
self.oldTime = time.time();
|
|
if sys.version_info[0] == 2:
|
|
urllib.urlretrieve(url,filename=filename,reporthook= self.DownloadHook)
|
|
else:
|
|
urllib.request.urlretrieve(url,filename=filename,reporthook= self.DownloadHook)
|
|
speed = self.GetSpeed()
|
|
speed['pre'] = 100;
|
|
speed['used'] = speed['total']
|
|
self.WriteLogs(json.dumps(speed));
|
|
except:
|
|
if self.timeoutCount > 5: return;
|
|
self.timeoutCount += 1
|
|
time.sleep(5)
|
|
self.DownloadFile(url,filename)
|
|
|
|
#下载文件进度回调
|
|
def DownloadHook(self,count, blockSize, totalSize):
|
|
used = count * blockSize
|
|
pre1 = int((100.0 * used / totalSize))
|
|
my_time = time.time()
|
|
if self.pre != pre1 or (my_time - self.writeTime) > 1:
|
|
dspeed = ((count -self.down_count) * blockSize) / (my_time - self.oldTime)
|
|
speed = {'name':public.GetMsg("DOWNLOAD_FILE"),'total':totalSize,'used':used,'pre':self.pre,'speed':dspeed}
|
|
self.WriteLogs(json.dumps(speed))
|
|
self.pre = pre1
|
|
self.writeTime = my_time
|
|
self.down_count = count
|
|
self.oldTime = my_time
|
|
|
|
#取下载进度
|
|
def GetSpeed(self):
|
|
speedLog = public.ReadFile(self.logPath)
|
|
if not speedLog: return {'name':public.GetMsg("DOWNLOAD_FILE"),'total':0,'used':0,'pre':0,'speed':0}
|
|
return json.loads(speedLog)
|
|
|
|
#写输出日志
|
|
def WriteLogs(self,logMsg):
|
|
public.WriteFile(self.logPath,logMsg)
|