def Clear(self):
''' Reinitialize the widget to reuse '''
self.Increase = 1
self.posMove = None
self.flag = None
self.T = 0
self.outFile = 'LDM_IIITA_DEVELOPMENT'
self.done = 0
self.dInfo = {}
self.dFile = {}
self.totalTime = 0
self.dLength = 0
self.downloader = YTube_Download(12)
self.progressBar.setValue(0)
self.dWidget.removeAllElements()
def Clear(self) :
''' Reinitialize the widget to reuse '''
self.Increase = 1
self.posMove=None
self.flag=None
self.T=0
self.outFile='LDM_IIITA_DEVELOPMENT'
self.done = 0
self.dInfo = {}
self.dFile={}
self.totalTime = 0
self.dLength = 0
self.downloader = YTube_Download(12)
self.progressBar.setValue(0)
self.dWidget.removeAllElements()
class DownloadWindow(QtGui.QWidget):
''' Main Window which contains download progress information & takes input from user '''
def __init__(self, PARTS, url='', parent=None):
QtGui.QWidget.__init__(self, parent)
self.PARTS = PARTS # Decides partitions
self.dFlag = Value(
'i') # Communicates actions through this shared variable
self.dFlag.value = 0
self.setMouseTracking(1)
self.recv = None
self.pause = QtGui.QPushButton(self)
self.resume = QtGui.QPushButton(self)
cancel = QtGui.QPushButton(self)
alwaysOnTop = QtGui.QRadioButton(self)
minimize = QtGui.QPushButton(self)
self.hideDL = QtGui.QPushButton(self)
self.setWindowFlags(QtCore.Qt.FramelessWindowHint)
label = QtGui.QLabel('Downloading from')
self.urlLabel = QtGui.QLabel(str(url))
self.progressBar = QtGui.QProgressBar()
self.progressBar.setMaximum(100)
self.dWidget = DownloadWidget()
self.Clear()
self.t = QtGui.QLabel('Time Elapsed: 0 sec')
self.Speed = QtGui.QLabel('Total Speed : 0 sec')
self.Size = QtGui.QLabel('Downloaded Size: 0 Bytes')
self.Remaining = QtGui.QLabel('Remaining : 0 Bytes')
self.pause.setText('Pause')
self.resume.setText('Resume')
cancel.setText('Cancel')
self.pause.setEnabled(False)
self.resume.setEnabled(False)
self.hideDL.setText("Hide Downloads")
alwaysOnTop.setText("On Top of all")
minimize.setText("Minimize To Tray")
hbox = QtGui.QHBoxLayout()
hbox.addWidget(self.pause)
hbox.addWidget(self.resume)
hbox.addWidget(cancel)
hbox1 = QtGui.QHBoxLayout()
hbox1.addWidget(alwaysOnTop)
hbox1.addWidget(self.hideDL)
hbox1.addWidget(minimize)
vbox = QtGui.QVBoxLayout()
vbox.addWidget(label)
vbox.addWidget(self.urlLabel)
vbox.addWidget(self.t)
vbox.addWidget(self.Speed)
vbox.addWidget(self.Size)
vbox.addWidget(self.Remaining)
vbox.addWidget(self.progressBar)
vbox.addLayout(hbox1)
vbox.addWidget(self.dWidget)
vbox.addLayout(hbox)
self.setLayout(vbox)
self.timer = QtCore.QTimer()
self.connect(self.timer, QtCore.SIGNAL('timeout()'),
self.UpdateInterface)
self.connect(self.hideDL, QtCore.SIGNAL('clicked()'),
self.hideDownloads)
self.connect(self.pause, QtCore.SIGNAL('clicked()'),
self.pauseDownload)
self.connect(self.resume, QtCore.SIGNAL('clicked()'),
self.resumeDownload)
self.connect(cancel, QtCore.SIGNAL('clicked()'), self.cancelDownload)
self.connect(alwaysOnTop, QtCore.SIGNAL('toggled(bool)'),
self.putOnTop)
self.connect(minimize, QtCore.SIGNAL('clicked()'), self.minimizeToTray)
def Clear(self):
''' Reinitialize the widget to reuse '''
self.Increase = 1
self.posMove = None
self.flag = None
self.T = 0
self.outFile = 'LDM_IIITA_DEVELOPMENT'
self.done = 0
self.dInfo = {}
self.dFile = {}
self.totalTime = 0
self.dLength = 0
self.downloader = YTube_Download(12)
self.progressBar.setValue(0)
self.dWidget.removeAllElements()
def minimizeToTray(self):
''' When widget is to be minimize it's information is added to the MainFrame Window which is informaed by setting a bit '''
if not (self.packet == None):
self.packet.flag = self.packet.flag | 0x0001
self.hide()
def putOnTop(self, bool):
if bool:
self.setWindowFlags(QtCore.Qt.Widget
| QtCore.Qt.WindowStaysOnTopHint
| QtCore.Qt.FramelessWindowHint)
self.show()
else:
self.setWindowFlags(QtCore.Qt.Widget
| QtCore.Qt.FramelessWindowHint)
self.show()
def hideDownloads(self):
if self.Increase == 1:
self.Increase = 0
self.resize(700, 225)
self.hideDL.setText("Show Downloads")
else:
self.Increase = 1
if self.dWidget.index > 1:
self.resize(700, 225 + (self.dWidget.index - 1) * 32)
self.hideDL.setText("Hide Downloads")
def pauseDownload(self):
self.dFlag.value = self.dFlag.value | 0x0002
self.pause.setEnabled(False)
def resumeDownload(self):
if self.dFlag.value & 0x0010:
self.downloader.resume()
self.pause.setEnabled(True)
self.resume.setEnabled(False)
self.dFlag.value = self.dFlag.value & 0xffef
def cancelDownload(self):
if self.downloader.paused == 0:
self.dFlag.value = self.dFlag.value | 0x0004
else:
if self.isHidden():
self.show()
ok = QtGui.QMessageBox(QtGui.QMessageBox.Warning, '',
"You can't cancel while pause",
QtGui.QMessageBox.Ok, self,
QtCore.Qt.FramelessWindowHint)
ok.exec_()
def keyPressEvent(self, event):
if event.key() == QtCore.Qt.Key_Escape:
self.hide()
def mousePressEvent(self, event):
self.posMove = event.pos()
def mouseMoveEvent(self, event):
if self.posMove:
self.move(self.pos().x() + event.pos().x() - self.posMove.x(),
self.pos().y() + event.pos().y() - self.posMove.y())
def mouseReleaseEvent(self, event):
self.posMove = None
def getPartDownload(self, obj, packet, wMutex, bit):
''' When a download to be resumes it's partial information is fetched & download is resumed '''
self.wMutex = wMutex
self.packet = packet
self.urlLabel.setText(str(obj.url)[0:80])
self.recv = Queue()
self.setWindowTitle(str(obj.url))
self.outFile = obj.fileName
self.dLength = int(obj.length)
info = Info()
info.url = obj.url
info.fileName = obj.fileName
info.date = obj.date
info.time = obj.time
info.length = obj.length
obj = dLoadInfo().getSpecificInformation(obj)
if obj == None:
raise urllib2.URLError
info.partition = obj.partition
for i in range(len(info.partition)):
self.dFile[obj.partition[i][0]] = None
if obj == None:
return 0
else:
if bit == 1:
fName, length, cmtd = self.downloader.YTubePartition(
info, self.recv, self.PARTS, wMutex, self.dFlag, bit)
self.PARTS = self.PARTS - cmtd
if fName == None:
self.dLength = int(obj.length) - length
self.timer.start(1000)
return 1
else:
self.packet.fileName = fName
self.packet.length = length
self.dLength = length
self.fileName = fName
self.outFile = fName
self.dLength = length
self.timer.start(1000)
return 2
self.timer.start(1000)
return 1
def getDownload(self, url, packet, wMutex, bit=0):
self.wMutex = wMutex
self.packet = packet
self.urlLabel.setText(str(url)[0:80])
self.recv = Queue()
self.setWindowTitle(url)
info = Info()
info.url = url
info.date = self.packet.dt
info.time = self.packet.tm
if bit == 0:
self.outFile, self.dLength = self.downloader.YTubeDownload(
url, self.recv, self.PARTS, info, wMutex, self.dFlag)
else:
self.outFile, self.dLength = self.downloader.Download(
url, self.recv, self.PARTS, info, wMutex, self.dFlag)
info.fileName = str(self.outFile)
info.length = self.dLength
self.info = info
self.Remaining.setText('Remaining : ' + str(self.dLength) + ' Bytes')
self.timer.start(1000)
return self.outFile, self.dLength
def getTime(self, t):
z = ''
if t / 60 > 0:
z = str(t / 60) + 'min ' + str(t % 60) + 'sec'
else:
z = str(t) + 'sec'
return z
def updateInfo(self, t, speed, size, remaining):
self.t.setText('Time Elapsed: ' + self.getTime(t))
if speed / 1024 > 0:
self.Speed.setText('Avg. Speed : ' + str(speed / 1024) + ' MBPS')
else:
self.Speed.setText('Avg. Speed : ' + str(speed) + ' KBPS')
self.Size.setText('Downloaded Size: ' + str(size) + ' Bytes')
self.Remaining.setText('Remaining : ' + str(remaining) + ' Bytes')
def Watch(self):
''' continously monitors the flags bits & checks if any signal is send & performs corresponding actions '''
if self.packet.flag & 0x0080:
self.cancelDownload()
if self.packet.flag & 0x0001:
if self.isVisible():
self.hide()
self.packet.flag = self.packet.flag & 0xfffe
if self.packet.flag & 0x0010:
if self.isHidden():
self.show()
self.packet.flag = self.packet.flag & 0xffef
if self.dFlag.value & 0x0010:
self.resume.setEnabled(True)
if self.dFlag.value & 0x0008:
self.packet.flag = self.packet.flag | 0x0002
ok = QtGui.QMessageBox(QtGui.QMessageBox.Information, 'Cancel',
"Download canceled", QtGui.QMessageBox.Ok,
self, QtCore.Qt.FramelessWindowHint)
ok.exec_()
self.dFlag.value = self.dFlag.value & 0xfff7
if self.dFlag.value & 0x0100:
self.timer.stop()
ok = QtGui.QMessageBox(QtGui.QMessageBox.Information, 'Info..',
"Download complete", QtGui.QMessageBox.Ok,
self, QtCore.Qt.FramelessWindowHint)
ok.exec_()
self.dFlag.value = self.dFlag.value & 0xfeff
self.packet.flag = self.packet.flag | 0x0002
def UpdateInterface(self):
''' If data is fetched progress bar & other information is updated '''
self.totalTime = self.totalTime + 1
total = 0
length = 0
for key in self.dInfo:
obj = self.dInfo[key]
obj.timeElapsed = obj.timeElapsed + 1
total += obj.downloadTillNow
length += obj.justDownload
if obj.refresh == 1:
self.dWidget.updateTable(
obj.index, (100 * obj.downloadTillNow) / (obj.UL - obj.LL),
obj.justDownload / (1024 * obj.timeElapsed),
str(self.dInfo[key].LL) + ' to ' + str(self.dInfo[key].UL))
obj.timeElapsed = 0
obj.refresh = 0
obj.justDownload = 0
if total == 0:
self.packet.timeLeft = 'inf'
else:
self.packet.timeLeft = str(
(self.dLength - total) * self.totalTime / total)
self.packet.downloaded = total
if self.dLength > 0:
self.progressBar.setValue((100 * total) / self.dLength)
self.updateInfo(self.totalTime, length / 1024, total,
self.dLength - total)
self.Tracker()
if not (self.done == 1):
self.timer.start(1000)
else:
self.timer.stop()
self.Watch()
def Tracker(self):
'''Monitors the progress of download from this function '''
length = 0
i = 0
if not (self.T == self.PARTS):
while not (self.recv.empty()) and i <= 20:
data = self.recv.get()
if not (data == None):
data = data.split(':')
if data[0] == 'Init': # If initialize add entry to the download window
self.pause.setEnabled(True)
obj = DownloadInfo()
obj.index = self.dWidget.index
obj.LL = int(data[1])
obj.UL = int(data[2])
self.dInfo[int(data[1])] = obj
self.dWidget.addDownloadProcess()
if self.dWidget.index > 1 and self.Increase:
self.resize(700,
225 + (self.dWidget.index - 1) * 32)
elif data[
0] == 'Done': # if done download increase commited process number
self.T = self.T + 1
else: # else just add entry to process it by other functions
obj = self.dInfo[int(data[1])]
obj.justDownload += int(data[0]) - obj.downloadTillNow
length += int(data[0]) - obj.downloadTillNow
obj.downloadTillNow = int(data[0])
obj.refresh = 1
i = i + 1
else: # if all downloads complete check if file avail & createfinaloutput
ok = QtGui.QMessageBox.information(self, 'Hello User',
"Download Complete")
if len(self.dFile) > 0:
thread.start_new_thread(self.createFinalOutput,
(self.dFile, self.outFile))
else:
thread.start_new_thread(self.createFinalOutput,
(self.dInfo, self.outFile))
self.done = 1
self.packet.free = 1
if self.isHidden():
self.show()
def createFinalOutput(self, dI, fileName):
''' it reads the given sufficient input & builds the final output file at the destination '''
key = []
for k in dI:
key.append(k)
key.sort()
outFile = fileName
temp = fileName
i = 0
try:
if os.path.exists(self.packet.dLocation):
while os.path.exists(self.packet.dLocation + '/' + outFile):
outFile = temp + '_' + str(i)
i = i + 1
f = open(self.packet.dLocation + '/' + outFile, 'wb')
except IOError:
f = open(self.packet.dLocation + '/' + outFile, 'w')
f.close()
f = open(self.packet.dLocation + '/' + outFile, 'wb')
for k in key:
f1 = open(os.getcwd() + '/Download/' + fileName + '_LDM' + '/' +
'LDM_' + str(k))
data = f1.read()
f.write(data)
f1.close()
os.remove(os.getcwd() + '/Download/' + fileName + '_LDM' + '/' +
'LDM_' + str(k))
os.rmdir(os.getcwd() + '/Download/' + fileName + '_LDM')
f.close()
self.dFile = {}
self.packet.flag = self.packet.flag | 0x0004 # update the flag information to say that download complete
class DownloadWindow(QtGui.QWidget) :
''' Main Window which contains download progress information & takes input from user '''
def __init__(self,PARTS,url='',parent=None) :
QtGui.QWidget.__init__(self,parent)
self.PARTS=PARTS # Decides partitions
self.dFlag = Value('i') # Communicates actions through this shared variable
self.dFlag.value = 0
self.setMouseTracking(1)
self.recv=None
self.pause = QtGui.QPushButton(self)
self.resume = QtGui.QPushButton(self)
cancel = QtGui.QPushButton(self)
alwaysOnTop = QtGui.QRadioButton(self)
minimize = QtGui.QPushButton(self)
self.hideDL = QtGui.QPushButton(self)
self.setWindowFlags(QtCore.Qt.FramelessWindowHint)
label = QtGui.QLabel('Downloading from')
self.urlLabel = QtGui.QLabel(str(url))
self.progressBar = QtGui.QProgressBar()
self.progressBar.setMaximum(100)
self.dWidget = DownloadWidget()
self.Clear()
self.t = QtGui.QLabel('Time Elapsed: 0 sec')
self.Speed = QtGui.QLabel('Total Speed : 0 sec')
self.Size = QtGui.QLabel('Downloaded Size: 0 Bytes')
self.Remaining = QtGui.QLabel('Remaining : 0 Bytes')
self.pause.setText('Pause')
self.resume.setText('Resume')
cancel.setText('Cancel')
self.pause.setEnabled(False)
self.resume.setEnabled(False)
self.hideDL.setText("Hide Downloads")
alwaysOnTop.setText("On Top of all")
minimize.setText("Minimize To Tray")
hbox = QtGui.QHBoxLayout()
hbox.addWidget(self.pause)
hbox.addWidget(self.resume)
hbox.addWidget(cancel)
hbox1 = QtGui.QHBoxLayout()
hbox1.addWidget(alwaysOnTop)
hbox1.addWidget(self.hideDL)
hbox1.addWidget(minimize)
vbox = QtGui.QVBoxLayout()
vbox.addWidget(label)
vbox.addWidget(self.urlLabel)
vbox.addWidget(self.t)
vbox.addWidget(self.Speed)
vbox.addWidget(self.Size)
vbox.addWidget(self.Remaining)
vbox.addWidget(self.progressBar)
vbox.addLayout(hbox1)
vbox.addWidget(self.dWidget)
vbox.addLayout(hbox)
self.setLayout(vbox)
self.timer = QtCore.QTimer()
self.connect(self.timer,QtCore.SIGNAL('timeout()'),self.UpdateInterface)
self.connect(self.hideDL,QtCore.SIGNAL('clicked()'),self.hideDownloads)
self.connect(self.pause,QtCore.SIGNAL('clicked()'),self.pauseDownload)
self.connect(self.resume,QtCore.SIGNAL('clicked()'),self.resumeDownload)
self.connect(cancel,QtCore.SIGNAL('clicked()'),self.cancelDownload)
self.connect(alwaysOnTop,QtCore.SIGNAL('toggled(bool)'),self.putOnTop)
self.connect(minimize,QtCore.SIGNAL('clicked()'),self.minimizeToTray)
def Clear(self) :
''' Reinitialize the widget to reuse '''
self.Increase = 1
self.posMove=None
self.flag=None
self.T=0
self.outFile='LDM_IIITA_DEVELOPMENT'
self.done = 0
self.dInfo = {}
self.dFile={}
self.totalTime = 0
self.dLength = 0
self.downloader = YTube_Download(12)
self.progressBar.setValue(0)
self.dWidget.removeAllElements()
def minimizeToTray(self) :
''' When widget is to be minimize it's information is added to the MainFrame Window which is informaed by setting a bit '''
if not(self.packet == None) :
self.packet.flag=self.packet.flag | 0x0001
self.hide()
def putOnTop(self,bool) :
if bool :
self.setWindowFlags(QtCore.Qt.Widget | QtCore.Qt.WindowStaysOnTopHint | QtCore.Qt.FramelessWindowHint)
self.show()
else :
self.setWindowFlags(QtCore.Qt.Widget | QtCore.Qt.FramelessWindowHint)
self.show()
def hideDownloads(self) :
if self.Increase == 1:
self.Increase = 0
self.resize(700,225)
self.hideDL.setText("Show Downloads")
else :
self.Increase = 1
if self.dWidget.index > 1:
self.resize(700,225+(self.dWidget.index-1)*32)
self.hideDL.setText("Hide Downloads")
def pauseDownload(self) :
self.dFlag.value =self.dFlag.value | 0x0002
self.pause.setEnabled(False)
def resumeDownload(self) :
if self.dFlag.value & 0x0010 :
self.downloader.resume()
self.pause.setEnabled(True)
self.resume.setEnabled(False)
self.dFlag.value = self.dFlag.value & 0xffef
def cancelDownload(self) :
if self.downloader.paused==0 :
self.dFlag.value = self.dFlag.value | 0x0004
else :
if self.isHidden() :
self.show()
ok = QtGui.QMessageBox(QtGui.QMessageBox.Warning,'',"You can't cancel while pause",QtGui.QMessageBox.Ok,self, QtCore.Qt.FramelessWindowHint)
ok.exec_()
def keyPressEvent(self,event) :
if event.key() == QtCore.Qt.Key_Escape :
self.hide()
def mousePressEvent(self,event) :
self.posMove = event.pos()
def mouseMoveEvent(self,event) :
if self.posMove :
self.move(self.pos().x()+event.pos().x()-self.posMove.x(),self.pos().y()+event.pos().y()-self.posMove.y())
def mouseReleaseEvent(self,event) :
self.posMove=None
def getPartDownload(self,obj,packet,wMutex,bit) :
''' When a download to be resumes it's partial information is fetched & download is resumed '''
self.wMutex=wMutex
self.packet=packet
self.urlLabel.setText(str(obj.url)[0:80])
self.recv = Queue()
self.setWindowTitle(str(obj.url))
self.outFile=obj.fileName
self.dLength=int(obj.length)
info = Info()
info.url=obj.url
info.fileName=obj.fileName
info.date=obj.date
info.time=obj.time
info.length=obj.length
obj=dLoadInfo().getSpecificInformation(obj)
if obj==None :
raise urllib2.URLError
info.partition=obj.partition
for i in range(len(info.partition)) :
self.dFile[obj.partition[i][0]]=None
if obj == None :
return 0
else :
if bit==1 :
fName,length,cmtd=self.downloader.YTubePartition(info,self.recv,self.PARTS,wMutex,self.dFlag,bit)
self.PARTS=self.PARTS-cmtd
if fName==None :
self.dLength=int(obj.length)-length
self.timer.start(1000)
return 1
else :
self.packet.fileName=fName
self.packet.length=length
self.dLength=length
self.fileName=fName
self.outFile = fName
self.dLength=length
self.timer.start(1000)
return 2
self.timer.start(1000)
return 1
def getDownload(self,url,packet,wMutex,bit=0) :
self.wMutex=wMutex
self.packet = packet
self.urlLabel.setText(str(url)[0:80])
self.recv = Queue()
self.setWindowTitle(url)
info = Info()
info.url=url
info.date=self.packet.dt
info.time=self.packet.tm
if bit == 0 :
self.outFile,self.dLength = self.downloader.YTubeDownload(url,self.recv,self.PARTS,info,wMutex,self.dFlag)
else :
self.outFile,self.dLength = self.downloader.Download(url,self.recv,self.PARTS,info,wMutex,self.dFlag)
info.fileName=str(self.outFile)
info.length=self.dLength
self.info=info
self.Remaining.setText('Remaining : '+str(self.dLength)+' Bytes')
self.timer.start(1000)
return self.outFile,self.dLength
def getTime(self,t) :
z=''
if t/60 > 0 :
z=str(t/60)+'min '+str(t%60)+'sec'
else :
z=str(t)+'sec'
return z
def updateInfo(self,t,speed,size,remaining) :
self.t.setText('Time Elapsed: '+self.getTime(t))
if speed/1024 > 0 :
self.Speed.setText('Avg. Speed : '+str(speed/1024)+' MBPS')
else :
self.Speed.setText('Avg. Speed : '+str(speed)+' KBPS')
self.Size.setText('Downloaded Size: '+str(size)+' Bytes')
self.Remaining.setText('Remaining : '+str(remaining)+' Bytes')
def Watch(self) :
''' continously monitors the flags bits & checks if any signal is send & performs corresponding actions '''
if self.packet.flag & 0x0080 :
self.cancelDownload()
if self.packet.flag & 0x0001 :
if self.isVisible() :
self.hide()
self.packet.flag=self.packet.flag & 0xfffe
if self.packet.flag & 0x0010 :
if self.isHidden() :
self.show()
self.packet.flag=self.packet.flag & 0xffef
if self.dFlag.value & 0x0010 :
self.resume.setEnabled(True)
if self.dFlag.value & 0x0008 :
self.packet.flag=self.packet.flag | 0x0002
ok = QtGui.QMessageBox(QtGui.QMessageBox.Information,'Cancel',"Download canceled",QtGui.QMessageBox.Ok,self,QtCore.Qt.FramelessWindowHint)
ok.exec_()
self.dFlag.value = self.dFlag.value & 0xfff7
if self.dFlag.value & 0x0100 :
self.timer.stop()
ok = QtGui.QMessageBox(QtGui.QMessageBox.Information,'Info..',"Download complete",QtGui.QMessageBox.Ok,self,QtCore.Qt.FramelessWindowHint)
ok.exec_()
self.dFlag.value=self.dFlag.value & 0xfeff
self.packet.flag = self.packet.flag | 0x0002
def UpdateInterface(self) :
''' If data is fetched progress bar & other information is updated '''
self.totalTime=self.totalTime+1
total=0
length=0
for key in self.dInfo :
obj=self.dInfo[key]
obj.timeElapsed=obj.timeElapsed+1
total+=obj.downloadTillNow
length+=obj.justDownload
if obj.refresh == 1 :
self.dWidget.updateTable(obj.index,(100*obj.downloadTillNow)/(obj.UL-obj.LL),obj.justDownload/(1024*obj.timeElapsed),str(self.dInfo[key].LL)+' to '+str(self.dInfo[key].UL))
obj.timeElapsed = 0
obj.refresh = 0
obj.justDownload=0
if total == 0:
self.packet.timeLeft='inf'
else :
self.packet.timeLeft=str((self.dLength-total)*self.totalTime/total)
self.packet.downloaded=total
if self.dLength > 0:
self.progressBar.setValue((100*total)/self.dLength)
self.updateInfo(self.totalTime,length/1024,total,self.dLength-total)
self.Tracker()
if not(self.done==1) :
self.timer.start(1000)
else :
self.timer.stop()
self.Watch()
def Tracker(self) :
'''Monitors the progress of download from this function '''
length = 0
i=0
if not(self.T==self.PARTS) :
while not(self.recv.empty()) and i <= 20 :
data = self.recv.get()
if not(data == None) :
data=data.split(':')
if data[0] == 'Init' : # If initialize add entry to the download window
self.pause.setEnabled(True)
obj = DownloadInfo()
obj.index = self.dWidget.index
obj.LL = int(data[1])
obj.UL = int(data[2])
self.dInfo[int(data[1])]=obj
self.dWidget.addDownloadProcess()
if self.dWidget.index > 1 and self.Increase :
self.resize(700,225+(self.dWidget.index-1)*32)
elif data[0] == 'Done': # if done download increase commited process number
self.T=self.T+1
else : # else just add entry to process it by other functions
obj = self.dInfo[int(data[1])]
obj.justDownload += int(data[0])-obj.downloadTillNow
length+=int(data[0])-obj.downloadTillNow
obj.downloadTillNow = int(data[0])
obj.refresh = 1
i=i+1
else : # if all downloads complete check if file avail & createfinaloutput
ok=QtGui.QMessageBox.information(self,'Hello User',"Download Complete")
if len(self.dFile) > 0 :
thread.start_new_thread(self.createFinalOutput,(self.dFile,self.outFile))
else :
thread.start_new_thread(self.createFinalOutput,(self.dInfo,self.outFile))
self.done = 1
self.packet.free=1
if self.isHidden() :
self.show()
def createFinalOutput(self,dI,fileName) :
''' it reads the given sufficient input & builds the final output file at the destination '''
key=[]
for k in dI :
key.append(k)
key.sort()
outFile=fileName
temp=fileName
i=0
try :
if os.path.exists(self.packet.dLocation) :
while os.path.exists(self.packet.dLocation+'/'+outFile) :
outFile=temp+'_'+str(i)
i=i+1
f = open(self.packet.dLocation+'/'+outFile,'wb')
except IOError :
f = open(self.packet.dLocation+'/'+outFile,'w')
f.close()
f = open(self.packet.dLocation+'/'+outFile,'wb')
for k in key :
f1=open(os.getcwd()+'/Download/'+fileName+'_LDM'+'/'+'LDM_'+str(k))
data = f1.read()
f.write(data)
f1.close()
os.remove(os.getcwd()+'/Download/'+fileName+'_LDM'+'/'+'LDM_'+str(k))
os.rmdir(os.getcwd()+'/Download/'+fileName+'_LDM')
f.close()
self.dFile={}
self.packet.flag = self.packet.flag | 0x0004 # update the flag information to say that download complete
