mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-25 23:49:11 +00:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from typing import Optional
|
|
|
|
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) -> Optional[str]:
|
|
if not self.PB_API:
|
|
return None
|
|
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
|
|
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 成功")
|
|
return result.headers.get("location")
|