更新代码

main.py

@@ -2,8 +2,9 @@ from source import get_image_link
 
 
 def example():
+    test_cookie = "abRequestId=25c57ab7-8cbf-5383-b020-08852c1704e1; webBuild=3.4.1; xsecappid=xhs-pc-web; a1=18a033d274338lwsfacj9x5bpf4fznhhc8xrakemj50000250510; webId=93c0636350d85103d93bca88da2959cd; websectiga=2a3d3ea002e7d92b5c9743590ebd24010cf3710ff3af8029153751e41a6af4a3; sec_poison_id=ae1f0190-4d0c-45f5-a75f-68c2d87c5573; web_session=030037a3ed570060b3a43845a6234a2f100ef4; gid=yY08qqfYy0SSyY08qqfJWYTd4qqY1EMi0SVjC6VC2DUi4F28iuIxx0888J282y880JfWY0Di; cache_feeds=[]"
     demo = "https://www.xiaohongshu.com/explore/64a3a5170000000031008914"
-    print(get_image_link(demo))
+    print(get_image_link(demo, cookie=test_cookie))
 
 
 if __name__ == '__main__':

source/Html.py

@@ -10,7 +10,15 @@ HEADERS = {
 }
 
 
-def get_html(url, params=None, proxies=None, timeout=10, **kwargs):
+def get_html(
+        url,
+        cookie=None,
+        params=None,
+        proxies=None,
+        timeout=10,
+        **kwargs):
+    if cookie:
+        update_cookie(cookie)
     response = requests.get(
         url,
         params=params,
@@ -19,3 +27,8 @@ def get_html(url, params=None, proxies=None, timeout=10, **kwargs):
         headers=HEADERS,
         **kwargs)
     return response.text
+
+
+def update_cookie(cookie: str):
+    global HEADERS
+    HEADERS["Cookie"] = cookie

source/__init__.py

@@ -3,9 +3,16 @@ from .Image import generate_url
 from .Image import get_id
 
 
-def get_image_link(url: str, params=None, proxies=None, timeout=10, **kwargs):
+def get_image_link(
+        url: str,
+        cookie=None,
+        params=None,
+        proxies=None,
+        timeout=10,
+        **kwargs):
     html = get_html(
         url,
+        cookie=cookie,
         params=params,
         proxies=proxies,
         timeout=timeout,