From 0d35e3390e751e624decaa2dc0b3b70680eb2275 Mon Sep 17 00:00:00 2001 From: felinae98 <731499577@qq.com> Date: Fri, 19 Feb 2021 15:54:56 +0800 Subject: [PATCH 1/2] Revert "add log for merge image" This reverts commit 385d47268f9d928a0594fa53f64a01dfe7f2485c. --- src/plugins/hk_reporter/post.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/src/plugins/hk_reporter/post.py b/src/plugins/hk_reporter/post.py index 6485d16..255c7b6 100644 --- a/src/plugins/hk_reporter/post.py +++ b/src/plugins/hk_reporter/post.py @@ -1,9 +1,7 @@ import base64 -import math from dataclasses import dataclass, field from io import BytesIO from typing import NoReturn, Optional -from nonebot import logger import httpx from PIL import Image from .plugin_config import plugin_config @@ -37,17 +35,15 @@ class Post: if len(self.pics) < 6: return first_image = await self._pic_url_to_image(self.pics[0]) - if abs(first_image.size[0] - first_image.size[1]) / first_image.size[0] > 0.05: + if first_image.size[0] != first_image.size[1]: return pic_size = first_image.size[0] images = [first_image] for pic in self.pics[1:]: cur_image = await self._pic_url_to_image(pic) - if abs(first_image.size[0] - pic_size) / pic_size > 0.05 or \ - abs(first_image.size[1] - pic_size) / pic_size > 0.05: + if cur_image.size[0] != pic_size or cur_image.size[1] != pic_size: break images.append(cur_image) - logger.debug('{} images is same size'.format(len(images))) if len(images) == 6: matrix = (3, 2) self.pics = self.pics[6:] @@ -56,7 +52,6 @@ class Post: self.pics = self.pics[9:] else: return - logger.info('trigger merge image with {} imgs'.format(len(images))) target = Image.new('RGB', (matrix[0] * pic_size, matrix[1] * pic_size)) for y in range(matrix[1]): for x in range(matrix[0]): From a2a91ee8043da4dc30475af47eb70a3a6813fb2a Mon Sep 17 00:00:00 2001 From: felinae98 <731499577@qq.com> Date: Fri, 19 Feb 2021 15:58:44 +0800 Subject: [PATCH 2/2] add logger for merge image --- src/plugins/hk_reporter/platform/weibo.py | 2 +- src/plugins/hk_reporter/post.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/plugins/hk_reporter/platform/weibo.py b/src/plugins/hk_reporter/platform/weibo.py index 078c4e1..aa2bd12 100644 --- a/src/plugins/hk_reporter/platform/weibo.py +++ b/src/plugins/hk_reporter/platform/weibo.py @@ -68,7 +68,7 @@ class Weibo(Platform): def _get_text(self, raw_text: str) -> str: text = raw_text.replace('
', '\n') - return bs(text).text + return bs(text, 'html.parser').text async def parse(self, raw_post: RawPost) -> Post: info = raw_post['mblog'] diff --git a/src/plugins/hk_reporter/post.py b/src/plugins/hk_reporter/post.py index 255c7b6..4609e6c 100644 --- a/src/plugins/hk_reporter/post.py +++ b/src/plugins/hk_reporter/post.py @@ -2,6 +2,7 @@ import base64 from dataclasses import dataclass, field from io import BytesIO from typing import NoReturn, Optional +from nonebot import logger import httpx from PIL import Image from .plugin_config import plugin_config @@ -47,11 +48,12 @@ class Post: if len(images) == 6: matrix = (3, 2) self.pics = self.pics[6:] - elif len(images) == 9: + elif len(images) >= 9: matrix = (3, 3) self.pics = self.pics[9:] else: return + logger.info('trigger merge image') target = Image.new('RGB', (matrix[0] * pic_size, matrix[1] * pic_size)) for y in range(matrix[1]): for x in range(matrix[0]):