Ejemplo n.º 1
0
 def getPic(self):
     global lst
     lst=[]
     self.photourls=[]
     self.db=Douban(self)
     self.ui.progressBar.setValue(0)
     self.orgLink=unicode(self.ui.lineEdit.text())
     self.dLink=self.db.tidyLink(self.orgLink.strip())
     if not self.dLink :
         self.ui.lineEdit.setText("")
         self.ui.lineEdit.setFocus()
         return
     logging.info("Ready to analysis album info")
     self.photourls=self.db.getPhotoLinks(self.dLink)
     if self.db.photoCount>=100:
         self.check()
         for i in self.checkedPages:
             self.finalLinks.extend(self.photourls[i*100:(i+1)*100-1])
     else:
         self.finalLinks=self.photourls
     if not self.finalLinks:
         self.ui.statusbar.showMessage(u"没有选择要下载的图片...")
         self.ui.lineEdit.setFocus()
         return
     self.ui.progressBar.setRange(0,len(self.finalLinks))
     lst=self.finalLinks
     for filename in lst:
         fname=filename.split("/")[-1]
         fname="photos/%s"%(fname)
         kv[fname]=0
     tasks=[]
     self.ui.statusbar.showMessage(u"开始下载图片...")
     logging.info("Ready to get pics")
     if QThread.idealThreadCount()>=10:
         threadCount=QThread.idealThreadCount()
     else:
         threadCount=10
     for i in range(threadCount):
         m=MutiDl(self)
         tasks.append(m)
     for task in tasks:
         time.sleep(0.5)
         task.start()
     logging.info("Muti DL started.")
     while self.isAlive(tasks):
         time.sleep(0.5)
         self.ui.progressBar.setValue(d)
     self.ui.progressBar.setValue(len(self.finalLinks))
     f=open("FailedFiles.txt","wa")
     for filename in failed:
         f.write("%s,"%filename)
     f.write("\n")
     f.close()
     self.db.alert(u"所有图片下载完成!")
     logging.info("Done!")
     self.ui.statusbar.showMessage(u"")
     del self.db
     del m
     self.ui.lineEdit.setFocus()
     self.ui.lineEdit.selectAll()
Ejemplo n.º 2
0
class DoubanPic(QMainWindow):
    def __init__(self,parent=None):
        super(DoubanPic,self).__init__(parent)
        self.ui=Ui_MainWindow()
        self.ui.setupUi(self)
        self.ui.progressBar.setValue(0)

        self.photourls=[]
        self.checkedPages=[]
        self.finalLinks=[]

        self.connect(self.ui.pushButton,SIGNAL("clicked()"),self.getPic)
        self.connect(self.ui.lineEdit,SIGNAL("returnPressed()"),self.getPic)


    def getPic(self):
        global lst
        lst=[]
        self.photourls=[]
        self.db=Douban(self)
        self.ui.progressBar.setValue(0)
        self.orgLink=unicode(self.ui.lineEdit.text())
        self.dLink=self.db.tidyLink(self.orgLink.strip())
        if not self.dLink :
            self.ui.lineEdit.setText("")
            self.ui.lineEdit.setFocus()
            return
        logging.info("Ready to analysis album info")
        self.photourls=self.db.getPhotoLinks(self.dLink)
        if self.db.photoCount>=100:
            self.check()
            for i in self.checkedPages:
                self.finalLinks.extend(self.photourls[i*100:(i+1)*100-1])
        else:
            self.finalLinks=self.photourls
        if not self.finalLinks:
            self.ui.statusbar.showMessage(u"没有选择要下载的图片...")
            self.ui.lineEdit.setFocus()
            return
        self.ui.progressBar.setRange(0,len(self.finalLinks))
        lst=self.finalLinks
        for filename in lst:
            fname=filename.split("/")[-1]
            fname="photos/%s"%(fname)
            kv[fname]=0
        tasks=[]
        self.ui.statusbar.showMessage(u"开始下载图片...")
        logging.info("Ready to get pics")
        if QThread.idealThreadCount()>=10:
            threadCount=QThread.idealThreadCount()
        else:
            threadCount=10
        for i in range(threadCount):
            m=MutiDl(self)
            tasks.append(m)
        for task in tasks:
            time.sleep(0.5)
            task.start()
        logging.info("Muti DL started.")
        while self.isAlive(tasks):
            time.sleep(0.5)
            self.ui.progressBar.setValue(d)
        self.ui.progressBar.setValue(len(self.finalLinks))
        f=open("FailedFiles.txt","wa")
        for filename in failed:
            f.write("%s,"%filename)
        f.write("\n")
        f.close()
        self.db.alert(u"所有图片下载完成!")
        logging.info("Done!")
        self.ui.statusbar.showMessage(u"")
        del self.db
        del m
        self.ui.lineEdit.setFocus()
        self.ui.lineEdit.selectAll()

    def check(self):
        if not self.photourls:
            QMessageBox.warning(self,u"提示",u"您还没有获取下载地址……")
            return
        d=Check(len(self.photourls),self)
        if d.exec_():
            self.checkedPages=d.checkedPages
    def isAlive(self,tasks):
        for task in tasks:
            if task.isAlive():
                return True
        return False