PaiGramTeam/PaiGram
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/PaiGram.git synced 2024-11-26 02:11:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

🎨 Improve error handling in create_pb

Browse Source
This commit is contained in:
洛水居室 2022-11-21 12:23:22 +08:00
parent 84f0f4d5aa
commit 60b8e46924
No known key found for this signature in database
GPG Key ID: C9DE87DA724B88FC
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
modules/error/pb.py
View File

@ -1,3 +1,5 @@
from typing import Optional
import httpx import httpx
from core.config import config from core.config import config
@ -12,11 +14,11 @@ class PbClient:
self.private: bool = True self.private: bool = True
self.max_lines: int = config.error.pb_max_lines self.max_lines: int = config.error.pb_max_lines
async def create_pb(self, content: str) -> str: async def create_pb(self, content: str) -> Optional[str]:
if not self.PB_API: if not self.PB_API:
return "" return None
logger.info("正在上传日记到 pb") logger.info("正在上传日记到 pb")
content = "\n".join(content.splitlines()[-self.max_lines :]) + "\n" content = "\n".join(content.splitlines()[-self.max_lines:]) + "\n"
data = { data = {
"c": content, "c": content,
} }
@ -24,6 +26,9 @@ class PbClient:
data["p"] = "1" data["p"] = "1"
if self.sunset: if self.sunset:
data["sunset"] = self.sunset data["sunset"] = self.sunset
data = await self.client.post(self.PB_API, data=data) # 需要错误处理 result = await self.client.post(self.PB_API, data=data)
if result.is_error:
logger.warning("上传日记到 pb 失败 status_code[%s]", result.status_code)
return None
logger.success("上传日记到 pb 成功") logger.success("上传日记到 pb 成功")
return data.headers["location"] return result.headers.get("location")