From 0e50d4aafd719087a2e5adbb12ce733e5d9bb9a2 Mon Sep 17 00:00:00 2001
From: AnDaoAi <535420980@qq.com>
Date: Tue, 21 Oct 2025 16:38:46 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=9D=E5=AD=98cookie=E5=89=8D?=
 =?UTF-8?q?=E5=88=9B=E5=BB=BAcookiesFile=E7=9B=AE=E5=BD=95=E4=BB=A5?=
 =?UTF-8?q?=E9=98=B2=E6=AD=A2=E6=96=87=E4=BB=B6=E4=B8=8D=E5=AD=98=E5=9C=A8?=
 =?UTF-8?q?=E9=94=99=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在所有平台的cookie生成函数中添加目录创建逻辑
- 修复cookiesFile目录不存在导致cookie保存失败的问题
- 影响的函数：douyin_cookie_gen、get_tencent_cookie、get_ks_cookie和xiaohongshu_cookie_gen

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 myUtils/login.py | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/myUtils/login.py b/myUtils/login.py
index 3136a95..56f3980 100644
--- a/myUtils/login.py
+++ b/myUtils/login.py
@@ -50,7 +50,10 @@ async def douyin_cookie_gen(id,status_queue):
             return None
         uuid_v1 = uuid.uuid1()
         print(f"UUID v1: {uuid_v1}")
-        await context.storage_state(path=Path(BASE_DIR / "cookiesFile" / f"{uuid_v1}.json"))
+        # 确保cookiesFile目录存在
+        cookies_dir = Path(BASE_DIR / "cookiesFile")
+        cookies_dir.mkdir(exist_ok=True)
+        await context.storage_state(path=cookies_dir / f"{uuid_v1}.json")
         result = await check_cookie(3, f"{uuid_v1}.json")
         if not result:
             status_queue.put("500")
@@ -125,7 +128,10 @@ async def get_tencent_cookie(id,status_queue):
             return None
         uuid_v1 = uuid.uuid1()
         print(f"UUID v1: {uuid_v1}")
-        await context.storage_state(path=Path(BASE_DIR / "cookiesFile" / f"{uuid_v1}.json"))
+        # 确保cookiesFile目录存在
+        cookies_dir = Path(BASE_DIR / "cookiesFile")
+        cookies_dir.mkdir(exist_ok=True)
+        await context.storage_state(path=cookies_dir / f"{uuid_v1}.json")
         result = await check_cookie(2,f"{uuid_v1}.json")
         if not result:
             status_queue.put("500")
@@ -196,7 +202,10 @@ async def get_ks_cookie(id,status_queue):
             return None
         uuid_v1 = uuid.uuid1()
         print(f"UUID v1: {uuid_v1}")
-        await context.storage_state(path=Path(BASE_DIR / "cookiesFile" / f"{uuid_v1}.json"))
+        # 确保cookiesFile目录存在
+        cookies_dir = Path(BASE_DIR / "cookiesFile")
+        cookies_dir.mkdir(exist_ok=True)
+        await context.storage_state(path=cookies_dir / f"{uuid_v1}.json")
         result = await check_cookie(4, f"{uuid_v1}.json")
         if not result:
             status_queue.put("500")
@@ -267,7 +276,10 @@ async def xiaohongshu_cookie_gen(id,status_queue):
             return None
         uuid_v1 = uuid.uuid1()
         print(f"UUID v1: {uuid_v1}")
-        await context.storage_state(path=Path(BASE_DIR / "cookiesFile" / f"{uuid_v1}.json"))
+        # 确保cookiesFile目录存在
+        cookies_dir = Path(BASE_DIR / "cookiesFile")
+        cookies_dir.mkdir(exist_ok=True)
+        await context.storage_state(path=cookies_dir / f"{uuid_v1}.json")
         result = await check_cookie(1, f"{uuid_v1}.json")
         if not result:
             status_queue.put("500")