1
1
mirror of https://github.com/KenanZhu/AutoLibrary.git synced 2026-06-18 23:43:02 +08:00

Compare commits

..

3 Commits

Author SHA1 Message Date
KenanZhu 631785122b fix(gui.ALConfigWidget): fix the width of web driver path line edit 2025-11-08 18:31:07 +08:00
KenanZhu 82ea40d3dc refactor(LibChecker): fix(AutoLib): extract reservation decode logic into separate function
: adjust default browser size to prevent page scale issues

Extract the reservation decode logic into a separate
function within LibChecker to improve code structure
and enable more structured return values for
decision-making.

Testing of the refactored LibChecker revealed that
page scaling could cause unexpected behavior.
Change the default browser size from
1280x720 to 1920x1080 to resolve this.
2025-11-08 12:53:24 +08:00
KenanZhu 1244084c75 style(*): remove the spare punctuation marks ',' at the end of function parameters 2025-11-08 12:51:33 +08:00
12 changed files with 172 additions and 114 deletions
+7 -6
View File
@@ -30,7 +30,7 @@ class AutoLib(MsgBase):
def __init__( def __init__(
self, self,
input_queue: queue.Queue, input_queue: queue.Queue,
output_queue: queue.Queue, output_queue: queue.Queue
): ):
super().__init__(input_queue, output_queue) super().__init__(input_queue, output_queue)
@@ -52,7 +52,8 @@ class AutoLib(MsgBase):
edge_options.add_argument("--no-sandbox") edge_options.add_argument("--no-sandbox")
edge_options.add_argument("--disable-dev-shm-usage") edge_options.add_argument("--disable-dev-shm-usage")
edge_options.add_argument("--window-size=1280,720") # must be 1920x1080, otherwise the page will cause some elements not accessible
edge_options.add_argument("--window-size=1920,1080")
edge_options.add_argument("--remote-allow-origins=*") edge_options.add_argument("--remote-allow-origins=*")
# omit ssl errors and verbose log level # omit ssl errors and verbose log level
@@ -114,7 +115,7 @@ class AutoLib(MsgBase):
def __waitResponseLoad( def __waitResponseLoad(
self, self
) -> bool: ) -> bool:
# wait for page load # wait for page load
@@ -154,7 +155,7 @@ class AutoLib(MsgBase):
self, self,
username: str, username: str,
password: str, password: str,
reserve_info: dict, reserve_info: dict
) -> int: ) -> int:
# result : 0 - success, 1 - failed, 2 - passed # result : 0 - success, 1 - failed, 2 - passed
@@ -202,7 +203,7 @@ class AutoLib(MsgBase):
def run( def run(
self, self,
system_config_reader: ConfigReader, system_config_reader: ConfigReader,
users_config_reader: ConfigReader, users_config_reader: ConfigReader
): ):
self.__system_config_reader = system_config_reader self.__system_config_reader = system_config_reader
@@ -245,7 +246,7 @@ class AutoLib(MsgBase):
def close( def close(
self, self
) -> bool: ) -> bool:
if self.__driver: if self.__driver:
+134 -77
View File
@@ -57,6 +57,83 @@ class LibChecker(LibOperator):
return True return True
def __decodeReserveInfo(
self,
info_elements
) -> str:
location = ""
status = ""
for info in info_elements:
if "已预约" in info.text:
status = "已预约"
elif "使用中" in info.text:
status = "使用中"
elif "已完成" in info.text:
status = "已完成"
elif "已结束使用" in info.text:
status = "已结束使用"
elif "已取消" in info.text:
status = "已取消"
elif "失约" in info.text:
status = "失约"
elif "图书馆" in info.text:
location = info.text.strip()
return {
"location": location,
"status": status,
}
def __decodeReserveRecord(
self,
reservation
) -> dict:
try:
time_element = reservation.find_element(
By.CSS_SELECTOR, "dt"
)
info_elements = reservation.find_elements(
By.CSS_SELECTOR, "a"
)
except:
return None
# process time element to get the date string
time_str = time_element.text.strip()
today = datetime.now().date()
if "明天" in time_str:
target_date = today + timedelta(days=1)
date_str = target_date.strftime("%Y-%m-%d")
elif "今天" in time_str:
target_date = today
date_str = target_date.strftime("%Y-%m-%d")
elif "昨天" in time_str:
target_date = today - timedelta(days=1)
date_str = target_date.strftime("%Y-%m-%d")
else:
date_match = re.search(r"(\d{4}-\d{1,2}-\d{1,2})", time_str)
if date_match:
date_str = date_match.group(1)
else:
date_str = ""
time_match = re.search(r"(\d{1,2}:\d{2}) -- (\d{1,2}:\d{2})", time_str)
if time_match:
begin_time = time_match.group(1)
end_time = time_match.group(2)
else:
time_str = ""
info = self.__decodeReserveInfo(info_elements)
return {
"date": date_str,
"time": {
"begin": begin_time,
"end": end_time,
},
"info": info
}
def __getReserveRecord( def __getReserveRecord(
self, self,
wanted_date: str, wanted_date: str,
@@ -66,75 +143,58 @@ class LibChecker(LibOperator):
if wanted_date is None: if wanted_date is None:
self._showTrace("日期未指定, 无法检查当前预约状态") self._showTrace("日期未指定, 无法检查当前预约状态")
return None return None
self._showTrace(f"正在检查用户在日期 {wanted_date} 是否有预约状态为 {wanted_status} 的预约记录......") self._showTrace(f"正在检查用户在 {wanted_date} 是否有预约状态为 {wanted_status} 的预约记录......")
date_obj = datetime.strptime(wanted_date, "%Y-%m-%d").date() date_obj = datetime.strptime(wanted_date, "%Y-%m-%d").date()
checked_count = 0 checked_count = 0
max_check_times = 3 # we only check (3*4=)12 reservations max_check_times = 6 # we only check (4*(6-1)=)20 reservations, the last time cant be checked
if not self.__navigateToReserveRecordPage(): if not self.__navigateToReserveRecordPage():
return None return None
for _ in range(max_check_times): for _ in range(max_check_times):
try: try:
# check if there's any reservation on the date # check if there's any reservation on the date
WebDriverWait(self.__driver, 2).until(
EC.presence_of_element_located((By.CSS_SELECTOR, ".myReserveList > dl"))
)
reservations = self.__driver.find_elements( reservations = self.__driver.find_elements(
By.CSS_SELECTOR, ".myReserveList dl" By.CSS_SELECTOR, ".myReserveList > dl:not(#moreBlock)"
) )
except: except:
self._showTrace("加载预约记录失败 !") self._showTrace("加载预约记录失败 !")
return None return None
for i in range(checked_count, len(reservations) - 1): # the last one is load button for i in range(checked_count, len(reservations)): # the last one is load button
reservation = reservations[i] reservation = reservations[i]
try: record = self.__decodeReserveRecord(reservation)
time_element = reservation.find_element( print(record)
By.CSS_SELECTOR, "dt" if record is None:
)
info_elements = reservation.find_elements(
By.CSS_SELECTOR, "a"
)
except:
self._showTrace(f"解析第 {i + 1} 条预约记录时发生未知错误 !")
continue continue
is_wanted = any(wanted_status in status.text for status in info_elements) record_date = record["date"]
# process time element to get the date string record_time = record["time"]
time_str = time_element.text.strip() status = record["info"]["status"]
today = datetime.now().date() location = record["info"]["location"]
if "明天" in time_str: if record_date == "" or record_time == {"begin": "", "end": ""}:
target_date = today + timedelta(days=1) continue
date_str = target_date.strftime("%Y-%m-%d") is_wanted = (status == wanted_status)
elif "今天" in time_str:
target_date = today
date_str = target_date.strftime("%Y-%m-%d")
elif "昨天" in time_str:
target_date = today - timedelta(days=1)
date_str = target_date.strftime("%Y-%m-%d")
else:
date_match = re.search(r'(\d{4}-\d{1,2}-\d{1,2})', time_str)
if date_match:
date_str = date_match.group(1)
else:
self._showTrace(f"无法解析第 {i + 1} 条预约记录的日期 ! 该记录的时间为 {time_str}")
continue
# reservation is later than the given date, check the next one # reservation is later than the given date, check the next one
if datetime.strptime(date_str, "%Y-%m-%d").date() > date_obj: if datetime.strptime(record_date, "%Y-%m-%d").date() > date_obj:
continue continue
# reservation is earlier than the given date, can reserve # reservation is earlier than the given date, can reserve
if datetime.strptime(date_str, "%Y-%m-%d").date() < date_obj: if datetime.strptime(record_date, "%Y-%m-%d").date() < date_obj:
return None return None
# query the wanted status # query the wanted status
if is_wanted: if is_wanted:
self._showTrace(f"寻找到第 {i + 1} 条预约记录, 状态为 {wanted_status}") self._showTrace(
time_match = re.search(r"(\d{1,2}:\d{2}) -- (\d{1,2}:\d{2})", time_str) f"寻找到用户第 {i + 1} 条状态为 {wanted_status} 的预约记录, "
if time_match is None: f"详细信息: {record_date} {record_time['begin']} - {record_time['end']} {location}"
self._showTrace(f"无法解析第 {i + 1} 条预约记录的时间 ! 该记录的时间为 {time_str}") )
continue
return { return {
"index": i, "index": i,
"date": date_str, "date": record_date,
"time_str": time_match.group(0), "time": record_time,
"status": wanted_status "status": wanted_status
} }
checked_count = len(reservations) - 1 checked_count = len(reservations)
# load new reservations if still not sure # load new reservations if still not sure
try: try:
more_btn = self.__driver.find_element(By.ID, "moreBtn") more_btn = self.__driver.find_element(By.ID, "moreBtn")
@@ -142,7 +202,7 @@ class LibChecker(LibOperator):
self.__driver.execute_script("arguments[0].scrollIntoView(true);", more_btn) self.__driver.execute_script("arguments[0].scrollIntoView(true);", more_btn)
self.__driver.execute_script("arguments[0].click();", more_btn) self.__driver.execute_script("arguments[0].click();", more_btn)
else: else:
self._showTrace("用户无法加载更多预约记录") self._showTrace("用户无法加载更多预约记录")
break break
except: except:
self._showTrace("加载更多预约记录失败 !") self._showTrace("加载更多预约记录失败 !")
@@ -159,10 +219,11 @@ class LibChecker(LibOperator):
# then can reserve # then can reserve
if self.__getReserveRecord(date, "已预约") is None: if self.__getReserveRecord(date, "已预约") is None:
if self.__getReserveRecord(date, "使用中") is None: if self.__getReserveRecord(date, "使用中") is None:
self._showTrace(f"用户在日期 {date} 可以预约") self._showTrace(f"用户在 {date} 可以预约")
return True return True
self._showTrace(f"用户在日期 {date} 有使用中的预约, 无法预约") self._showTrace(f"用户在 {date} 有使用中的预约, 无法预约")
self._showTrace(f"用户在日期 {date} 已存在有效预约, 无法预约") return False
self._showTrace(f"用户在 {date} 已存在有效预约, 无法预约")
return False return False
@@ -174,34 +235,30 @@ class LibChecker(LibOperator):
# have a reserved record in the given date # have a reserved record in the given date
record = self.__getReserveRecord(date, "已预约") record = self.__getReserveRecord(date, "已预约")
if record is not None: if record is not None:
time_match = re.search(r"(\d{1,2}:\d{2})", record["time_str"]) begin_time = record["time"]["begin"]
if time_match: begin_time = datetime.strptime(f"{date} {begin_time}", "%Y-%m-%d %H:%M")
begin_time = time_match.group(0) time_diff = datetime.now() - begin_time
begin_time = datetime.strptime(f"{date} {begin_time}", "%Y-%m-%d %H:%M") time_diff_seconds = time_diff.total_seconds()
time_diff = datetime.now() - begin_time # before 30 minutes, cant checkin
time_diff_seconds = time_diff.total_seconds() if time_diff_seconds < -30*60:
# before 30 minutes, cant checkin self._showTrace(
if time_diff_seconds < -30*60: f"用户在 {date} 的预约开始时间为 {begin_time}, "
self._showTrace( f"距离当前时间还有 {abs(time_diff_seconds)/60:.2f} 分钟, 无法签到"
f"用户在日期 {date} 的预约开始时间为 {begin_time}, " )
f"距离当前时间还有 {abs(time_diff_seconds)/60:.2f} 分钟, 无法签到" return False
) # before in 30 minutes, can checkin
return False elif -30*60 <= time_diff_seconds < 0:
# before in 30 minutes, can checkin self._showTrace(
elif -30*60 <= time_diff_seconds < 0: f"用户在 {date} 的预约开始时间为 {begin_time}, "
self._showTrace( f"距离当前时间还有 {abs(time_diff_seconds)/60:.2f} 分钟, 可以签到"
f"用户在日期 {date} 的预约开始时间为 {begin_time}, " )
f"距离当前时间还有 {abs(time_diff_seconds)/60:.2f} 分钟, 可以签到" return True
) # past less than 30 minutes, can checkin
return True elif 0 <= time_diff_seconds < 30*60:
# past less than 30 minutes, can checkin self._showTrace(
elif 0 <= time_diff_seconds < 30*60: f"用户在 {date} 的预约开始时间为 {begin_time}, "
self._showTrace( f"当前时间已经 {abs(time_diff_seconds)/60:.2f} 分钟, 可以签到"
f"用户在日期 {date} 的预约开始时间为 {begin_time}, " )
f"当前时间已经 {abs(time_diff_seconds)/60:.2f} 分钟, 可以签到" return True
) self._showTrace(f"用户在 {date} 有没有有效预约记录, 无法签到")
return True
else:
self._showTrace(f"用户在日期 {date} 的预约时间格式错误, 无法签到")
self._showTrace(f"用户在日期 {date} 有没有有效预约记录, 无法签到")
return False return False
+1 -1
View File
@@ -34,7 +34,7 @@ class LibCheckout(LibOperator):
def _waitResponseLoad( def _waitResponseLoad(
self, self
) -> bool: ) -> bool:
pass pass
+8 -8
View File
@@ -36,7 +36,7 @@ class LibLogin(LibOperator):
def _waitResponseLoad( def _waitResponseLoad(
self, self
) -> bool: ) -> bool:
# wait to verify login success # wait to verify login success
@@ -59,7 +59,7 @@ class LibLogin(LibOperator):
def __fillLogInElements( def __fillLogInElements(
self, self,
username: str, username: str,
password: str, password: str
) -> bool: ) -> bool:
# ensure elements presence and fill them # ensure elements presence and fill them
@@ -77,7 +77,7 @@ class LibLogin(LibOperator):
def __autoRecognizeCaptcha( def __autoRecognizeCaptcha(
self, self
) -> str: ) -> str:
# auto recognize captcha # auto recognize captcha
@@ -99,7 +99,7 @@ class LibLogin(LibOperator):
def __manualRecognizeCaptcha( def __manualRecognizeCaptcha(
self, self
) -> str: ) -> str:
# manual recognize captcha # manual recognize captcha
@@ -117,7 +117,7 @@ class LibLogin(LibOperator):
def __refreshCaptcha( def __refreshCaptcha(
self, self
): ):
# refresh captcha # refresh captcha
@@ -136,7 +136,7 @@ class LibLogin(LibOperator):
def __solveCaptcha( def __solveCaptcha(
self, self,
auto_captcha: bool = True, auto_captcha: bool = True
) -> str: ) -> str:
max_attempts = 5 max_attempts = 5
@@ -155,7 +155,7 @@ class LibLogin(LibOperator):
def __fillCaptchaElement( def __fillCaptchaElement(
self, self,
captcha_text: str, captcha_text: str
) -> bool: ) -> bool:
try: try:
@@ -174,7 +174,7 @@ class LibLogin(LibOperator):
username: str, username: str,
password: str, password: str,
max_attempts: int = 5, max_attempts: int = 5,
auto_captcha: bool = True, auto_captcha: bool = True
) -> bool: ) -> bool:
if self.__driver is None: if self.__driver is None:
+3 -3
View File
@@ -22,7 +22,7 @@ class LibLogout(LibOperator):
self, self,
input_queue: queue.Queue, input_queue: queue.Queue,
output_queue: queue.Queue, output_queue: queue.Queue,
driver, driver
): ):
super().__init__(input_queue, output_queue) super().__init__(input_queue, output_queue)
@@ -31,7 +31,7 @@ class LibLogout(LibOperator):
def _waitResponseLoad( def _waitResponseLoad(
self, self
) -> bool: ) -> bool:
return True return True
@@ -39,7 +39,7 @@ class LibLogout(LibOperator):
def logout( def logout(
self, self,
username: str, username: str
) -> bool: ) -> bool:
if self.__driver is None: if self.__driver is None:
+2 -2
View File
@@ -17,14 +17,14 @@ class LibOperator(MsgBase):
def __init__( def __init__(
self, self,
input_queue: queue.Queue, input_queue: queue.Queue,
output_queue: queue.Queue, output_queue: queue.Queue
): ):
super().__init__(input_queue, output_queue) super().__init__(input_queue, output_queue)
def _waitResponseLoad( def _waitResponseLoad(
self, self
) -> bool: ) -> bool:
pass pass
+1 -1
View File
@@ -28,7 +28,7 @@ class LibRenew(LibOperator):
def _waitResponseLoad( def _waitResponseLoad(
self, self
) -> bool: ) -> bool:
pass pass
+1 -1
View File
@@ -265,7 +265,7 @@ class LibReserve(LibOperator):
trigger_locator: tuple, trigger_locator: tuple,
fail_msg: str, fail_msg: str,
success_msg: str, success_msg: str,
option_locator: tuple = None, option_locator: tuple = None
) -> bool: ) -> bool:
try: try:
+3 -3
View File
@@ -16,7 +16,7 @@ class MsgBase:
def __init__( def __init__(
self, self,
input_queue: queue.Queue, input_queue: queue.Queue,
output_queue: queue.Queue, output_queue: queue.Queue
): ):
self._class_name = self.__class__.__name__ self._class_name = self.__class__.__name__
@@ -43,7 +43,7 @@ class MsgBase:
def _waitMsg( def _waitMsg(
self, self,
timeout: float = 1.0, timeout: float = 1.0
) -> str: ) -> str:
try: try:
@@ -55,7 +55,7 @@ class MsgBase:
def _inputMsg( def _inputMsg(
self, self,
timeout: float = 1.0, timeout: float = 1.0
) -> bool: ) -> bool:
try: try:
+1 -1
View File
@@ -569,7 +569,7 @@ class ALConfigWidget(QWidget, Ui_ALConfigWidget):
@Slot() @Slot()
def onFloorComboBoxCurrentIndexChanged( def onFloorComboBoxCurrentIndexChanged(
self, self
): ):
floor = self.FloorComboBox.currentText() floor = self.FloorComboBox.currentText()
+2 -2
View File
@@ -1127,13 +1127,13 @@
<widget class="QLineEdit" name="BrowseBrowserDriverEdit"> <widget class="QLineEdit" name="BrowseBrowserDriverEdit">
<property name="minimumSize"> <property name="minimumSize">
<size> <size>
<width>265</width> <width>250</width>
<height>25</height> <height>25</height>
</size> </size>
</property> </property>
<property name="maximumSize"> <property name="maximumSize">
<size> <size>
<width>260</width> <width>300</width>
<height>25</height> <height>25</height>
</size> </size>
</property> </property>
+9 -9
View File
@@ -163,7 +163,7 @@ class ALMainWindow(QMainWindow, Ui_ALMainWindow):
def closeEvent( def closeEvent(
self, self,
event: QCloseEvent, event: QCloseEvent
): ):
if self.__timer and self.__timer.isActive(): if self.__timer and self.__timer.isActive():
@@ -175,7 +175,7 @@ class ALMainWindow(QMainWindow, Ui_ALMainWindow):
def appendToTextEdit( def appendToTextEdit(
self, self,
text: str, text: str
): ):
cursor = self.MessageIOTextEdit.textCursor() cursor = self.MessageIOTextEdit.textCursor()
@@ -200,7 +200,7 @@ class ALMainWindow(QMainWindow, Ui_ALMainWindow):
self, self,
config_button_enabled: bool, config_button_enabled: bool,
start_button_enabled: bool, start_button_enabled: bool,
stop_button_enabled: bool, stop_button_enabled: bool
): ):
self.ConfigButton.setEnabled(config_button_enabled) self.ConfigButton.setEnabled(config_button_enabled)
@@ -210,7 +210,7 @@ class ALMainWindow(QMainWindow, Ui_ALMainWindow):
@Slot() @Slot()
def showMsg( def showMsg(
self, self,
msg: str, msg: str
): ):
self.appendToTextEdit(f"[{self.__class_name:<12}] >>> : {msg}") self.appendToTextEdit(f"[{self.__class_name:<12}] >>> : {msg}")
@@ -218,7 +218,7 @@ class ALMainWindow(QMainWindow, Ui_ALMainWindow):
@Slot() @Slot()
def showTrace( def showTrace(
self, self,
msg: str, msg: str
): ):
timestamp = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) timestamp = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
@@ -226,7 +226,7 @@ class ALMainWindow(QMainWindow, Ui_ALMainWindow):
@Slot() @Slot()
def pollMsgQueue( def pollMsgQueue(
self, self
): ):
try: try:
@@ -239,7 +239,7 @@ class ALMainWindow(QMainWindow, Ui_ALMainWindow):
@Slot(dict) @Slot(dict)
def onConfigWidgetClosed( def onConfigWidgetClosed(
self, self,
config_paths: dict, config_paths: dict
): ):
self.__alConfigWidget = None self.__alConfigWidget = None
@@ -250,7 +250,7 @@ class ALMainWindow(QMainWindow, Ui_ALMainWindow):
@Slot() @Slot()
def onConfigButtonClicked( def onConfigButtonClicked(
self, self
): ):
if self.__alConfigWidget is None: if self.__alConfigWidget is None:
@@ -268,7 +268,7 @@ class ALMainWindow(QMainWindow, Ui_ALMainWindow):
@Slot() @Slot()
def onStartButtonClicked( def onStartButtonClicked(
self, self
): ):
self.setControlButtons(False, False, True) self.setControlButtons(False, False, True)