MibooGram/modules/error/pb.py
2022-10-28 15:11:14 +08:00

30 lines
949 B
Python

import httpx
from core.config import config
from utils.log import logger
class PbClient:
def __init__(self):
self.client = httpx.AsyncClient()
self.PB_API = config.error.pb_url
self.sunset: int = config.error.pb_sunset # 自动销毁时间 单位为秒
self.private: bool = True
self.max_lines: int = config.error.pb_max_lines
async def create_pb(self, content: str) -> str:
if not self.PB_API:
return ""
logger.info("正在上传日记到 pb")
content = "\n".join(content.splitlines()[-self.max_lines :]) + "\n"
data = {
"c": content,
}
if self.private:
data["p"] = "1"
if self.sunset:
data["sunset"] = self.sunset
data = await self.client.post(self.PB_API, data=data) # 需要错误处理
logger.success("上传日记到 pb 成功")
return data.headers["location"]