mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-09-24 15:30:46 +08:00
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
import base64
|
|
import typing
|
|
import io
|
|
from urllib.parse import urlparse, parse_qs
|
|
import ssl
|
|
|
|
import aiohttp
|
|
import PIL.Image
|
|
|
|
|
|
def get_qq_image_downloadable_url(image_url: str) -> tuple[str, dict]:
|
|
"""获取QQ图片的下载链接"""
|
|
parsed = urlparse(image_url)
|
|
query = parse_qs(parsed.query)
|
|
return f"http://{parsed.netloc}{parsed.path}", query
|
|
|
|
|
|
async def get_qq_image_bytes(image_url: str, query: dict={}) -> tuple[bytes, str]:
|
|
"""[弃用]获取QQ图片的bytes"""
|
|
image_url, query_in_url = get_qq_image_downloadable_url(image_url)
|
|
query = {**query, **query_in_url}
|
|
ssl_context = ssl.create_default_context()
|
|
ssl_context.check_hostname = False
|
|
ssl_context.verify_mode = ssl.CERT_NONE
|
|
async with aiohttp.ClientSession(trust_env=False) as session:
|
|
async with session.get(image_url, params=query, ssl=ssl_context) as resp:
|
|
resp.raise_for_status()
|
|
file_bytes = await resp.read()
|
|
content_type = resp.headers.get('Content-Type')
|
|
if not content_type:
|
|
image_format = 'jpeg'
|
|
elif not content_type.startswith('image/'):
|
|
pil_img = PIL.Image.open(io.BytesIO(file_bytes))
|
|
image_format = pil_img.format.lower()
|
|
else:
|
|
image_format = content_type.split('/')[-1]
|
|
return file_bytes, image_format
|
|
|
|
|
|
async def qq_image_url_to_base64(
|
|
image_url: str
|
|
) -> typing.Tuple[str, str]:
|
|
"""[弃用]将QQ图片URL转为base64，并返回图片格式
|
|
|
|
Args:
|
|
image_url (str): QQ图片URL
|
|
|
|
Returns:
|
|
typing.Tuple[str, str]: base64编码和图片格式
|
|
"""
|
|
image_url, query = get_qq_image_downloadable_url(image_url)
|
|
|
|
# Flatten the query dictionary
|
|
query = {k: v[0] for k, v in query.items()}
|
|
|
|
file_bytes, image_format = await get_qq_image_bytes(image_url, query)
|
|
|
|
base64_str = base64.b64encode(file_bytes).decode()
|
|
|
|
return base64_str, image_format
|
|
|
|
async def extract_b64_and_format(image_base64_data: str) -> typing.Tuple[str, str]:
|
|
"""提取base64编码和图片格式
|
|
|
|
data:image/jpeg;base64,xxx
|
|
提取出base64编码和图片格式
|
|
"""
|
|
base64_str = image_base64_data.split(',')[-1]
|
|
image_format = image_base64_data.split(':')[-1].split(';')[0].split('/')[-1]
|
|
return base64_str, image_format |