Initial YakPanel commit
This commit is contained in:
64
class/downloadFile.py
Normal file
64
class/downloadFile.py
Normal file
@@ -0,0 +1,64 @@
|
||||
#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)
|
||||
Reference in New Issue
Block a user