Merge branch 'main' into main

CLAUDE.md

@@ -0,0 +1,101 @@
+## Project Overview
+
+This project, `social-auto-upload`, is a powerful automation tool designed to help content creators and operators efficiently publish video content to multiple domestic and international mainstream social media platforms in one click. The project implements video upload, scheduled release and other functions for platforms such as `Douyin`, `Bilibili`, `Xiaohongshu`, `Kuaishou`, `WeChat Channel`, `Baijiahao` and `TikTok`.
+
+The project consists of a Python backend and a Vue.js frontend.
+
+**Backend:**
+
+*   Framework: Flask
+*   Core Functionality:
+    *   Handles file uploads and management.
+    *   Interacts with a SQLite database to store information about files and user accounts.
+    *   Uses `playwright` for browser automation to interact with social media platforms.
+    *   Provides a RESTful API for the frontend to consume.
+    *   Uses Server-Sent Events (SSE) for real-time communication with the frontend during the login process.
+
+**Frontend:**
+
+*   Framework: Vue.js
+*   Build Tool: Vite
+*   UI Library: Element Plus
+*   State Management: Pinia
+*   Routing: Vue Router
+*   Core Functionality:
+    *   Provides a web interface for managing social media accounts, video files, and publishing videos.
+    *   Communicates with the backend via a RESTful API.
+
+**Command-line Interface:**
+
+The project also provides a command-line interface (CLI) for users who prefer to work from the terminal. The CLI supports two main actions:
+
+*   `login`: To log in to a social media platform.
+*   `upload`: To upload a video to a social media platform, with an option to schedule the upload.
+
+## Building and Running
+
+### Backend
+
+1.  **Install dependencies:**
+    ```bash
+    pip install -r requirements.txt
+    ```
+
+2.  **Install Playwright browser drivers:**
+    ```bash
+    playwright install chromium
+    ```
+
+3.  **Initialize the database:**
+    ```bash
+    python db/createTable.py
+    ```
+
+4.  **Run the backend server:**
+    ```bash
+    python sau_backend.py
+    ```
+    The backend server will start on `http://localhost:5409`.
+
+### Frontend
+
+1.  **Navigate to the frontend directory:**
+    ```bash
+    cd sau_frontend
+    ```
+
+2.  **Install dependencies:**
+    ```bash
+    npm install
+    ```
+
+3.  **Run the development server:**
+    ```bash
+    npm run dev
+    ```
+    The frontend development server will start on `http://localhost:5173`.
+
+### Command-line Interface
+
+To use the CLI, you can run the `cli_main.py` script with the appropriate arguments.
+
+**Login:**
+
+```bash
+python cli_main.py <platform> <account_name> login
+```
+
+**Upload:**
+
+```bash
+python cli_main.py <platform> <account_name> upload <video_file> [-pt {0,1}] [-t YYYY-MM-DD HH:MM]
+```
+
+## Development Conventions
+
+*   The backend code is located in the root directory and the `myUtils` and `uploader` directories.
+*   The frontend code is located in the `sau_frontend` directory.
+*   The project uses a SQLite database for data storage. The database file is located at `db/database.db`.
+*   The `conf.example.py` file should be copied to `conf.py` and configured with the appropriate settings.
+*   The `requirements.txt` file lists the Python dependencies.
+*   The `package.json` file in the `sau_frontend` directory lists the frontend dependencies.

myUtils/postVideo.py

@@ -10,7 +10,7 @@ from utils.constant import TencentZoneTypes
 from utils.files_times import generate_schedule_time_next_day
 
 
-def post_video_tencent(title,files,tags,account_file,category=TencentZoneTypes.LIFESTYLE.value,enableTimer=False,videos_per_day = 1, daily_times=None,start_days = 0):
+def post_video_tencent(title,files,tags,account_file,category=TencentZoneTypes.LIFESTYLE.value,enableTimer=False,videos_per_day = 1, daily_times=None,start_days = 0, is_draft=False):
     # 生成文件的完整路径
     account_file = [Path(BASE_DIR / "cookiesFile" / file) for file in account_file]
     files = [Path(BASE_DIR / "videoFile" / file) for file in files]
@@ -25,7 +25,7 @@ def post_video_tencent(title,files,tags,account_file,category=TencentZoneTypes.L
             print(f"视频文件名：{file}")
             print(f"标题：{title}")
             print(f"Hashtag：{tags}")
-            app = TencentVideo(title, str(file), tags, publish_datetimes[index], cookie, category)
+            app = TencentVideo(title, str(file), tags, publish_datetimes[index], cookie, category, is_draft)
             asyncio.run(app.main(), debug=False)
 
 

sau_backend.py

@@ -139,9 +139,10 @@ def upload_save():
         }), 200
 
     except Exception as e:
+        print(f"Upload failed: {e}")
         return jsonify({
             "code": 500,
-            "msg": str("upload failed!"),
+            "msg": f"upload failed: {e}",
             "data": None
         }), 500
 
@@ -157,14 +158,26 @@ def get_all_files():
             cursor.execute("SELECT * FROM file_records")
             rows = cursor.fetchall()
 
-            # 将结果转为字典列表
-            data = [dict(row) for row in rows]
-
-        return jsonify({
-            "code": 200,
-            "msg": "success",
-            "data": data
-        }), 200
+            # 将结果转为字典列表，并提取UUID
+            data = []
+            for row in rows:
+                row_dict = dict(row)
+                # 从 file_path 中提取 UUID (文件名的第一部分，下划线前)
+                if row_dict.get('file_path'):
+                    file_path_parts = row_dict['file_path'].split('_', 1)  # 只分割第一个下划线
+                    if len(file_path_parts) > 0:
+                        row_dict['uuid'] = file_path_parts[0]  # UUID 部分
+                    else:
+                        row_dict['uuid'] = ''
+                else:
+                    row_dict['uuid'] = ''
+                data.append(row_dict)
+
+            return jsonify({
+                "code": 200,
+                "msg": "success",
+                "data": data
+            }), 200
     except Exception as e:
         return jsonify({
             "code": 500,
@@ -173,6 +186,37 @@ def get_all_files():
         }), 500
 
 
+@app.route("/getAccounts", methods=['GET'])
+def getAccounts():
+    """快速获取所有账号信息，不进行cookie验证"""
+    try:
+        with sqlite3.connect(Path(BASE_DIR / "db" / "database.db")) as conn:
+            conn.row_factory = sqlite3.Row
+            cursor = conn.cursor()
+            cursor.execute('''
+            SELECT * FROM user_info''')
+            rows = cursor.fetchall()
+            rows_list = [list(row) for row in rows]
+
+            print("\n📋 当前数据表内容（快速获取）：")
+            for row in rows:
+                print(row)
+
+            return jsonify(
+                {
+                    "code": 200,
+                    "msg": None,
+                    "data": rows_list
+                }), 200
+    except Exception as e:
+        print(f"获取账号列表时出错: {str(e)}")
+        return jsonify({
+            "code": 500,
+            "msg": f"获取账号列表失败: {str(e)}",
+            "data": None
+        }), 500
+
+
 @app.route("/getValidAccounts",methods=['GET'])
 async def getValidAccounts():
     with sqlite3.connect(Path(BASE_DIR / "db" / "database.db")) as conn:
@@ -234,6 +278,18 @@ def delete_file():
 
             record = dict(record)
 
+            # 获取文件路径并删除实际文件
+            file_path = Path(BASE_DIR / "videoFile" / record['file_path'])
+            if file_path.exists():
+                try:
+                    file_path.unlink()  # 删除文件
+                    print(f"✅ 实际文件已删除: {file_path}")
+                except Exception as e:
+                    print(f"⚠️ 删除实际文件失败: {e}")
+                    # 即使删除文件失败，也要继续删除数据库记录，避免数据不一致
+            else:
+                print(f"⚠️ 实际文件不存在: {file_path}")
+
             # 删除数据库记录
             cursor.execute("DELETE FROM file_records WHERE id = ?", (file_id,))
             conn.commit()
@@ -338,6 +394,7 @@ def postVideo():
     productLink = data.get('productLink', '')
     productTitle = data.get('productTitle', '')
     thumbnail_path = data.get('thumbnail', '')
+    is_draft = data.get('isDraft', False)  # 新增参数：是否保存为草稿
 
     videos_per_day = data.get('videosPerDay')
     daily_times = data.get('dailyTimes')
@@ -351,7 +408,7 @@ def postVideo():
                                start_days)
         case 2:
             post_video_tencent(title, file_list, tags, account_list, category, enableTimer, videos_per_day, daily_times,
-                               start_days)
+                               start_days, is_draft)
         case 3:
             post_video_DouYin(title, file_list, tags, account_list, category, enableTimer, videos_per_day, daily_times,
                       start_days, thumbnail_path, productLink, productTitle)
@@ -450,6 +507,127 @@ def postVideoBatch():
             "data": None
         }), 200
 
+# Cookie文件上传API
+@app.route('/uploadCookie', methods=['POST'])
+def upload_cookie():
+    try:
+        if 'file' not in request.files:
+            return jsonify({
+                "code": 500,
+                "msg": "没有找到Cookie文件",
+                "data": None
+            }), 400
+
+        file = request.files['file']
+        if file.filename == '':
+            return jsonify({
+                "code": 500,
+                "msg": "Cookie文件名不能为空",
+                "data": None
+            }), 400
+
+        if not file.filename.endswith('.json'):
+            return jsonify({
+                "code": 500,
+                "msg": "Cookie文件必须是JSON格式",
+                "data": None
+            }), 400
+
+        # 获取账号信息
+        account_id = request.form.get('id')
+        platform = request.form.get('platform')
+
+        if not account_id or not platform:
+            return jsonify({
+                "code": 500,
+                "msg": "缺少账号ID或平台信息",
+                "data": None
+            }), 400
+
+        # 从数据库获取账号的文件路径
+        with sqlite3.connect(Path(BASE_DIR / "db" / "database.db")) as conn:
+            conn.row_factory = sqlite3.Row
+            cursor = conn.cursor()
+            cursor.execute('SELECT filePath FROM user_info WHERE id = ?', (account_id,))
+            result = cursor.fetchone()
+
+        if not result:
+            return jsonify({
+                "code": 500,
+                "msg": "账号不存在",
+                "data": None
+            }), 404
+
+        # 保存上传的Cookie文件到对应路径
+        cookie_file_path = Path(BASE_DIR / "cookiesFile" / result['filePath'])
+        cookie_file_path.parent.mkdir(parents=True, exist_ok=True)
+
+        file.save(str(cookie_file_path))
+
+        # 更新数据库中的账号信息（可选，比如更新更新时间）
+        # 这里可以根据需要添加额外的处理逻辑
+
+        return jsonify({
+            "code": 200,
+            "msg": "Cookie文件上传成功",
+            "data": None
+        }), 200
+
+    except Exception as e:
+        print(f"上传Cookie文件时出错: {str(e)}")
+        return jsonify({
+            "code": 500,
+            "msg": f"上传Cookie文件失败: {str(e)}",
+            "data": None
+        }), 500
+
+
+# Cookie文件下载API
+@app.route('/downloadCookie', methods=['GET'])
+def download_cookie():
+    try:
+        file_path = request.args.get('filePath')
+        if not file_path:
+            return jsonify({
+                "code": 500,
+                "msg": "缺少文件路径参数",
+                "data": None
+            }), 400
+
+        # 验证文件路径的安全性，防止路径遍历攻击
+        cookie_file_path = Path(BASE_DIR / "cookiesFile" / file_path).resolve()
+        base_path = Path(BASE_DIR / "cookiesFile").resolve()
+
+        if not cookie_file_path.is_relative_to(base_path):
+            return jsonify({
+                "code": 500,
+                "msg": "非法文件路径",
+                "data": None
+            }), 400
+
+        if not cookie_file_path.exists():
+            return jsonify({
+                "code": 500,
+                "msg": "Cookie文件不存在",
+                "data": None
+            }), 404
+
+        # 返回文件
+        return send_from_directory(
+            directory=str(cookie_file_path.parent),
+            path=cookie_file_path.name,
+            as_attachment=True
+        )
+
+    except Exception as e:
+        print(f"下载Cookie文件时出错: {str(e)}")
+        return jsonify({
+            "code": 500,
+            "msg": f"下载Cookie文件失败: {str(e)}",
+            "data": None
+        }), 500
+
+
 # 包装函数：在线程中运行异步函数
 def run_async_function(type,id,status_queue):
     match type:

sau_frontend/.env.development

@@ -2,7 +2,7 @@
 NODE_ENV=development
 
 # API 基础地址
-VITE_API_BASE_URL=http://localhost:5409
+VITE_API_BASE_URL=/api
 
 # 应用端口
 VITE_PORT=5173

sau_frontend/src/api/account.js

@@ -2,21 +2,26 @@ import { http } from '@/utils/request'
 
 // 账号管理相关API
 export const accountApi = {
-  // 获取有效账号列表
+  // 获取有效账号列表（带验证）
   getValidAccounts() {
     return http.get('/getValidAccounts')
   },
-  
+
+  // 获取账号列表（不带验证，快速加载）
+  getAccounts() {
+    return http.get('/getAccounts')
+  },
+
   // 添加账号
   addAccount(data) {
     return http.post('/account', data)
   },
-  
+
   // 更新账号
   updateAccount(data) {
     return http.post('/updateUserinfo', data)
   },
-  
+
   // 删除账号
   deleteAccount(id) {
     return http.get(`/deleteAccount?id=${id}`)

sau_frontend/src/api/material.js

@@ -8,9 +8,9 @@ export const materialApi = {
   },
   
   // 上传素材
-  uploadMaterial: (formData) => {
+  uploadMaterial: (formData, onUploadProgress) => {
     // 使用http.upload方法，它已经配置了正确的Content-Type
-    return http.upload('/uploadSave', formData)
+    return http.upload('/uploadSave', formData, onUploadProgress)
   },
   
   // 删除素材

sau_frontend/src/utils/request.js

@@ -87,11 +87,12 @@ export const http = {
     return request.delete(url, { params })
   },
   
-  upload(url, formData) {
+  upload(url, formData, onUploadProgress) {
     return request.post(url, formData, {
       headers: {
         'Content-Type': 'multipart/form-data'
-      }
+      },
+      onUploadProgress
     })
   }
 }

sau_frontend/src/views/AccountManagement.vue

@@ -30,7 +30,7 @@
               <el-table :data="filteredAccounts" style="width: 100%">
                 <el-table-column label="头像" width="80">
                   <template #default="scope">
-                    <el-avatar :src="scope.row.avatar" :size="40" />
+                    <el-avatar :src="getDefaultAvatar(scope.row.name)" :size="40" />
                   </template>
                 </el-table-column>
                 <el-table-column prop="name" label="名称" width="180" />
@@ -47,9 +47,14 @@
                 <el-table-column prop="status" label="状态">
                   <template #default="scope">
                     <el-tag
-                      :type="scope.row.status === '正常' ? 'success' : 'danger'"
+                      :type="getStatusTagType(scope.row.status)"
                       effect="plain"
+                      :class="{'clickable-status': isStatusClickable(scope.row.status)}"
+                      @click="handleStatusClick(scope.row)"
                     >
+                      <el-icon :class="scope.row.status === '验证中' ? 'is-loading' : ''" v-if="scope.row.status === '验证中'">
+                        <Loading />
+                      </el-icon>
                       {{ scope.row.status }}
                     </el-tag>
                   </template>
@@ -57,6 +62,8 @@
                 <el-table-column label="操作">
                   <template #default="scope">
                     <el-button size="small" @click="handleEdit(scope.row)">编辑</el-button>
+                    <el-button size="small" type="primary" :icon="Download" @click="handleDownloadCookie(scope.row)">下载Cookie</el-button>
+                    <el-button size="small" type="info" :icon="Upload" @click="handleUploadCookie(scope.row)">上传Cookie</el-button>
                     <el-button size="small" type="danger" @click="handleDelete(scope.row)">删除</el-button>
                   </template>
                 </el-table-column>
@@ -93,7 +100,7 @@
               <el-table :data="filteredKuaishouAccounts" style="width: 100%">
                 <el-table-column label="头像" width="80">
                   <template #default="scope">
-                    <el-avatar :src="scope.row.avatar" :size="40" />
+                    <el-avatar :src="getDefaultAvatar(scope.row.name)" :size="40" />
                   </template>
                 </el-table-column>
                 <el-table-column prop="name" label="名称" width="180" />
@@ -110,9 +117,14 @@
                 <el-table-column prop="status" label="状态">
                   <template #default="scope">
                     <el-tag
-                      :type="scope.row.status === '正常' ? 'success' : 'danger'"
+                      :type="getStatusTagType(scope.row.status)"
                       effect="plain"
+                      :class="{'clickable-status': isStatusClickable(scope.row.status)}"
+                      @click="handleStatusClick(scope.row)"
                     >
+                      <el-icon :class="scope.row.status === '验证中' ? 'is-loading' : ''" v-if="scope.row.status === '验证中'">
+                        <Loading />
+                      </el-icon>
                       {{ scope.row.status }}
                     </el-tag>
                   </template>
@@ -120,6 +132,8 @@
                 <el-table-column label="操作">
                   <template #default="scope">
                     <el-button size="small" @click="handleEdit(scope.row)">编辑</el-button>
+                    <el-button size="small" type="primary" :icon="Download" @click="handleDownloadCookie(scope.row)">下载Cookie</el-button>
+                    <el-button size="small" type="info" :icon="Upload" @click="handleUploadCookie(scope.row)">上传Cookie</el-button>
                     <el-button size="small" type="danger" @click="handleDelete(scope.row)">删除</el-button>
                   </template>
                 </el-table-column>
@@ -156,7 +170,7 @@
               <el-table :data="filteredDouyinAccounts" style="width: 100%">
                 <el-table-column label="头像" width="80">
                   <template #default="scope">
-                    <el-avatar :src="scope.row.avatar" :size="40" />
+                    <el-avatar :src="getDefaultAvatar(scope.row.name)" :size="40" />
                   </template>
                 </el-table-column>
                 <el-table-column prop="name" label="名称" width="180" />
@@ -173,9 +187,14 @@
                 <el-table-column prop="status" label="状态">
                   <template #default="scope">
                     <el-tag
-                      :type="scope.row.status === '正常' ? 'success' : 'danger'"
+                      :type="getStatusTagType(scope.row.status)"
                       effect="plain"
+                      :class="{'clickable-status': isStatusClickable(scope.row.status)}"
+                      @click="handleStatusClick(scope.row)"
                     >
+                      <el-icon :class="scope.row.status === '验证中' ? 'is-loading' : ''" v-if="scope.row.status === '验证中'">
+                        <Loading />
+                      </el-icon>
                       {{ scope.row.status }}
                     </el-tag>
                   </template>
@@ -183,6 +202,8 @@
                 <el-table-column label="操作">
                   <template #default="scope">
                     <el-button size="small" @click="handleEdit(scope.row)">编辑</el-button>
+                    <el-button size="small" type="primary" :icon="Download" @click="handleDownloadCookie(scope.row)">下载Cookie</el-button>
+                    <el-button size="small" type="info" :icon="Upload" @click="handleUploadCookie(scope.row)">上传Cookie</el-button>
                     <el-button size="small" type="danger" @click="handleDelete(scope.row)">删除</el-button>
                   </template>
                 </el-table-column>
@@ -219,7 +240,7 @@
               <el-table :data="filteredChannelsAccounts" style="width: 100%">
                 <el-table-column label="头像" width="80">
                   <template #default="scope">
-                    <el-avatar :src="scope.row.avatar" :size="40" />
+                    <el-avatar :src="getDefaultAvatar(scope.row.name)" :size="40" />
                   </template>
                 </el-table-column>
                 <el-table-column prop="name" label="名称" width="180" />
@@ -236,9 +257,14 @@
                 <el-table-column prop="status" label="状态">
                   <template #default="scope">
                     <el-tag
-                      :type="scope.row.status === '正常' ? 'success' : 'danger'"
+                      :type="getStatusTagType(scope.row.status)"
                       effect="plain"
+                      :class="{'clickable-status': isStatusClickable(scope.row.status)}"
+                      @click="handleStatusClick(scope.row)"
                     >
+                      <el-icon :class="scope.row.status === '验证中' ? 'is-loading' : ''" v-if="scope.row.status === '验证中'">
+                        <Loading />
+                      </el-icon>
                       {{ scope.row.status }}
                     </el-tag>
                   </template>
@@ -246,6 +272,8 @@
                 <el-table-column label="操作">
                   <template #default="scope">
                     <el-button size="small" @click="handleEdit(scope.row)">编辑</el-button>
+                    <el-button size="small" type="primary" :icon="Download" @click="handleDownloadCookie(scope.row)">下载Cookie</el-button>
+                    <el-button size="small" type="info" :icon="Upload" @click="handleUploadCookie(scope.row)">上传Cookie</el-button>
                     <el-button size="small" type="danger" @click="handleDelete(scope.row)">删除</el-button>
                   </template>
                 </el-table-column>
@@ -282,7 +310,7 @@
               <el-table :data="filteredXiaohongshuAccounts" style="width: 100%">
                 <el-table-column label="头像" width="80">
                   <template #default="scope">
-                    <el-avatar :src="scope.row.avatar" :size="40" />
+                    <el-avatar :src="getDefaultAvatar(scope.row.name)" :size="40" />
                   </template>
                 </el-table-column>
                 <el-table-column prop="name" label="名称" width="180" />
@@ -299,9 +327,14 @@
                 <el-table-column prop="status" label="状态">
                   <template #default="scope">
                     <el-tag
-                      :type="scope.row.status === '正常' ? 'success' : 'danger'"
+                      :type="getStatusTagType(scope.row.status)"
                       effect="plain"
+                      :class="{'clickable-status': isStatusClickable(scope.row.status)}"
+                      @click="handleStatusClick(scope.row)"
                     >
+                      <el-icon :class="scope.row.status === '验证中' ? 'is-loading' : ''" v-if="scope.row.status === '验证中'">
+                        <Loading />
+                      </el-icon>
                       {{ scope.row.status }}
                     </el-tag>
                   </template>
@@ -309,6 +342,8 @@
                 <el-table-column label="操作">
                   <template #default="scope">
                     <el-button size="small" @click="handleEdit(scope.row)">编辑</el-button>
+                    <el-button size="small" type="primary" :icon="Download" @click="handleDownloadCookie(scope.row)">下载Cookie</el-button>
+                    <el-button size="small" type="info" :icon="Upload" @click="handleUploadCookie(scope.row)">上传Cookie</el-button>
                     <el-button size="small" type="danger" @click="handleDelete(scope.row)">删除</el-button>
                   </template>
                 </el-table-column>
@@ -393,7 +428,7 @@
 
 <script setup>
 import { ref, reactive, computed, onMounted, onBeforeUnmount } from 'vue'
-import { Refresh, CircleCheckFilled, CircleCloseFilled } from '@element-plus/icons-vue'
+import { Refresh, CircleCheckFilled, CircleCloseFilled, Download, Upload, Loading } from '@element-plus/icons-vue'
 import { ElMessage, ElMessageBox } from 'element-plus'
 import { accountApi } from '@/api/account'
 import { useAccountStore } from '@/stores/account'
@@ -410,12 +445,31 @@ const activeTab = ref('all')
 // 搜索关键词
 const searchKeyword = ref('')
 
-// 获取账号数据
+// 获取账号数据（快速，不验证）
+const fetchAccountsQuick = async () => {
+  try {
+    const res = await accountApi.getAccounts()
+    if (res.code === 200 && res.data) {
+      // 将所有账号的状态暂时设为"验证中"
+      const accountsWithPendingStatus = res.data.map(account => {
+        // account[4] 是状态字段，暂时设为"验证中"
+        const updatedAccount = [...account];
+        updatedAccount[4] = '验证中'; // 临时状态
+        return updatedAccount;
+      });
+      accountStore.setAccounts(accountsWithPendingStatus);
+    }
+  } catch (error) {
+    console.error('快速获取账号数据失败:', error)
+  }
+}
+
+// 获取账号数据（带验证）
 const fetchAccounts = async () => {
   if (appStore.isAccountRefreshing) return
-  
+
   appStore.setAccountRefreshing(true)
-  
+
   try {
     const res = await accountApi.getValidAccounts()
     if (res.code === 200 && res.data) {
@@ -436,12 +490,30 @@ const fetchAccounts = async () => {
   }
 }
 
+// 后台验证所有账号（优化版本，使用setTimeout避免阻塞UI）
+const validateAllAccountsInBackground = async () => {
+  // 使用setTimeout将验证过程放在下一个事件循环，避免阻塞UI
+  setTimeout(async () => {
+    try {
+      const res = await accountApi.getValidAccounts()
+      if (res.code === 200 && res.data) {
+        accountStore.setAccounts(res.data)
+      }
+    } catch (error) {
+      console.error('后台验证账号失败:', error)
+    }
+  }, 0)
+}
+
 // 页面加载时获取账号数据
 onMounted(() => {
-  // 只有第一次进入时才获取数据
-  if (appStore.isFirstTimeAccountManagement) {
-    fetchAccounts()
-  }
+  // 快速获取账号列表（不验证），立即显示
+  fetchAccountsQuick()
+
+  // 在后台验证所有账号
+  setTimeout(() => {
+    validateAllAccountsInBackground()
+  }, 100) // 稍微延迟一下，让用户看到快速加载的效果
 })
 
 // 获取平台标签类型
@@ -455,10 +527,34 @@ const getPlatformTagType = (platform) => {
   return typeMap[platform] || 'info'
 }
 
+// 判断状态是否可点击（异常状态可点击）
+const isStatusClickable = (status) => {
+  return status === '异常'; // 只有异常状态可点击，验证中不可点击
+}
+
+// 获取状态标签类型
+const getStatusTagType = (status) => {
+  if (status === '验证中') {
+    return 'info'; // 验证中使用灰色
+  } else if (status === '正常') {
+    return 'success'; // 正常使用绿色
+  } else {
+    return 'danger'; // 无效使用红色
+  }
+}
+
+// 处理状态点击事件
+const handleStatusClick = (row) => {
+  if (isStatusClickable(row.status)) {
+    // 触发重新登录流程
+    handleReLogin(row)
+  }
+}
+
 // 过滤后的账号列表
 const filteredAccounts = computed(() => {
   if (!searchKeyword.value) return accountStore.accounts
-  return accountStore.accounts.filter(account => 
+  return accountStore.accounts.filter(account =>
     account.name.includes(searchKeyword.value)
   )
 })
@@ -528,7 +624,12 @@ const handleAddAccount = () => {
 // 编辑账号
 const handleEdit = (row) => {
   dialogType.value = 'edit'
-  Object.assign(accountForm, { ...row })
+  Object.assign(accountForm, {
+    id: row.id,
+    name: row.name,
+    platform: row.platform,
+    status: row.status
+  })
   dialogVisible.value = true
 }
 
@@ -547,7 +648,7 @@ const handleDelete = (row) => {
       try {
         // 调用API删除账号
         const response = await accountApi.deleteAccount(row.id)
-        
+
         if (response.code === 200) {
           // 从状态管理中删除账号
           accountStore.deleteAccount(row.id)
@@ -568,6 +669,108 @@ const handleDelete = (row) => {
     })
 }
 
+// 下载Cookie文件
+const handleDownloadCookie = (row) => {
+  // 从后端获取Cookie文件
+  const baseUrl = import.meta.env.VITE_API_BASE_URL || 'http://localhost:5409'
+  const downloadUrl = `${baseUrl}/downloadCookie?filePath=${encodeURIComponent(row.filePath)}`
+
+  // 创建一个隐藏的链接来触发下载
+  const link = document.createElement('a')
+  link.href = downloadUrl
+  link.download = `${row.name}_cookie.json`
+  link.target = '_blank'
+  link.style.display = 'none'
+  document.body.appendChild(link)
+  link.click()
+  document.body.removeChild(link)
+}
+
+// 上传Cookie文件
+const handleUploadCookie = (row) => {
+  // 创建一个隐藏的文件输入框
+  const input = document.createElement('input')
+  input.type = 'file'
+  input.accept = '.json'
+  input.style.display = 'none'
+  document.body.appendChild(input)
+
+  input.onchange = async (event) => {
+    const file = event.target.files[0]
+    if (!file) return
+
+    // 检查文件类型
+    if (!file.name.endsWith('.json')) {
+      ElMessage.error('请选择JSON格式的Cookie文件')
+      document.body.removeChild(input)
+      return
+    }
+
+    try {
+      // 创建FormData对象
+      const formData = new FormData()
+      formData.append('file', file)
+      formData.append('id', row.id)
+      formData.append('platform', row.platform)
+
+      // 发送上传请求
+      const baseUrl = import.meta.env.VITE_API_BASE_URL || 'http://localhost:5409'
+      const response = await fetch(`${baseUrl}/uploadCookie`, {
+        method: 'POST',
+        body: formData
+      })
+
+      const result = await response.json()
+
+      if (result.code === 200) {
+        ElMessage.success('Cookie文件上传成功')
+        // 刷新账号列表以显示更新
+        fetchAccounts()
+      } else {
+        ElMessage.error(result.msg || 'Cookie文件上传失败')
+      }
+    } catch (error) {
+      console.error('上传Cookie文件失败:', error)
+      ElMessage.error('Cookie文件上传失败')
+    } finally {
+      document.body.removeChild(input)
+    }
+  }
+
+  input.click()
+}
+
+// 重新登录账号
+const handleReLogin = (row) => {
+  // 设置表单信息
+  dialogType.value = 'edit'
+  Object.assign(accountForm, {
+    id: row.id,
+    name: row.name,
+    platform: row.platform,
+    status: row.status
+  })
+
+  // 重置SSE状态
+  sseConnecting.value = false
+  qrCodeData.value = ''
+  loginStatus.value = ''
+
+  // 显示对话框
+  dialogVisible.value = true
+
+  // 立即开始登录流程
+  setTimeout(() => {
+    connectSSE(row.platform, row.name)
+  }, 300)
+}
+
+// 获取默认头像
+const getDefaultAvatar = (name) => {
+  // 使用简单的默认头像，可以基于用户名生成不同的颜色
+  return `https://ui-avatars.com/api/?name=${encodeURIComponent(name)}&background=random`
+}
+
 // SSE事件源对象
 let eventSource = null
 
@@ -583,12 +786,12 @@ const closeSSEConnection = () => {
 const connectSSE = (platform, name) => {
   // 关闭可能存在的连接
   closeSSEConnection()
-  
+
   // 设置连接状态
   sseConnecting.value = true
   qrCodeData.value = ''
   loginStatus.value = ''
-  
+
   // 获取平台类型编号
   const platformTypeMap = {
     '小红书': '1',
@@ -596,20 +799,20 @@ const connectSSE = (platform, name) => {
     '抖音': '3',
     '快手': '4'
   }
-  
+
   const type = platformTypeMap[platform] || '1'
-  
+
   // 创建SSE连接
   const baseUrl = import.meta.env.VITE_API_BASE_URL || 'http://localhost:5409'
   const url = `${baseUrl}/login?type=${type}&id=${encodeURIComponent(name)}`
-  
+
   eventSource = new EventSource(url)
-  
+
   // 监听消息
   eventSource.onmessage = (event) => {
     const data = event.data
     console.log('SSE消息:', data)
-    
+
     // 如果还没有二维码数据，且数据长度较长，认为是二维码
     if (!qrCodeData.value && data.length > 100) {
       try {
@@ -625,30 +828,32 @@ const connectSSE = (platform, name) => {
       } catch (error) {
         console.error('处理二维码数据出错:', error)
       }
-    } 
+    }
     // 如果收到状态码
     else if (data === '200' || data === '500') {
       loginStatus.value = data
-      
+
       // 如果登录成功
       if (data === '200') {
         setTimeout(() => {
           // 关闭连接
           closeSSEConnection()
-          
+
           // 1秒后关闭对话框并开始刷新
           setTimeout(() => {
             dialogVisible.value = false
             sseConnecting.value = false
-            ElMessage.success('账号添加成功')
-            
+
+            // 根据是否是重新登录显示不同提示
+            ElMessage.success(dialogType.value === 'edit' ? '重新登录成功' : '账号添加成功')
+
             // 显示更新账号信息提示
             ElMessage({
               type: 'info',
               message: '正在同步账号信息...',
               duration: 0
             })
-            
+
             // 触发刷新操作
             fetchAccounts().then(() => {
               // 刷新完成后关闭提示
@@ -660,7 +865,7 @@ const connectSSE = (platform, name) => {
       } else {
         // 登录失败，关闭连接
         closeSSEConnection()
-        
+
         // 2秒后重置状态，允许重试
         setTimeout(() => {
           sseConnecting.value = false
@@ -670,7 +875,7 @@ const connectSSE = (platform, name) => {
       }
     }
   }
-  
+
   // 监听错误
   eventSource.onerror = (error) => {
     console.error('SSE连接错误:', error)
@@ -690,14 +895,29 @@ const submitAccountForm = () => {
       } else {
         // 编辑账号逻辑
         try {
+          // 将平台名称转换为类型数字
+          const platformTypeMap = {
+            '快手': 1,
+            '抖音': 2,
+            '视频号': 3,
+            '小红书': 4
+          };
+          const type = platformTypeMap[accountForm.platform] || 1;
+
           const res = await accountApi.updateAccount({
             id: accountForm.id,
-            type: Number(accountForm.platform === '快手' ? 1 : accountForm.platform === '抖音' ? 2 : accountForm.platform === '视频号' ? 3 : 4),
+            type: type,
             userName: accountForm.name
           })
           if (res.code === 200) {
             // 更新状态管理中的账号
-            accountStore.updateAccount(accountForm.id, accountForm)
+            const updatedAccount = {
+              id: accountForm.id,
+              name: accountForm.name,
+              platform: accountForm.platform,
+              status: accountForm.status // Keep the existing status
+            };
+            accountStore.updateAccount(accountForm.id, updatedAccount)
             ElMessage.success('更新成功')
             dialogVisible.value = false
             // 刷新账号列表
@@ -785,6 +1005,16 @@ onBeforeUnmount(() => {
   }
   
   // 二维码容器样式
+  .clickable-status {
+    cursor: pointer;
+    transition: all 0.3s;
+
+    &:hover {
+      transform: scale(1.05);
+      box-shadow: 0 0 8px rgba(0, 0, 0, 0.15);
+    }
+  }
+
   .qrcode-container {
     margin-top: 20px;
     display: flex;

sau_frontend/src/views/MaterialManagement.vue

@@ -25,6 +25,7 @@
       
       <div v-if="filteredMaterials.length > 0" class="material-list">
         <el-table :data="filteredMaterials" style="width: 100%">
+          <el-table-column prop="uuid" label="UUID" width="180" />
           <el-table-column prop="filename" label="文件名" width="300" />
           <el-table-column prop="filesize" label="文件大小" width="120">
             <template #default="scope">
@@ -58,7 +59,8 @@
           <el-form-item label="文件名称:">
             <el-input
               v-model="customFilename"
-              placeholder="选填"
+              placeholder="选填 (仅单个文件时生效)"
+              :disabled="customFilenameDisabled"
               clearable
             />
           </el-form-item>
@@ -66,10 +68,11 @@
             <el-upload
               class="upload-demo"
               drag
+              multiple
               :auto-upload="false"
               :on-change="handleFileChange"
+              :on-remove="handleFileRemove"
               :file-list="fileList"
-              :limit="1"
             >
               <el-icon class="el-icon--upload"><Upload /></el-icon>
               <div class="el-upload__text">
@@ -77,11 +80,26 @@
               </div>
               <template #tip>
                 <div class="el-upload__tip">
-                  支持视频、图片等格式文件，只能上传一个文件
+                  支持视频、图片等格式文件，可一次选择多个文件
                 </div>
               </template>
             </el-upload>
           </el-form-item>
+          <el-form-item label="上传列表" v-if="fileList.length > 0">
+            <div class="upload-file-list">
+              <div v-for="file in fileList" :key="file.uid" class="upload-file-item">
+                <span class="file-name">{{ file.name }}</span>
+                <el-progress
+                  :percentage="uploadProgress[file.uid]?.percentage || 0"
+                  :text-inside="true"
+                  :stroke-width="20"
+                  style="width: 100%; margin-top: 5px;"
+                >
+                  <span>{{ uploadProgress[file.uid]?.speed || '' }}</span>
+                </el-progress>
+              </div>
+            </div>
+          </el-form-item>
         </el-form>
       </div>
       <template #footer>
@@ -123,7 +141,7 @@
 </template>
 
 <script setup>
-import { ref, reactive, computed, onMounted } from 'vue'
+import { ref, computed, onMounted, watch } from 'vue'
 import { Refresh, Upload } from '@element-plus/icons-vue'
 import { ElMessage, ElMessageBox } from 'element-plus'
 import { materialApi } from '@/api/material'
@@ -145,6 +163,17 @@ const currentMaterial = ref(null)
 // 文件上传
 const fileList = ref([])
 const customFilename = ref('')
+const customFilenameDisabled = computed(() => fileList.value.length > 1)
+const uploadProgress = ref({}); // { [uid]: { percentage: 0, speed: '' } }
+
+
+watch(fileList, (newList) => {
+  if (newList.length <= 1) {
+    // If you want to clear the custom name when going back to single file, uncomment below
+    // customFilename.value = ''
+  }
+});
+
 
 // 获取素材列表
 const fetchMaterials = async () => {
@@ -186,6 +215,7 @@ const handleUploadMaterial = () => {
   // 清空变量
   fileList.value = []
   customFilename.value = ''
+  uploadProgress.value = {};
   uploadDialogVisible.value = true
 }
 
@@ -193,16 +223,24 @@ const handleUploadMaterial = () => {
 const handleUploadDialogClose = () => {
   fileList.value = []
   customFilename.value = ''
+  uploadProgress.value = {};
 }
 
 // 文件选择变更
 const handleFileChange = (file, uploadFileList) => {
-  // 只保留最新选择的文件
-  console.log('选择的文件:', file)
-  if (file.raw) {
-    // 确保获取到原始文件对象
-    fileList.value = [file]
+  fileList.value = uploadFileList;
+  const newProgress = {};
+  for (const f of uploadFileList) {
+    newProgress[f.uid] = { percentage: 0, speed: '' };
   }
+  uploadProgress.value = newProgress;
+}
+
+const handleFileRemove = (file, uploadFileList) => {
+  fileList.value = uploadFileList;
+  const newProgress = { ...uploadProgress.value };
+  delete newProgress[file.uid];
+  uploadProgress.value = newProgress;
 }
 
 // 提交上传
@@ -212,45 +250,67 @@ const submitUpload = async () => {
     return
   }
   
-  // 确保文件对象存在
-  const fileObj = fileList.value[0]
-  if (!fileObj || !fileObj.raw) {
-    ElMessage.warning('文件对象无效，请重新选择文件')
-    return
-  }
-  
   isUploading.value = true
   
-  try {
-    // 使用FormData进行表单提交
-    const formData = new FormData()
-    
-    // 添加文件，确保使用正确的文件对象
-    console.log('上传文件对象:', fileObj.raw)
-    formData.append('file', fileObj.raw)
-    
-    // 如果用户输入了自定义文件名，则添加到表单中
-    if (customFilename.value.trim()) {
-      formData.append('filename', customFilename.value.trim())
-      console.log('自定义文件名:', customFilename.value.trim())
-    }
-    
-    const response = await materialApi.uploadMaterial(formData)
-    
-    if (response.code === 200) {
-      ElMessage.success('上传成功')
-      uploadDialogVisible.value = false
-      // 上传成功后直接刷新素材列表
-      await fetchMaterials()
-    } else {
-      ElMessage.error(response.msg || '上传失败')
+  for (const file of fileList.value) {
+    try {
+      // 确保文件对象存在
+      if (!file || !file.raw) {
+        ElMessage.warning(`文件 ${file.name} 对象无效，已跳过`)
+        continue
+      }
+      
+      const formData = new FormData()
+      formData.append('file', file.raw)
+      
+      // 只有当只有一个文件时，自定义文件名才生效
+      if (fileList.value.length === 1 && customFilename.value.trim()) {
+        formData.append('filename', customFilename.value.trim())
+      }
+      
+      let lastLoaded = 0;
+      let lastTime = Date.now();
+
+      const response = await materialApi.uploadMaterial(formData, (progressEvent) => {
+        const progressData = uploadProgress.value[file.uid];
+        if (!progressData) return;
+
+        const progress = Math.round((progressEvent.loaded * 100) / progressEvent.total)
+        progressData.percentage = progress;
+
+        const currentTime = Date.now();
+        const timeDiff = (currentTime - lastTime) / 1000; // in seconds
+        const loadedDiff = progressEvent.loaded - lastLoaded;
+
+        if (timeDiff > 0.5) { // Update speed every 0.5 seconds
+          const speed = loadedDiff / timeDiff; // bytes per second
+          if (speed > 1024 * 1024) {
+            progressData.speed = (speed / (1024 * 1024)).toFixed(2) + ' MB/s';
+          } else {
+            progressData.speed = (speed / 1024).toFixed(2) + ' KB/s';
+          }
+          lastLoaded = progressEvent.loaded;
+          lastTime = currentTime;
+        }
+      })
+      
+      if (response.code === 200) {
+        ElMessage.success(`文件 ${file.name} 上传成功`)
+        const progressData = uploadProgress.value[file.uid];
+        if(progressData) progressData.speed = '完成';
+      } else {
+        ElMessage.error(`文件 ${file.name} 上传失败: ${response.msg || '未知错误'}`)
+      }
+    } catch (error) {
+      console.error(`上传文件 ${file.name} 出错:`, error)
+      ElMessage.error(`文件 ${file.name} 上传失败: ${error.message || '未知错误'}`)
     }
-  } catch (error) {
-    console.error('上传素材出错:', error)
-    ElMessage.error('上传失败: ' + (error.message || '未知错误'))
-  } finally {
-    isUploading.value = false
   }
+  
+  isUploading.value = false
+  // Keep dialog open to show results
+  // uploadDialogVisible.value = false 
+  await fetchMaterials()
 }
 
 // 预览素材
@@ -430,6 +490,24 @@ onMounted(() => {
   justify-content: flex-end;
 }
 
+.upload-file-list {
+  width: 100%;
+}
+
+.upload-file-item {
+  border: 1px solid #dcdfe6;
+  border-radius: 4px;
+  padding: 10px;
+  margin-bottom: 10px;
+}
+
+.upload-file-item .file-name {
+  font-size: 14px;
+  color: #606266;
+  margin-bottom: 5px;
+  display: block;
+}
+
 /* 覆盖Element Plus对话框样式 */
 :deep(.el-dialog__body) {
   padding: 20px 0;
@@ -445,4 +523,35 @@ onMounted(() => {
   padding-top: 10px;
   padding-bottom: 15px;
 }
-</style>
+
+/* 修改上传进度条样式 */
+:deep(.el-progress__text) {
+  color: #303133 !important; /* 深灰色字体，确保在各种背景上都可见 */
+  font-size: 12px;
+}
+
+:deep(.el-progress--line) {
+  margin-bottom: 10px;
+}
+
+.upload-file-item {
+  border: 1px solid #dcdfe6;
+  border-radius: 6px; /* 增加圆角 */
+  padding: 12px; /* 增加内边距 */
+  margin-bottom: 12px; /* 增加外边距 */
+  background-color: #fafafa; /* 轻微背景色 */
+  transition: box-shadow 0.3s; /* 添加过渡效果 */
+}
+
+.upload-file-item:hover {
+  box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1); /* 悬停效果 */
+}
+
+.upload-file-item .file-name {
+  font-size: 14px;
+  color: #303133; /* 深灰色字体 */
+  margin-bottom: 8px; /* 增加底部间距 */
+  display: block;
+  font-weight: 500;
+}
+</style>

sau_frontend/src/views/PublishCenter.vue

@@ -296,6 +296,36 @@
             </el-radio-group>
           </div>
 
+          <!-- 草稿选项 (仅在视频号可见) -->
+          <div v-if="tab.selectedPlatform === 2" class="draft-section">
+            <el-checkbox
+              v-model="tab.isDraft"
+              label="视频号仅保存草稿(用手机发布)"
+              class="draft-checkbox"
+            />
+          </div>
+
+          <!-- 标签 (仅在抖音可见) -->
+          <div v-if="tab.selectedPlatform === 3" class="product-section">
+            <h3>商品链接</h3>
+            <el-input
+              v-model="tab.productTitle"
+              type="text"
+              :rows="1"
+              placeholder="请输入商品名称"
+              maxlength="200"
+              class="product-name-input"
+            />
+            <el-input
+              v-model="tab.productLink"
+              type="text"
+              :rows="1"
+              placeholder="请输入商品链接"
+              maxlength="200"
+              class="product-link-input"
+            />
+          </div>
+
           <!-- 标题输入 -->
           <div class="title-section">
             <h3>标题</h3>
@@ -457,7 +487,14 @@
           <!-- 操作按钮 -->
           <div class="action-buttons">
             <el-button size="small" @click="cancelPublish(tab)">取消</el-button>
-            <el-button size="small" type="primary" @click="confirmPublish(tab)">发布</el-button>
+            <el-button
+              size="small"
+              type="primary"
+              @click="confirmPublish(tab)"
+              :loading="tab.publishing || false"
+            >
+              {{ tab.publishing ? '发布中...' : '发布' }}
+            </el-button>
           </div>
         </div>
       </div>
@@ -526,7 +563,9 @@ const defaultTabInit = {
   videosPerDay: 1, // 每天发布视频数量
   dailyTimes: ['10:00'], // 每天发布时间点列表
   startDays: 0, // 从今天开始计算的发布天数，0表示明天，1表示后天
-  publishStatus: null // 发布状态，包含message和type
+  publishStatus: null, // 发布状态，包含message和type
+  publishing: false, // 发布状态，用于控制按钮loading效果
+  isDraft: false // 是否保存为草稿，仅视频号平台可见
 }
 
 // helper to create a fresh deep-copied tab from defaultTabInit
@@ -733,29 +772,40 @@ const cancelPublish = (tab) => {
 
 // 确认发布
 const confirmPublish = async (tab) => {
+  // 防止重复点击
+  if (tab.publishing) {
+    return Promise.reject(new Error('正在发布中，请稍候...'))
+  }
+
+  tab.publishing = true // 设置发布状态为进行中
+
   return new Promise((resolve, reject) => {
     // 数据验证
     if (tab.fileList.length === 0) {
       ElMessage.error('请先上传视频文件')
+      tab.publishing = false // 重置发布状态
       reject(new Error('请先上传视频文件'))
       return
     }
     if (!tab.title.trim()) {
       ElMessage.error('请输入标题')
+      tab.publishing = false // 重置发布状态
       reject(new Error('请输入标题'))
       return
     }
     if (!tab.selectedPlatform) {
       ElMessage.error('请选择发布平台')
+      tab.publishing = false // 重置发布状态
       reject(new Error('请选择发布平台'))
       return
     }
     if (tab.selectedAccounts.length === 0) {
       ElMessage.error('请选择发布账号')
+      tab.publishing = false // 重置发布状态
       reject(new Error('请选择发布账号'))
       return
     }
-    
+
     // 构造发布数据，符合后端API格式
     const publishData = {
       type: tab.selectedPlatform,
@@ -772,9 +822,10 @@ const confirmPublish = async (tab) => {
       startDays: tab.scheduleEnabled ? tab.startDays || 0 : 0, // 从今天开始计算的发布天数，0表示明天，1表示后天
       category: 0, //表示非原创
       productLink: tab.productLink.trim() || '', // 商品链接
-      productTitle: tab.productTitle.trim() || '' // 商品名称
+      productTitle: tab.productTitle.trim() || '', // 商品名称
+      isDraft: tab.isDraft // 是否保存为草稿，仅视频号平台使用
     }
-    
+
     // 调用后端发布API
     fetch(`${apiBaseUrl}/postVideo`, {
       method: 'POST',
@@ -815,6 +866,9 @@ const confirmPublish = async (tab) => {
       }
       reject(error)
     })
+    .finally(() => {
+      tab.publishing = false // 重置发布状态
+    })
   })
 }
 
@@ -1261,10 +1315,19 @@ const batchPublish = async () => {
           padding-top: 20px;
           border-top: 1px solid #ebeef5;
         }
+
+        .draft-section {
+          margin: 20px 0;
+
+          .draft-checkbox {
+            display: block;
+            margin: 10px 0;
+          }
+        }
       }
     }
   }
-  
+
   // 已上传文件列表样式
   .uploaded-files {
     margin-top: 20px;

sau_frontend/vite.config.js

@@ -19,7 +19,14 @@ export default defineConfig({
   },
   server: {
     port: 5173,
-    open: true
+    open: true,
+    proxy: {
+      '/api': {
+        target: 'http://localhost:5409',
+        changeOrigin: true,
+        rewrite: (path) => path.replace(/^\/api/, '')
+      }
+    }
   },
   build: {
     outDir: 'dist',

uploader/tencent_uploader/main.py

@@ -82,15 +82,16 @@ async def weixin_setup(account_file, handle=False):
 
 
 class TencentVideo(object):
-    def __init__(self, title, file_path, tags, publish_date: datetime, account_file, category=None):
+    def __init__(self, title, file_path, tags, publish_date: datetime, account_file, category=None, is_draft=False):
         self.title = title  # 视频标题
         self.file_path = file_path
         self.tags = tags
         self.publish_date = publish_date
         self.account_file = account_file
         self.category = category
-        self.local_executable_path = LOCAL_CHROME_PATH
         self.headless = LOCAL_CHROME_HEADLESS
+        self.is_draft = is_draft  # 是否保存为草稿
+        self.local_executable_path = LOCAL_CHROME_PATH or None
 
     async def set_schedule_time_tencent(self, page, publish_date):
         label_element = page.locator("label").filter(has_text="定时").nth(1)
@@ -186,21 +187,37 @@ class TencentVideo(object):
     async def click_publish(self, page):
         while True:
             try:
-                publish_buttion = page.locator('div.form-btns button:has-text("发表")')
-                if await publish_buttion.count():
-                    await publish_buttion.click()
-                await page.wait_for_url("https://channels.weixin.qq.com/platform/post/list", timeout=5000)
-                tencent_logger.success("  [-]视频发布成功")
+                if self.is_draft:
+                    # 点击"保存草稿"按钮
+                    draft_button = page.locator('div.form-btns button:has-text("保存草稿")')
+                    if await draft_button.count():
+                        await draft_button.click()
+                    # 等待跳转到草稿箱页面或确认保存成功
+                    await page.wait_for_url("**/post/list**", timeout=5000)  # 使用通配符匹配包含post/list的URL
+                    tencent_logger.success("  [-]视频草稿保存成功")
+                else:
+                    # 点击"发表"按钮
+                    publish_button = page.locator('div.form-btns button:has-text("发表")')
+                    if await publish_button.count():
+                        await publish_button.click()
+                    await page.wait_for_url("https://channels.weixin.qq.com/platform/post/list", timeout=5000)
+                    tencent_logger.success("  [-]视频发布成功")
                 break
             except Exception as e:
                 current_url = page.url
-                if "https://channels.weixin.qq.com/platform/post/list" in current_url:
-                    tencent_logger.success("  [-]视频发布成功")
-                    break
+                if self.is_draft:
+                    # 检查是否在草稿相关的页面
+                    if "post/list" in current_url or "draft" in current_url:
+                        tencent_logger.success("  [-]视频草稿保存成功")
+                        break
                 else:
-                    tencent_logger.exception(f"  [-] Exception: {e}")
-                    tencent_logger.info("  [-] 视频正在发布中...")
-                    await asyncio.sleep(0.5)
+                    # 检查是否在发布列表页面
+                    if "https://channels.weixin.qq.com/platform/post/list" in current_url:
+                        tencent_logger.success("  [-]视频发布成功")
+                        break
+                tencent_logger.exception(f"  [-] Exception: {e}")
+                tencent_logger.info("  [-] 视频正在发布中...")
+                await asyncio.sleep(0.5)
 
     async def detect_upload_status(self, page):
         while True: