# -*- coding: utf-8 -*- """ Copyright (c) 2025 KenanZhu. All rights reserved. This software is provided "as is", without any warranty of any kind. You may use, modify, and distribute this file under the terms of the MIT License. See the LICENSE file for details. """ import re import time import queue from datetime import datetime, timedelta from selenium.webdriver.common.by import By from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC from LibOperator import LibOperator class LibReserve(LibOperator): def __init__( self, input_queue: queue.Queue, output_queue: queue.Queue, driver ): super().__init__(input_queue, output_queue) self.__driver = driver # library floor and room mapping in website self.__floor_map = { "2": "二层", "3": "三层", "4": "四层", "5": "五层" } self.__room_map = { "1": "二层内环", "2": "二层外环", "3": "三层内环", "4": "三层外环", "5": "四层内环", "6": "四层外环", "7": "四层期刊区", "8": "五层考研" } def _waitResponseLoad( self, ) -> bool: try: WebDriverWait(self.__driver, 5).until( EC.presence_of_element_located((By.CLASS_NAME, "layoutSeat")) ) title_elements = [] # reserve failed without title elements, so we need to try try: WebDriverWait(self.__driver, 1).until( EC.presence_of_element_located((By.CSS_SELECTOR, ".layoutSeat dt")) ) title_elements = self.__driver.find_elements( By.CSS_SELECTOR, ".layoutSeat dt" ) except: pass content_elements = self.__driver.find_elements( By.CSS_SELECTOR, ".layoutSeat dd" ) if not content_elements: self._showTrace("未找到预约结果") raise title = title_elements[0].text if title_elements else "" contents = [element.text for element in content_elements if element.text.strip()] for message in contents: if "预约失败" in message or "已有1个有效预约" in message: self._showTrace(f"预约失败 - {"".join(contents)}") raise if "预定好了" in title or "预约成功" in title or "操作成功" in title: if len(contents) >= 6: date_val = contents[1].split(" ： ")[1].strip() if " ： " in contents[1] else contents[1].strip() time_val = contents[2].split(" ： ")[1].strip() if " ： " in contents[2] else contents[2].strip() seat_val = contents[3].split(" ： ")[1].strip() if " ： " in contents[3] else contents[3].strip() checkin_val = contents[5].strip() self._showTrace(f"\n"\ f" 预约成功 !\n"\ f" 预约日期: {date_val}, \n"\ f" 预约时间: {time_val}, \n"\ f" 预约座位: {seat_val}, \n"\ f" 签到时间: {checkin_val}") else: self._showTrace(f"\n"\ f" 预约成功 !\n"\ f" 未找获取到详细信息") return True except: self._showTrace(f"预约结果加载失败 !") return False @staticmethod def __timeToMins( time_str: str ) -> int: hour, minute = map(int, time_str.split(":")) return hour*60 + minute @staticmethod def __minsToTime( mins: int ) -> str: hour, minute = divmod(mins, 60) return f"{hour:02d}:{minute:02d}" def __checkReserveInfo( self, reserve_info: dict ) -> bool: try: # check the required information # reserve_info["place"] if reserve_info.get("floor") is None: raise ValueError("未指定楼层") if reserve_info["floor"] not in self.__floor_map: raise ValueError(f"楼层 '{reserve_info['floor']}' 不存在") if reserve_info.get("room") is None: raise ValueError("未指定房间") if reserve_info["room"] not in self.__room_map: raise ValueError(f"房间 '{reserve_info['room']}' 不存在") if reserve_info.get("seat_id") is None: raise ValueError("未指定座位") except ValueError as e: self._showTrace( f"预约信息错误 ! : {e}, "\ f"由于缺少必要的预约信息， 无法开始预约流程, 请检查预约信息是否完整" ) return False # check and try to fix the time errors cur_time_str = time.strftime("%Y-%m-%d %H:%M", time.localtime()) cur_date, curr_time = cur_time_str.split() if not reserve_info.get("date"): reserve_info["date"] = cur_date self._showTrace(f"预约日期未指定, 自动设置为当前日期: {cur_date}") else: if reserve_info["date"] < cur_date: self._showTrace( f"预约日期错误 ! :"\ f"{reserve_info['date']} 早于当前日期 {cur_date}, 自动设置为当前日期" ) reserve_info["date"] = cur_date # check the begin time begin_time = reserve_info.get("begin_time") if not begin_time: reserve_info["begin_time"] = { "time": curr_time, "max_diff": 30, "prefer_early": True } self._showTrace(f"开始时间未指定, 自动设置为当前时间: {curr_time}, 最大时间差为 30 分钟, 优先选择更早预约时间") else: begin_time = reserve_info["begin_time"] if "time" not in begin_time: begin_time["time"] = curr_time self._showTrace(f"开始时间未指定, 自动设置为当前时间: {curr_time}") if "max_diff" not in begin_time: begin_time["max_diff"] = 30 self._showTrace(f"最大时间差未指定, 自动设置为 30 分钟") if "prefer_early" not in begin_time: begin_time["prefer_early"] = True self._showTrace(f"是否优先选择更早预约时间未指定, 自动设置为 True") expect_duration = reserve_info.get("expect_duration") if not expect_duration: reserve_info["expect_duration"] = 4 expect_duration = 4 self._showTrace("预约持续时间未指定, 使用默认时长为 4 小时") if not reserve_info.get("satisfy_duration"): reserve_info["satisfy_duration"] = True self._showTrace("预约满足时长要求未指定, 默认满足") # check the end time if not reserve_info.get("end_time"): begin_mins = self.__timeToMins(reserve_info["begin_time"]["time"]) end_mins = begin_mins + reserve_info["expect_duration"] * 60 end_time_str = self.__minsToTime(end_mins) reserve_info["end_time"] = { "time": end_time_str, "max_diff": 30, "prefer_early": False } self._showTrace(f"结束时间未指定, 自动设置为开始时间加上期望时长: {end_time_str}, 最大时间差为 30 分钟, 优先选择较晚预约时间") else: end_time = reserve_info["end_time"] if "time" not in end_time: begin_mins = self.__timeToMins(reserve_info["begin_time"]["time"]) end_mins = begin_mins + reserve_info["expect_duration"] * 60 end_time["time"] = self.__minsToTime(end_mins) self._showTrace(f"结束时间未指定, 自动设置为开始时间加上期望时长: {end_time['time']}") if "max_diff" not in end_time: end_time["max_diff"] = 30 self._showTrace(f"最大时间差未指定, 自动设置为 30 分钟") if "prefer_early" not in end_time: end_time["prefer_early"] = False self._showTrace(f"是否优先选择较早预约时间未指定, 自动设置为 False") # check the reserve time boundary and fix the errors # # get time string for message show begin_time_str = reserve_info["begin_time"]["time"] end_time_str = reserve_info["end_time"]["time"] # minute time for check and fix them begin_mins = self.__timeToMins(begin_time_str) end_mins = self.__timeToMins(end_time_str) # ensure begin time is not later than end time if begin_mins > end_mins: reserve_info["begin_time"]["time"], reserve_info["end_time"]["time"] = end_time_str, begin_time_str reserve_info["begin_time"]["prefer_early"], reserve_info["end_time"]["prefer_early"] = \ reserve_info["end_time"]["prefer_early"], reserve_info["begin_time"]["prefer_early"] self._showTrace("预约开始时间晚于预约结束时间，自动调换开始时间和结束时间") # update the begin_mins and end_mins after swap begin_time_str, end_time_str = end_time_str, begin_time_str begin_mins, end_mins = end_mins, begin_mins # ensure end time is not later than 22:30 max_end_mins = self.__timeToMins("22:30") if end_mins > max_end_mins: reserve_info["end_time"]["time"] = "22:30" end_time_str = "22:30" end_mins = max_end_mins self._showTrace("预约结束时间超过 22:30, 自动设置为 22:30") # ensure expect duration is shorter than 8 hours max_duration_mins = 8 * 60 duration_mins = end_mins - begin_mins if duration_mins > max_duration_mins: new_end_mins = begin_mins + max_duration_mins reserve_info["end_time"]["time"] = self.__minsToTime(new_end_mins) self._showTrace("预约持续时间超过8小时, 自动设置为 8 小时") self._showTrace( f"预约信息检查完成，准备预约 " f"{reserve_info['date']} " f"{reserve_info['begin_time']["time"]} - " f"{reserve_info['end_time']["time"]} " f"图书馆 " f"{self.__floor_map[reserve_info['floor']]} " f"{self.__room_map[reserve_info['room']]} " f"的座位 {reserve_info['seat_id']}" ) return True def __clickElement( self, trigger_locator: tuple, fail_msg: str, success_msg: str, option_locator: tuple = None, ) -> bool: try: # click the trigger element WebDriverWait(self.__driver, 5).until( EC.element_to_be_clickable(trigger_locator) ).click() if option_locator: # select the option element if specified WebDriverWait(self.__driver, 5).until( EC.element_to_be_clickable(option_locator) ).click() self._showTrace(success_msg) return True except: self._showTrace(fail_msg) return False def __selectDate( self, date_str: str ) -> bool: return self.__clickElement( trigger_locator=(By.ID, "onDate_select"), option_locator=(By.XPATH, f"//p[@id='options_onDate']/a[@value='{date_str}']"), success_msg=f"日期 {date_str} 选择成功 !", fail_msg=f"选择日期失败 ! : {date_str} 不可用" ) def __selectPlace( self, place: str ) -> bool: actual_place = "1" if place == "图书馆" else "1" return self.__clickElement( trigger_locator=(By.ID, "display_building"), option_locator=(By.XPATH, f"//p[@id='options_building']/a[@value='{actual_place}']"), success_msg=f"预约场所 {place} 选择成功 !", fail_msg=f"选择预约场所失败 ! : {place} 不可用" ) def __selectFloor( self, floor: str ) -> bool: display_floor = self.__floor_map.get(floor) return self.__clickElement( trigger_locator=(By.ID, "floor_select"), option_locator=(By.XPATH, f"//p[@id='options_floor']/a[@value='{floor}']"), success_msg=f"楼层 {display_floor} 选择成功 !", fail_msg=f"选择楼层失败 ! : {display_floor} 不可用" ) def __selectRoom( self, room: str ) -> bool: display_room = self.__room_map.get(room) return self.__clickElement( trigger_locator=(By.ID, f"room_{room}"), option_locator=None, success_msg=f"房间 {display_room} 选择成功 !", fail_msg=f"选择房间失败 ! : {display_room} 不可用" ) def __selectSeat( self, seat_id: str ) -> bool: try: # wait fot seat layout element to load WebDriverWait(self.__driver, 5).until( EC.presence_of_element_located((By.ID, "seatLayout")) ) all_seats = self.__driver.find_elements( By.CSS_SELECTOR, "li[id^='seat_']" ) seat_id_upper = seat_id.lstrip('0').upper() for seat in all_seats: if not seat_id_upper == seat.text.lstrip('0'): continue seat_link = seat.find_element(By.TAG_NAME, "a") WebDriverWait(self.__driver, 5).until( EC.element_to_be_clickable(seat_link) ) seat_link.click() seat_status = seat_link.get_attribute("title") self._showTrace(f"座位 {seat_id} 选择成功 ! : 当前状态 - '{seat_status}'") return True self._showTrace(f"座位 {seat_id} 在该楼层区域中不存在, 请检查座位号是否正确") except: self._showTrace(f"座位选择失败 !") return False def __selectNearestTime( self, time_id: str, time_type: str, target_time: int, max_time_diff: int = 30, prefer_earlier: bool = True ) -> int: try: all_time_opts = self.__driver.find_elements( By.CSS_SELECTOR, f"#{time_id} ul li a" ) free_times = [] best_time_diff = max_time_diff best_actual_diff = None best_time_opt = None for time_opt in all_time_opts: time_attr = time_opt.get_attribute("time") if time_attr == "now": now = datetime.now() time_val = int(now.hour*60 + now.minute) elif time_attr and time_attr.isdigit(): time_val = int(time_attr) else: continue free_times.append(self.__minsToTime(time_val)) actual_diff = time_val - target_time abs_diff = abs(actual_diff) if abs_diff < best_time_diff or ( abs_diff == best_time_diff and ( # prefer earlier time (prefer_earlier and actual_diff <= 0) or # prefer later time (not prefer_earlier and actual_diff >= 0) ) ): best_time_diff = abs_diff best_actual_diff = actual_diff best_time_opt = time_opt if best_time_opt is not None: best_time_opt.click() abs_time_diff = abs(best_actual_diff) if best_actual_diff < 0: time_relation = f"早了 {abs_time_diff} 分钟" elif best_actual_diff > 0: time_relation = f"晚了 {abs_time_diff} 分钟" else: time_relation = f"正好等于 {time_type}" target_time += best_actual_diff self._showTrace( f"选择距离期望 {time_type} 最近的 {best_time_opt.text}, "\ f"与期望 {time_type} 相比 {time_relation}" ) return target_time self._showTrace( f"无法选择最近的 {time_type} {self.__minsToTime(target_time)}, "\ f"所有可选时间与目标时间相差都超过 {max_time_diff} 分钟" ) self._showTrace(f"当前可供预约的 {time_type} 有: {free_times}") return -1 except: self._showTrace(f"{time_type} {self.__minsToTime(target_time)} 选择失败 !") return -1 def __selectSeatTime( self, begin_time: dict, end_time: dict, expct_duration: int = 4, satisfy_duration: bool = True ) -> bool: expect_begin_time = actual_begin_time = begin_time["time"] expect_end_time = actual_end_time = end_time["time"] expect_begin_mins = self.__timeToMins(expect_begin_time) expect_end_mins = self.__timeToMins(expect_end_time) # select the begin time if self.__selectNearestTime( time_id="startTime", # dont change into begin, this is the element in the page time_type="开始时间", target_time=expect_begin_mins, max_time_diff=begin_time["max_diff"], prefer_earlier=begin_time["prefer_early"] ) == -1: return False else: actual_begin_time = self.__minsToTime(expect_begin_mins) # if 'satisfy_duration' is True. # select the end time based on the begin time # (because it may be changed under the 'max time diff' strategy) and expect duration. if satisfy_duration: expect_end_mins = int(expect_begin_mins + expct_duration*60) self._showTrace( f"需要满足期望预约持续时间: {expct_duration} 小时, "\ f"根据开始时间 {actual_begin_time} 计算结束时间: {self.__minsToTime(expect_end_mins)}" ) # select the end time if self.__selectNearestTime( time_id="endTime", time_type="结束时间", target_time=expect_end_mins, max_time_diff=end_time["max_diff"], prefer_earlier=end_time["prefer_early"] ) == -1: return False else: actual_end_time = self.__minsToTime(expect_end_mins) self._showTrace( f"期望预约时间段: {expect_begin_time} - {expect_end_time}, " f"实际预约时间段: {actual_begin_time} - {actual_end_time}" ) return True def canReserve( self, date: str ) -> bool: if date is None: self._showTrace("日期未指定, 无法检查预约状态") return True else: self._showTrace(f"正在检查用户在日期 {date} 是否可预约......") date_obj = datetime.strptime(date, "%Y-%m-%d").date() try: # we need to navigate to the history page to check if we can reserve WebDriverWait(self.__driver, 5).until( EC.element_to_be_clickable((By.XPATH, "//a[@href='/history?type=SEAT']")) ).click() WebDriverWait(self.__driver, 2).until( EC.presence_of_element_located((By.CLASS_NAME, "myReserveList")) ) except: self._showTrace("加载预约记录页面失败 !") return False checked_count = 0 max_attemots = 3 # we only check (3*4=)12 reservations for _ in range(max_attemots): try: # check if there's any reservation on the date reservations = self.__driver.find_elements( By.CSS_SELECTOR, ".myReserveList dl" ) except: self._showTrace("加载预约记录失败 !") return False for i in range(checked_count, len(reservations) - 1): # the last one is load button reservation = reservations[i] try: time_element = reservation.find_element( By.CSS_SELECTOR, "dt" ) status_elements = reservation.find_elements( By.CSS_SELECTOR, "a" ) is_reserved = any("已预约" in status.text for status in status_elements) # 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: continue # reservation is earlier than the given date, can reserve if datetime.strptime(date_str, "%Y-%m-%d").date() < date_obj: self._showTrace(f"用户在 {date} 可预约") return True # reservation is later than the given date, check the next one elif datetime.strptime(date_str, "%Y-%m-%d").date() > date_obj: continue # compare with the given date if date_str == date and is_reserved: self._showTrace(f"用户在 {date} 已存在有效预约, 无法预约") return False except: self._showTrace(f"解析第 {i + 1} 条预约记录时发生未知错误 !") continue checked_count = len(reservations) - 1 # load new reservations if still not sure try: more_btn = self.__driver.find_element(By.ID, "moreBtn") if more_btn.is_displayed() and more_btn.is_enabled(): self.__driver.execute_script("arguments[0].scrollIntoView(true);", more_btn) self.__driver.execute_script("arguments[0].click();", more_btn) else: break except: break self._showTrace(f"用户在 {date} 可预约") return True def reserve( self, reserve_info: dict ) -> bool: submit_reserve = False reserve_success = False have_hover_on_page = False # reserve info if not self.__checkReserveInfo(reserve_info): return False # map page try: WebDriverWait(self.__driver, 5).until( EC.element_to_be_clickable((By.XPATH, "//a[@href='/map']")) ).click() WebDriverWait(self.__driver, 2).until( EC.presence_of_element_located((By.ID, "seatLayout")) ) except: self._showTrace(f"加载预约选座页面失败 !") return False # date, place, floor if not self.__selectDate(reserve_info["date"]): return False if not self.__selectPlace(reserve_info["place"]): return False if not self.__selectFloor(reserve_info["floor"]): return False # room find try: WebDriverWait(self.__driver, 2).until( EC.element_to_be_clickable((By.ID, "findRoom")) ).click() except: self._showTrace("加载房间/区域失败 !") return False # room if not self.__selectRoom(reserve_info["room"]): return False else: have_hover_on_page = True # seat selections if not self.__selectSeat(reserve_info["seat_id"]): pass elif not self.__selectSeatTime( begin_time=reserve_info["begin_time"], end_time=reserve_info["end_time"], expct_duration=reserve_info["expect_duration"], satisfy_duration=reserve_info["satisfy_duration"] ): pass else: try: WebDriverWait(self.__driver, 2).until( EC.element_to_be_clickable((By.ID, "reserveBtn")) ).click() submit_reserve = True if not self._waitResponseLoad(): raise reserve_success = True except: self._showTrace(f"预约提交失败 !") if not submit_reserve and have_hover_on_page: self.__driver.refresh() return reserve_success