fix: harden sqlite concurrency and media cache delivery

Author: genz27

src/core/account_tiers.py

@@ -33,7 +33,7 @@ def get_paygate_tier_label(user_paygate_tier: Optional[str]) -> str:
         return "Ult"
     if normalized == PAYGATE_TIER_ONE:
         return "Pro"
-    return "Normal"
+    return "Free"
 
 
 def get_required_paygate_tier_for_model(model_name: Optional[str]) -> str:

src/core/database.py

@@ -126,6 +126,13 @@ def _set_request_fingerprint(self, fingerprint: Optional[Dict[str, Any]]):
         """设置当前请求链路的浏览器指纹上下文。"""
         self._request_fingerprint_ctx.set(dict(fingerprint) if fingerprint else None)
 
+    def get_request_fingerprint(self) -> Optional[Dict[str, Any]]:
+        """获取当前请求链路绑定的浏览器指纹快照。"""
+        fingerprint = self._request_fingerprint_ctx.get()
+        if not isinstance(fingerprint, dict) or not fingerprint:
+            return None
+        return dict(fingerprint)
+
     def clear_request_fingerprint(self):
         """清理请求链路绑定的浏览器指纹。"""
         self._set_request_fingerprint(None)

src/services/file_cache.py

@@ -275,3 +275,43 @@ async def select_token(
 
         debug_logger.log_info(f"[LOAD_BALANCER] ❌ 候选Token均不可用 (图片生成={for_image_generation}, 视频生成={for_video_generation})")
         return None
+
+    async def get_unavailable_reason(
+        self,
+        *,
+        for_image_generation: bool = False,
+        for_video_generation: bool = False,
+        model: Optional[str] = None,
+    ) -> Optional[str]:
+        """给出更明确的“无可用账号”原因，优先用于分辨率/tier 档位提示。"""
+        active_tokens = await self.token_manager.get_active_tokens()
+        if not active_tokens:
+            return None
+
+        required_tier = get_required_paygate_tier_for_model(model)
+        supported_tokens = []
+        for token in active_tokens:
+            normalized_tier = normalize_user_paygate_tier(token.user_paygate_tier)
+            if model and not supports_model_for_tier(model, normalized_tier):
+                continue
+            supported_tokens.append(token)
+
+        if model and not supported_tokens:
+            tier_label = get_paygate_tier_label(required_tier)
+            return f"当前模型需要 {tier_label} 账号，但没有可用的 {tier_label} 账号: {model}"
+
+        capability_tokens = []
+        for token in supported_tokens:
+            if for_image_generation and not token.image_enabled:
+                continue
+            if for_video_generation and not token.video_enabled:
+                continue
+            capability_tokens.append(token)
+
+        if supported_tokens and not capability_tokens:
+            if for_image_generation:
+                return "当前有符合档位的账号，但图片生成功能已全部禁用。"
+            if for_video_generation:
+                return "当前有符合档位的账号，但视频生成功能已全部禁用。"
+
+        return None

src/services/flow_client.py

@@ -403,7 +403,8 @@ async def _do_refresh_at(self, token_id: int, st: str) -> bool:
                 credits_result = await self.flow_client.get_credits(new_at)
                 await self.db.update_token(
                     token_id,
-                    credits=credits_result.get("credits", 0)
+                    credits=credits_result.get("credits", 0),
+                    user_paygate_tier=credits_result.get("userPaygateTier"),
                 )
                 debug_logger.log_info(f"[AT_REFRESH] Token {token_id}: AT 验证成功（余额: {credits_result.get('credits', 0)}）")
                 return True
@@ -619,9 +620,14 @@ async def refresh_credits(self, token_id: int) -> int:
         try:
             result = await self.flow_client.get_credits(token.at)
             credits = result.get("credits", 0)
+            user_paygate_tier = result.get("userPaygateTier")
 
             # 更新数据库
-            await self.db.update_token(token_id, credits=credits)
+            await self.db.update_token(
+                token_id,
+                credits=credits,
+                user_paygate_tier=user_paygate_tier,
+            )
 
             return credits
         except Exception as e:

src/services/generation_handler.py

@@ -785,7 +785,7 @@ <h3 class="text-lg font-semibold">导入 Token</h3>
         formatSora2=(t)=>{if(t.sora2_supported===true){const remaining=t.sora2_total_count-t.sora2_redeemed_count;const tooltipText=`邀请码: ${t.sora2_invite_code||'无'}\n可用次数: ${remaining}/${t.sora2_total_count}\n已用次数: ${t.sora2_redeemed_count}`;return`<div class="inline-flex items-center gap-1"><span class="inline-flex items-center rounded px-2 py-0.5 text-xs bg-green-50 text-green-700 cursor-pointer" title="${tooltipText}" onclick="copySora2Code('${t.sora2_invite_code||''}')">支持</span><span class="text-xs text-muted-foreground" title="${tooltipText}">${remaining}/${t.sora2_total_count}</span></div>`}else if(t.sora2_supported===false){return`<span class="inline-flex items-center rounded px-2 py-0.5 text-xs bg-gray-100 text-gray-700 cursor-pointer" title="点击使用邀请码激活" onclick="openSora2Modal(${t.id})">不支持</span>`}else{return'-'}},
         formatPlanTypeWithTooltip=(t)=>{const tooltipText=t.subscription_end?`套餐到期: ${new Date(t.subscription_end).toLocaleDateString('zh-CN',{year:'numeric',month:'2-digit',day:'2-digit'}).replace(/\//g,'-')} ${new Date(t.subscription_end).toLocaleTimeString('zh-CN',{hour:'2-digit',minute:'2-digit',hour12:false})}`:'';return`<span class="inline-flex items-center rounded px-2 py-0.5 text-xs bg-blue-50 text-blue-700 cursor-pointer" title="${tooltipText||t.plan_title||'-'}">${formatPlanType(t.plan_type)}</span>`},
         formatSora2Remaining=(t)=>{if(t.sora2_supported===true){const remaining=t.sora2_remaining_count||0;return`<span class="text-xs">${remaining}</span>`}else{return'-'}},
-        formatAccountType=(tier)=>{if(tier==='PAYGATE_TIER_NOT_PAID'||!tier){return`<span class="inline-flex items-center rounded px-2 py-0.5 text-xs bg-gray-100 text-gray-700">普通</span>`}if(tier==='PAYGATE_TIER_ONE'){return`<span class="inline-flex items-center rounded px-2 py-0.5 text-xs bg-blue-50 text-blue-700">Pro</span>`}if(tier==='PAYGATE_TIER_TWO'){return`<span class="inline-flex items-center rounded px-2 py-0.5 text-xs bg-purple-50 text-purple-700">Ult</span>`}return`<span class="inline-flex items-center rounded px-2 py-0.5 text-xs bg-amber-50 text-amber-700" title="${tier}">${tier}</span>`},
+        formatAccountType=(tier)=>{if(tier==='PAYGATE_TIER_NOT_PAID'||!tier){return`<span class="inline-flex items-center rounded px-2 py-0.5 text-xs bg-gray-100 text-gray-700">Free</span>`}if(tier==='PAYGATE_TIER_ONE'){return`<span class="inline-flex items-center rounded px-2 py-0.5 text-xs bg-blue-50 text-blue-700">Pro</span>`}if(tier==='PAYGATE_TIER_TWO'){return`<span class="inline-flex items-center rounded px-2 py-0.5 text-xs bg-purple-50 text-purple-700">Ult</span>`}return`<span class="inline-flex items-center rounded px-2 py-0.5 text-xs bg-amber-50 text-amber-700" title="${tier}">${tier}</span>`},
         formatProjectNameDisplay=name=>{if(!name)return'<span class="text-muted-foreground">-</span>';const text=String(name);const match=text.match(/^(.*?)(?:\s+(P\d+))$/);if(match){return`<div class="flex flex-col items-center justify-center gap-1 leading-none"><span class="text-xs font-medium text-foreground whitespace-nowrap">${escapeLogHtml(match[1])}</span><span class="inline-flex items-center rounded-full border border-slate-200 bg-slate-100 px-2 py-0.5 text-[11px] font-medium text-slate-700">${escapeLogHtml(match[2])}</span></div>`}return`<div class="text-center text-xs font-medium text-foreground whitespace-nowrap">${escapeLogHtml(text)}</div>`},
         formatProjectIdDisplay=projectId=>{if(!projectId)return'<span class="text-muted-foreground">-</span>';const text=String(projectId);const short=text.length>8?`${text.substring(0,8)}...`:text;return`<button onclick="copyProjectId('${projectId}')" class="inline-flex items-center justify-center rounded-full border border-blue-200 bg-blue-50 px-2 py-0.5 text-[10px] font-mono font-medium text-blue-700 transition-colors hover:bg-blue-100 hover:text-blue-800" title="点击复制项目ID&#10;${escapeLogHtml(text)}">${escapeLogHtml(short)}</button>`},
         renderTokenActions=t=>{const toggleLabel=t.is_active?'禁用':'启用';return`<div class="inline-flex flex-nowrap items-center justify-end gap-1"><button onclick="refreshTokenAT(${t.id})" class="inline-flex h-7 items-center justify-center rounded-md px-2 text-xs font-medium hover:bg-blue-50 hover:text-blue-700">刷新AT</button><button onclick="openEditModal(${t.id})" class="inline-flex h-7 items-center justify-center rounded-md px-2 text-xs font-medium hover:bg-green-50 hover:text-green-700">详情</button><button onclick="toggleToken(${t.id},${t.is_active})" class="inline-flex h-7 items-center justify-center rounded-md px-2 text-xs font-medium hover:bg-accent">${toggleLabel}</button><button onclick="deleteToken(${t.id})" class="inline-flex h-7 items-center justify-center rounded-md px-2 text-xs font-medium hover:bg-destructive/10 hover:text-destructive">删除</button></div>`},