feeds.py
1.14 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
"""首页 Feed 列表，对应 Go xiaohongshu/feeds.go。"""
from __future__ import annotations
import json
import logging
import time
from .cdp import Page
from .errors import NoFeedsError
from .types import Feed
from .urls import HOME_URL
logger = logging.getLogger(__name__)
# 从 __INITIAL_STATE__ 提取 feeds 的 JS
_EXTRACT_FEEDS_JS = """
(() => {
if (window.__INITIAL_STATE__ &&
window.__INITIAL_STATE__.feed &&
window.__INITIAL_STATE__.feed.feeds) {
const feeds = window.__INITIAL_STATE__.feed.feeds;
const feedsData = feeds.value !== undefined ? feeds.value : feeds._value;
if (feedsData) {
return JSON.stringify(feedsData);
}
}
return "";
})()
"""
def list_feeds(page: Page) -> list[Feed]:
"""获取首页 Feed 列表。
Raises:
NoFeedsError: 没有捕获到 feeds 数据。
"""
page.navigate(HOME_URL)
page.wait_for_load()
page.wait_dom_stable()
time.sleep(1)
result = page.evaluate(_EXTRACT_FEEDS_JS)
if not result:
raise NoFeedsError()
feeds_data = json.loads(result)
return [Feed.from_dict(f) for f in feeds_data]