fix get_bilibili_live_rooms

_playwright/bilibili.py

@@ -70,26 +70,22 @@ async def live(page):
     for room_id in list(bilibili_live_rooms.keys()):
         await page.goto(f"{bilibili_live_url}/{str(room_id)}")
         # 输入框选择 xpath 绝对定位
-        input_box = await page.locator(
+        input_box = page.locator(
             "xpath=/html/body/div[3]/main/div[1]/section[1]/div[3]/div[12]/div/div[2]/div[2]/textarea",
         )
         # 输入内容
         await input_box.fill("1")
         # 发送按钮
-        send_button = await page.locator(
+        send_button = page.locator(
             "xpath=/html/body/div[3]/main/div[1]/section[1]/div[3]/div[12]/div/div[3]/div/button/span"
         )
         if await send_button.is_visible():
             # 点击发送按钮
             await send_button.click()
-            room_msg = (
-                f"直播间：【{bilibili_live_rooms[room_id]}】打卡成功，打卡时间：【{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}】"
-            )
+            room_msg = f"直播间：【{bilibili_live_rooms[room_id]}】打卡成功，打卡时间：【{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}】"
             bilibili_logger.info(room_msg)
         else:
-            room_msg = (
-                f"直播间：【{bilibili_live_rooms[room_id]}】打卡失败，打卡时间：【{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}】"
-            )
+            room_msg = f"直播间：【{bilibili_live_rooms[room_id]}】打卡失败，打卡时间：【{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}】"
             bilibili_logger.error(room_msg)
         msg.append(room_msg)
 

utils.py

@@ -97,7 +97,9 @@ def get_pi_ip(host: str) -> str:
 
 
 # gotify 推送
-def push_msg(title: str = "无标题", message: str = "无内容", img_url: str = gotify_img) -> dict:
+def push_msg(
+    title: str = "无标题", message: str = "无内容", img_url: str = gotify_img
+) -> dict:
     gotify_ip = get_pi_ip(gotify_host)
     url = f"http://{gotify_ip}:{gotify_port}/message?token={gotify_token}"
     data = {
@@ -117,7 +119,9 @@ async def get_bilibili_live_rooms_from_pi() -> dict:
     sql = f"SELECT room_id, username FROM live_rooms"
     pool = await get_aiomysql_instance()
     data = await pool.query(sql)
-    bilibili_live_rooms = data[0]
+    bilibili_live_rooms = {}
+    for d in data:
+        bilibili_live_rooms[d["room_id"]] = d["username"]
     return bilibili_live_rooms