feat: 禁言检测 - 面板显示禁言状态与到期时间

account_manager.py

@@ -16,6 +16,8 @@ class Account:
     in_use: bool = False
     error_count: int = 0
     logged_in: bool = False
+    is_muted: bool = False
+    muted_until: str = ""
 
 
 class AccountManager:
@@ -85,6 +87,9 @@ class AccountManager:
                 )
                 await account.browser.start()
                 account.logged_in = True
+                # Check mute status
+                account.is_muted = account.browser.is_muted()
+                account.muted_until = account.browser.muted_until()
             return account.browser
         except Exception as e:
             print(f"Error creating browser: {e}")
@@ -112,10 +117,25 @@ class AccountManager:
         in_use = sum(1 for a in self.accounts.values() if a.in_use)
         available = sum(1 for a in self.accounts.values() if not a.in_use and a.error_count < 3)
         logged_in = sum(1 for a in self.accounts.values() if a.logged_in)
+        muted = sum(1 for a in self.accounts.values() if a.is_muted)
+        accounts_list = [
+            {
+                "email": a.email,
+                "name": a.name,
+                "in_use": a.in_use,
+                "logged_in": a.logged_in,
+                "is_muted": a.is_muted,
+                "muted_until": a.muted_until,
+                "error_count": a.error_count,
+            }
+            for a in self.accounts.values()
+        ]
         return {
             "total": total,
             "in_use": in_use,
             "available": available,
             "logged_in": logged_in,
+            "muted": muted,
             "queue_size": len(self.queue),
+            "accounts": accounts_list,
         }

deepseek_browser.py

@@ -61,15 +61,50 @@ class DeepSeekBrowser:
             except Exception:
                 await self._auto_login()
 
+        # Check if account is muted after login
+        if self._logged_in:
+            await self._check_mute()
+
+    async def _check_mute(self):
+        """Check if account is muted and extract mute expiry."""
+        try:
+            muted, until = await self.page.evaluate("""() => {
+                const text = document.body.innerText || '';
+                // Match: 禁言至 YYYY年M月D日 HH:MM or 禁言至 YYYY-MM-DD HH:MM
+                const match = text.match(/禁言至\\s*(\\d{4}[-年]\\d{1,2}[-月]\\d{1,2}[日]?\\s*\\d{1,2}:\\d{2})/);
+                if (match) return [true, match[1]];
+                if (text.includes('禁言')) return [true, ''];
+                return [false, ''];
+            }""")
+            self._is_muted = muted
+            self._muted_until = until
+            if muted:
+                print(f"[mute] {self.email} is muted until {until}")
+        except Exception:
+            self._is_muted = False
+            self._muted_until = ""
+
+    def is_muted(self) -> bool:
+        return getattr(self, '_is_muted', False)
+
+    def muted_until(self) -> str:
+        return getattr(self, '_muted_until', "")
+
     async def _auto_login(self):
         print(f"Logging in as {self.email}...")
 
         try:
-            email_input = self.page.locator('input[placeholder*="邮箱"], input[placeholder*="手机"], input[type="text"]').first
+            email_input = self.page.locator('input[placeholder*="邮箱"], input[placeholder*="手机"], input[placeholder*="Email"], input[placeholder*="email"], input[type="text"]').first
             await email_input.wait_for(state="visible", timeout=10000)
             await email_input.fill(self.email)
             await asyncio.sleep(0.5)
         except Exception as e:
+            # Take screenshot to debug
+            try:
+                await self.page.screenshot(path=f"/tmp/login_fail_{self.email.replace('@','_at_')}.png")
+                print(f"Screenshot saved to /tmp/login_fail_{self.email.replace('@','_at_')}.png")
+            except Exception:
+                pass
             print(f"Email input error: {e}")
             raise
 
@@ -113,22 +148,98 @@ class DeepSeekBrowser:
 
     async def delete_chat(self):
         try:
-            more_btn = self.page.locator('button:has-text("更多"), .ds-icon-button:has-text("...")').first
-            if await more_btn.count() > 0:
-                await more_btn.click()
-                await asyncio.sleep(0.5)
-
-                delete_btn = self.page.locator('button:has-text("删除"), div:has-text("删除对话")').first
-                if await delete_btn.count() > 0:
-                    await delete_btn.click()
-                    await asyncio.sleep(0.5)
+            # Find the sidebar and active conversation
+            chat_list = self.page.locator(
+                'nav, aside, [class*="sidebar"], [class*="Sidebar"], div:has-text("开启新对话")'
+            ).first
+            chat_list_count = await chat_list.count()
+            if chat_list_count == 0:
+                print(f"[delete_chat] no sidebar")
+                return
+
+            active_item = chat_list.locator(
+                '[class*="active"], [class*="selected"], [class*="current"]'
+            ).first
+            active_count = await active_item.count()
+            if active_count == 0:
+                # No active item yet (first chat), skip deletion
+                print(f"[delete_chat] no active item, skipping")
+                return
+
+            # Get bounding box and click near right edge where "..." should be
+            box = await active_item.bounding_box()
+            if not box:
+                print(f"[delete_chat] no bbox")
+                return
+
+            # Instead of position-based click, find the "..." element in DOM
+            click_result = await self.page.evaluate("""() => {
+                // Find the active/highlighted conversation item
+                const active = document.querySelector('[class*="active"], [class*="selected"]');
+                if (!active) return 'no-active';
+                
+                // Walk down to find a clickable child that looks like "..."
+                // The "..." is often a button or div with no text (SVG only)
+                const walk = (node, depth) => {
+                    if (depth > 10) return null;
+                    for (const child of node.children || []) {
+                        const tag = child.tagName;
+                        const cls = (child.className || '').toString();
+                        // Look for small icon-like elements
+                        if ((tag === 'BUTTON' || tag === 'svg' || cls.includes('icon') || cls.includes('more') || cls.includes('menu') || cls.includes('action')) && 
+                            child.offsetWidth < 40 && child.offsetWidth > 0) {
+                            return child;
+                        }
+                        const found = walk(child, depth + 1);
+                        if (found) return found;
+                    }
+                    return null;
+                };
+                
+                const icon = walk(active, 0);
+                if (icon) {
+                    icon.click();
+                    return 'clicked:' + icon.tagName + ':' + (icon.className || '').substring(0, 40);
+                }
+                
+                // Fallback: find any button/svg in active item
+                const btn = active.querySelector('button, [role="button"]');
+                if (btn) {
+                    btn.click();
+                    return 'fallback:' + btn.tagName;
+                }
+                return 'no-icon';
+            }""")
+            print(f"[delete_chat] icon click: {click_result}")
+            await asyncio.sleep(0.5)
 
-                    confirm_btn = self.page.locator('button:has-text("确认"), button:has-text("删除")').last
-                    if await confirm_btn.count() > 0:
-                        await confirm_btn.click()
-                        await asyncio.sleep(1)
-        except Exception:
-            pass
+            # Search for "删除" or "Delete" anywhere on page
+            delete_btn = self.page.locator(
+                ':has-text("删除"), :has-text("Delete")'
+            ).first
+            delete_count = await delete_btn.count()
+            
+            if delete_count == 0:
+                print(f"[delete_chat] no delete option found")
+                return
+
+            await delete_btn.click()
+            await asyncio.sleep(0.5)
+
+            # Confirm
+            confirm_btn = self.page.locator(
+                'button:has-text("确认"), button:has-text("删除"), '
+                'button:has-text("Confirm"), button:has-text("Delete")'
+            ).last
+            if await confirm_btn.count() > 0:
+                await confirm_btn.click()
+                await asyncio.sleep(1)
+                print(f"[delete_chat] done!")
+            else:
+                print(f"[delete_chat] no confirm btn")
+                
+        except Exception as e:
+            print(f"[delete_chat] error: {e}")
 
     async def switch_model(self, model: str):
         try:

main.py

@@ -401,8 +401,8 @@ textarea::placeholder{color:var(--dim)}
     </div>
     <div class="panel-body" style="padding-bottom:8px">
       <table class="tbl">
-        <thead><tr><th>邮箱</th><th class="hide-mobile">备注</th><th>登录</th><th>状态</th><th class="hide-mobile">错误</th></tr></thead>
-        <tbody id="tbl"><tr><td colspan="5" class="empty">加载中…</td></tr></tbody>
+        <thead><tr><th>邮箱</th><th class="hide-mobile">备注</th><th>登录</th><th>状态</th><th>禁言</th><th class="hide-mobile">错误</th></tr></thead>
+        <tbody id="tbl"><tr><td colspan="6" class="empty">加载中…</td></tr></tbody>
       </table>
     </div>
   </div>
@@ -503,6 +503,7 @@ async function loadStats(){
        <span>活跃 <b>${s.accounts.in_use}</b></span>
        <span>可用 <b>${s.accounts.available}</b></span>
        <span>在线 <b>${s.accounts.logged_in}</b></span>
+       <span>禁言 <b style="color:var(--red)">${s.accounts.muted||0}</b></span>
        <span>排队 <b>${s.accounts.queue_size}</b></span>`
   }catch(e){}
 }
@@ -516,12 +517,13 @@ async function loadAccounts(){
         <td class="hide-mobile">${a.name||'—'}</td>
         <td><span class="badge ${a.logged_in?'badge-on':'badge-off'}">${a.logged_in?'在线':'离线'}</span></td>
         <td><span class="badge ${a.in_use?'badge-on':'badge-idle'}">${a.in_use?'使用中':'空闲'}</span></td>
+        <td>${a.is_muted?`<span class="badge badge-off" title="${a.muted_until||'已禁言'}">禁言</span>`:'<span class="badge badge-idle">正常</span>'}</td>
         <td class="hide-mobile">${a.error_count>0?'<span class="badge badge-off">'+a.error_count+'</span>':'—'}</td>
       </tr>`
     }
-    document.getElementById('tbl').innerHTML=r||'<tr><td colspan="5" class="empty">暂无账号</td></tr>'
+    document.getElementById('tbl').innerHTML=r||'<tr><td colspan="6" class="empty">暂无账号</td></tr>'
   }catch(e){
-    document.getElementById('tbl').innerHTML='<tr><td colspan="5" style="color:var(--red)">'+e.message+'</td></tr>'
+    document.getElementById('tbl').innerHTML='<tr><td colspan="6" style="color:var(--red)">'+e.message+'</td></tr>'
   }
 }
 async function loadAll(){await loadStats();await loadAccounts()}