1
import requests
2
import json
3
from datetime import datetime
4
import pytz
5
IC_COOKIE = ""
6
def timestamp_to_gmt8_time_of_day(timestamp_ms):
7
    """
8
    将毫秒级时间戳转化为 GMT+08:00 时区一天里的几点。
9

10
    Args:
11
        timestamp_ms: 毫秒级时间戳。
12

13
    Returns:
14
        一个字符串，表示 GMT+08:00 时区一天里的几点，格式为 "HH:MM:SS"。
15
        如果输入不是数字或无法转换，则返回 None。
16
    """
17
    try:
18
        # 将毫秒转换为秒
19
        timestamp_sec = timestamp_ms / 1000.0
20
        # 创建 UTC 时间的 datetime 对象
21
        utc_datetime = datetime.utcfromtimestamp(timestamp_sec)
22
        # 获取 GMT+08:00 时区
23
        gmt8_tz = pytz.timezone('Etc/GMT-8')  # 注意：pytz 中 GMT+08:00 可能表示为 'Etc/GMT-8' 或 'Asia/Shanghai' 等
24
        # 将 UTC 时间转换为 GMT+08:00 时间
25
        gmt8_datetime = utc_datetime.replace(tzinfo=pytz.utc).astimezone(gmt8_tz)
26
        # 格式化为一天里的时间
27
        time_of_day = gmt8_datetime.strftime('%H:%M:%S')
28
        return time_of_day
29
    except (TypeError, ValueError):
30
        return None
31

32
def fetch_room_reservations(date,kind):
33
    """
34
    Fetches room reservation data from the given URL and extracts occupied times.
35

36
    Args:
37
        url (str): The URL to fetch the reservation data from.
38

39
    Returns:
40
        dict: A dictionary where keys are room names and values are lists of
41
              occupied time slots (start and end times in milliseconds).
42
    """
43
    kindid = {
44
        "oldlib_quietroom":101766982,
45
        "oldlib_groupstudyroom":101420505,
46
        "newlib_quietroom":101505911,
47
        "newlib_groupstudyroom":100462149,
48
    }[kind]
49
    url = f"https://librooms.cuhk.edu.cn/ic-web/reserve?sysKind=1&resvDates={date}&page=1&pageSize=30&kindIds={kindid}&labId="
50
    try:
51
        headers={'Cookie':'ic-cookie=IC_COOKIE'}
52
        response = requests.get(url,headers=headers)
53
        response.raise_for_status()  # Raise an exception for bad status codes
54
        data = response.json()
55
        if data.get("code") == 0 and data.get("message") == "查询成功":
56
            reservations = {}
57
            for room in data['data']:
58
                room_name = room['roomSn']
59
                for reserve in room['resvInfo']:
60
                    if not reservations.get(room_name):
61
                        reservations[room_name]=[]
62
                    reservations[room_name].append({'startTime':timestamp_to_gmt8_time_of_day(reserve['startTime']),'endTime':timestamp_to_gmt8_time_of_day(reserve['endTime'])})
63
            return reservations
64
        else:
65
            print(f"Error: {data.get('message')}")
66
            return None
67
    except requests.exceptions.RequestException as e:
68
        print(f"Request error: {e}")
69
        return None
70
    except json.JSONDecodeError:
71
        print("Error decoding JSON response.")
72
        return None