From 1de1d923ab34daff469b8f7d6bb3e8835d8b3025 Mon Sep 17 00:00:00 2001
From: "weiye.wang" <kj_wangweiye@126.com>
Date: Sun, 31 Mar 2024 21:05:57 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20=E4=B8=8B=E8=BD=BD?=
 =?UTF-8?q?=E5=B0=8F=E9=97=B2=E7=AD=94=E9=A2=98=E6=95=B0=E6=8D=AE.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 工具v2/下载小闲答题数据.py | 25 ++++++++++++++++++-------
 1 file changed, 18 insertions(+), 7 deletions(-)

diff --git a/工具v2/下载小闲答题数据.py b/工具v2/下载小闲答题数据.py
index c5ecdffc..bd06f38f 100644
--- a/工具v2/下载小闲答题数据.py
+++ b/工具v2/下载小闲答题数据.py
@@ -1,4 +1,5 @@
 from PySide6.QtWidgets import QWidget, QApplication, QFileDialog, QTableWidgetItem
+from PySide6.QtGui import QColor
 from Ui_下载小闲答题数据 import Ui_Form
 from selenium import webdriver
 from selenium.webdriver.common.by import By
@@ -155,9 +156,12 @@ def RenameRecentzip(folder,message): #将folder中最新的zip文件名加上(me
     filelist.sort(key = lambda x:os.path.getmtime(os.path.join(folder,x)),reverse = True)
     filepath = os.path.join(folder,filelist[0])
     renamedfilepath = filepath[:-4]+f"({message}).zip"
-    os.rename(filepath,renamedfilepath)
-    print(f"已重命名为: {os.path.split(renamedfilepath)[-1]}")
-
+    try:
+        os.rename(filepath,renamedfilepath)
+        print(f"已重命名为: {os.path.split(renamedfilepath)[-1]}")
+    except:
+        return 1
+    return 0
 
 def DownloadZips(driver,alist=[]): #下载页面上的指定zip文件
     page_info = GetPageInfo(driver)
@@ -283,8 +287,10 @@ class MyWindow(QWidget,Ui_Form):
         self.homeworklist = GetValidHomeworks(self.driver,self.lineEdit_startdate.text(),self.lineEdit_enddate.text(),self.lineEdit_graderegex.text())
         for i in range(len(self.homeworklist)):
             self.tableWidget.insertRow(i)
-            self.tableWidget.setItem(i,0,QTableWidgetItem(self.homeworklist[i][0]))
+            self.tableWidget.setItem(i,0,QTableWidgetItem(f"{self.homeworklist[i][0]['name']} / {self.homeworklist[i][0]['grade']}"))
+            self.tableWidget.setItem(i,1,QTableWidgetItem(""))
     def exec(self):
+        i = 0
         for hmwk,page in self.homeworklist:
             foundhmwk = False
             downloaded = False
@@ -302,12 +308,17 @@ class MyWindow(QWidget,Ui_Form):
                 print(f"正在下载 {hmwk['name']} 的 zip 文件")
                 try:
                     message = DownloadZipwithDetail(self.driver,hmwkindex)
-                    RenameRecentzip(self.outputfolder,message)
-                    downloaded = True
+                    renamed = RenameRecentzip(self.outputfolder,message)
+                    if renamed == 0:
+                        downloaded = True
+                        self.tableWidget.item(i,1).setBackground(QColor("green"))
+                        QApplication.processEvents()
                 except:
                     print(Exception)
             if not downloaded:
-                print(f"@@@@@@未能下载 {hmwk['name']}")
+                print(f"@@@@@@未能下载并改名 {hmwk['name']}")
+                self.tableWidget.item(i,1).setBackground(QColor("red"))
+            i += 1