34 lines
888 B
Python
34 lines
888 B
Python
from pathlib import Path
|
|
|
|
from nonebot.log import logger
|
|
|
|
from .config import config
|
|
|
|
|
|
def save_pic(img: bytes, img_type: str, name: str) -> None:
|
|
"""
|
|
保存图片
|
|
"""
|
|
save_path = Path(config.whatpic_res_path) / f"{img_type}_pic" / (name + ".jpg")
|
|
if isinstance(img, bytes):
|
|
with Path.open(save_path, "wb") as f:
|
|
f.write(img)
|
|
else:
|
|
logger.error(f"img must be bytes, but got {type(img)}")
|
|
raise TypeError("img must be bytes")
|
|
|
|
|
|
def delete_pic(img_type: str, name: str) -> None:
|
|
"""
|
|
删除图片
|
|
"""
|
|
delete_path = Path(config.whatpic_res_path) / f"{img_type}_pic" / (name + ".jpg")
|
|
if delete_path.exists():
|
|
try:
|
|
delete_path.unlink()
|
|
except OSError as e:
|
|
logger.error(f"Error: {e}")
|
|
raise
|
|
else:
|
|
raise FileNotFoundError(f"{delete_path} not found")
|